from pprint import pprint

from transfer import models as transfer_models
from transfer.serializers.partner import PartnerSerializer
from transfer.serializers import product as product_serializers


def transfer_partner():
    queryset = transfer_models.EstablishmentBacklinks.objects.filter(type="Partner")

    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}")


def transfer_wine_color():
    queryset = transfer_models.WineColor.objects.all()
    serialized_data = product_serializers.WineColorSerializer(
        data=list(queryset.values()),
        many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"WineColorSerializer errors: {serialized_data.errors}")


def transfer_wine_standard():
    queryset = transfer_models.WineClassification.objects.filter(parent_id__isnull=True)
    serialized_data = product_serializers.WineStandardSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"WineStandardSerializer errors: {serialized_data.errors}")


def transfer_wine_classifications():
    queryset = transfer_models.WineClassification.objects.filter(parent_id__isnull=False)
    serialized_data = product_serializers.WineStandardClassificationSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"WineStandardClassificationSerializer errors: {serialized_data.errors}")


data_types = {
    "partner": [transfer_partner],
    "wine_color": [transfer_wine_color],
    "wine_standard": [transfer_wine_standard],
    "wine_classification": [transfer_wine_classifications],
}