from pprint import pprint

from transfer.models import EstablishmentBacklinks, WineColor
from transfer.serializers.partner import PartnerSerializer
from transfer.serializers.product import WineColorSerializer


def transfer_partner():
    queryset = 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 = WineColor.objects.all()
    serialized_data = WineColorSerializer(data=list(queryset.values()),
                                          many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"CharacteristicSerializer errors: {serialized_data.errors}")


data_types = {
    "partner": [transfer_partner],
    "wine_color": [transfer_wine_color],
}