sharix-guide-webservice-running/serializer/orders_local.py

from rest_framework import serializers
from webservice_running.models import OrdersLocal
from tickets.models import *
from dbsynce.models import Orders
# from dbsynce.serializer import OrdersSerializer
from rest_framework.utils import model_meta
import pprint
from django.db import connection


class OrderNotDefined(Exception):
    """
    If order_synced field is not defined throw exception
    """
    def __init__(self, message="order_synced is not defined"):
        self.message = message
        super().__init__(self.message)


class OrdersLocalSerializer(serializers.ModelSerializer):
    #service = serializers.IntegerField(source="order_synced.service.pk", allow_null=True)    
    #service_type = serializers.IntegerField(source="order_synced.service_type.pk", allow_null=True) 
    #state = serializers.CharField(source="order_synced.state", allow_null=True) 
    #id_metaservice = serializers.IntegerField(source="order_synced.id_metaservice", allow_null=True) 
    
    #provider = serializers.IntegerField(source="order_synced.provider.pk", allow_null=True) 
    #provider_screen_name = serializers.CharField(source="order_synced.provider.user_id.full_name", allow_null=True) 

    #receiver = serializers.IntegerField(source="order_synced.receiver.pk", allow_null=True)
    #receiver_screen_name = serializers.CharField(source="order_synced.receiver.full_name", allow_null=True) 
    
    #client_id = serializers.CharField(source="order_synced.client_id.pk", allow_null=True) 
    #client_screen_name = serializers.CharField(source="order_synced.client.user.full_name", allow_null=True) 

    #time_created = serializers.DateTimeField(source="order_synced.time_created", allow_null=True) 
    #time_placed = serializers.DateTimeField(source="order_synced.time_placed", allow_null=True) 
    #time_start = serializers.DateTimeField(source="order_synced.time_start", allow_null=True) 
    #time_start_real = serializers.DateTimeField(source="order_synced.time_start_real", allow_null=True)
    #time_start_predicted = serializers.DateTimeField(source="order_synced.time_start_predicted", allow_null=True)
    #time_finish_predicted = serializers.DateTimeField(source="order_synced.time_finish_predicted", allow_null=True) 
    #time_finish_real = serializers.DateTimeField(source="order_synced.time_finish_real", allow_null=True) 
    
    #predicted_price = serializers.FloatField(source="order_synced.predicted_price", allow_null=True) 
    #real_price = serializers.FloatField(source="order_synced.real_price", allow_null=True) 
    #is_global = serializers.BooleanField(source="order_synced.is_global", allow_null=True) 
    #is_visible = serializers.BooleanField(source="order_synced.is_visible", allow_null=True) 
    #ticket = serializers.IntegerField(source="order_synced.ticket.pk", allow_null=True) 
    #status = serializers.IntegerField(source="order_synced.ticket.stauts", allow_null=True)
    #priority = serializers.IntegerField(source="order_synced.ticket.priority", allow_null=True)
    #title = serializers.CharField(source="order_synced.ticket.title", allow_null=True)
    #note = serializers.CharField(source="order_synced.ticket.note", allow_null=True)
    #due_date = serializers.DateField(source="order_synced.ticket.due_date", allow_null=True)
    #status_changed_date = serializers.DateTimeField(source="order_synced.ticket.status_changed_date", allow_null=True)
    # order_synced = OrdersSerializer(read_only=True)
    service_date = serializers.DateTimeField(source="order_synced.service_date", allow_null=True)
    service_time = serializers.TimeField(source="order_synced.service_time", allow_null=True)
    discription = serializers.CharField(source="order_synced.discription", allow_null=True)
    price = serializers.IntegerField(source="order_synced.price", allow_null=True)   
    name_order = serializers.CharField(source="order_synced.name_order", allow_null=True)
    comment = serializers.CharField(source="order_synced.comment", allow_null=True)
    place_start = serializers.CharField(source="order_synced.place_start", allow_null=True)
    place_end = serializers.CharField(source="order_synced.place_end", allow_null=True)
    guide_type = serializers.IntegerField(source="order_synced.guide_type", allow_null=True)
    movement_way = serializers.IntegerField(source="order_synced.movement_way", allow_null=True)
    numb_of_people = serializers.IntegerField(source="order_synced.numb_of_people", allow_null=True)

    class Meta:
        model = OrdersLocal
        fields = "__all__"
        # fields = ["order", "order_synced"]
        # depth = 2


    def update(self, instance, validated_data):
        """
        Normally order_synced is not null, but in testing it is null=True, so it can produce exceptions
        """

        order_instance = instance.order_synced
        if not order_instance:
            raise OrderNotDefined()
        ticket_instance = instance.order_synced.ticket

        for attr, value in validated_data.items():
            if attr == "order_synced":
                for order_synced_attr, order_synced_value in value.items():
                    if order_synced_attr == "ticket":
                        for ticket_attr, ticket_value in order_synced_value.items():
                            setattr(ticket_instance, ticket_attr, ticket_value)
                            continue
                    else:
                        setattr(order_instance, order_synced_attr, order_synced_value)
            else:
                setattr(instance, attr, value)

        instance.save()
        ticket_instance.save()
        order_instance.save()
        return instance