gault-millau/apps/product/transfer_data.py

from pprint import pprint

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


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"transfer_wine_color errors: {serialized_data.errors}")


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


def transfer_wine_bottles_produced():
    raw_queryset = transfer_models.Products.objects.raw(
        """
        SELECT
           DISTINCT bottles_produced,
           1 as id
        FROM products
        WHERE bottles_produced IS NOT NULL 
        AND bottles_produced !='';
        """
    )
    queryset = [vars(query) for query in raw_queryset]
    serialized_data = product_serializers.WineBottlesProducedSerializer(
      data=queryset,
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"transfer_wine_bottles_produced errors: {serialized_data.errors}")


def transfer_wine_classification_type():
    raw_queryset = transfer_models.ProductClassification.objects.raw(
        """
        SELECT
           DISTINCT name,
           1 as id
        FROM wine_classifications;
        """
    )
    queryset = [vars(query) for query in raw_queryset]
    serialized_data = product_serializers.WineClassificationTypeSerializer(
      data=queryset,
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"transfer_classification errors: {serialized_data.errors}")


def transfer_wine_standard():
    queryset = transfer_models.ProductClassification.objects.filter(parent_id__isnull=True) \
                                                         .exclude(type='Classification')
    serialized_data = product_serializers.ProductStandardSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"transfer_wine_standard errors: {serialized_data.errors}")


def transfer_wine_classifications():
    queryset = transfer_models.ProductClassification.objects.filter(type='Classification')
    serialized_data = product_serializers.ProductClassificationSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        pprint(f"transfer_wine_classifications errors: {serialized_data.errors}")


def transfer_product():
    queryset = transfer_models.Products.objects.all()
    serialized_data = product_serializers.ProductSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        errors = []
        for d in serialized_data.errors: errors.append(d) if d else None
        pprint(f"transfer_product errors: {errors}")


def transfer_plate():
    queryset = transfer_models.Merchandise.objects.all()
    serialized_data = product_serializers.PlateSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        errors = []
        for d in serialized_data.errors: errors.append(d) if d else None
        pprint(f"transfer_plates errors: {errors}")


def transfer_plate_image():
    queryset = transfer_models.Merchandise.objects.all()
    serialized_data = product_serializers.PlateImageSerializer(
      data=list(queryset.values()),
      many=True)
    if serialized_data.is_valid():
        serialized_data.save()
    else:
        errors = []
        for d in serialized_data.errors: errors.append(d) if d else None
        pprint(f"transfer_plates_images errors: {errors}")


data_types = {
    "partner": [transfer_partner],
    "wine_characteristics": [
        transfer_wine_sugar_content,
        transfer_wine_color,
        transfer_wine_bottles_produced,
        transfer_wine_classification_type,
        transfer_wine_standard,
        transfer_wine_classifications,
        ],
    "product": [
        transfer_product,
        # transfer_plate,
        # transfer_plate_image,
    ],
}