from pprint import pprint

from establishment.models import Establishment
from transfer.models import EstablishmentBacklinks
from transfer.serializers.partner import PartnerSerializer


def transfer_partner():
    establishments = Establishment.objects.filter(old_id__isnull=False).values_list('old_id', flat=True)
    queryset = EstablishmentBacklinks.objects.filter(
        establishment_id__in=list(establishments),
    ).values(
        'id',
        'establishment_id',
        # 'aasm_state',
        # 'created_at',
        # 'establishment_id',
        # 'mark',
        # 'vintage',
    )

    # queryset = EstablishmentBacklinks.objects.all() # Partner and Sponsor

    serialized_data = PartnerSerializer(data=list(queryset.values()), many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"News serializer errors: {serialized_data.errors}")


data_types = {
    "partner": [transfer_partner]
}