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

from rest_framework import serializers

from webservice_running.models import OrdersLocal


# from dbsynce.serializer import OrdersSerializer


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)

    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