sharix-open-backend/serializer/orders.py

from rest_framework import serializers
from dbsynce.models import Orders, Client
from dbsynce.ejabber import *
from tickets.models import Task, TaskList
from core.config import DEBUG, WEBSERVICE_NAME, EJ_HOST, EJ_SERVICE
import traceback
import requests
    

class OrdersSerializer(serializers.ModelSerializer):
    class Meta:
        model = Orders
        fields = '__all__'

    
    def create(self, validated_data):
        ModelClass = self.Meta.model
        
        
        if "ticket" not in validated_data:
            # Create ticket with minimal required data
            task_data = {
                "task_list": TaskList.objects.get(pk=9), # MS-SUP входящие обычные заявки ST_REQUEST
                "status": None,
                "created_date": "2000-01-01", # mandatory
                "status_changed_date": None,
                "created_by": None,
                "priority": 0, # default: 0
                "type": None,
                "title": "New ticket title", # mandatory
                "note": None,
                "due_date": None,
                "assigned_to": None,
            }
            ticket_instance = Task._default_manager.create(**task_data)
            validated_data["ticket"] = ticket_instance

        try:
            instance = ModelClass._default_manager.create(**validated_data)
        except TypeError:
            tb = traceback.format_exc()
            msg = (
                'Got a `TypeError` when calling `%s.%s.create()`. '
                'This may be because you have a writable field on the '
                'serializer class that is not a valid argument to '
                '`%s.%s.create()`. You may need to make the field '
                'read-only, or override the %s.create() method to handle '
                'this correctly.\nOriginal exception was:\n %s' %
                (
                    ModelClass.__name__,
                    ModelClass._default_manager.name,
                    ModelClass.__name__,
                    ModelClass._default_manager.name,
                    self.__class__.__name__,
                    tb
                )
            )
            raise TypeError(msg)
        else:
            if DEBUG: print("[CREATING EJABBER ROOMS]")
            # Get neccessary data
            order_num = instance.pk
            support = pick_support()
            client = instance.client_id.user.phone_number
            provider = instance.provider.user_id.phone_number

            for room_name, room_members in get_rooms(WEBSERVICE_NAME, order_num, support, client, provider).items():
                # Create ejabber rooms
                data = {
                    "name": room_name,
                    "service": EJ_SERVICE,
                    "host": EJ_HOST,
                }
                res = ej_execute("create_room", data)
                if DEBUG: print(f"[RESPONSE CREATED]: {res.json()}")
                if DEBUG: print(f"[ROOM]: {room_name}")

                # Set rooms to members_only
                data = {
                    "name": room_name,
                    "service": EJ_SERVICE,
                    "option": "members_only",
                    "value": "true"
                }
                res = ej_execute("change_room_option", data)
                if DEBUG: print(f"[SET ROOM TO MEMBERS ONLY]: {res.json()}")

                """
                Add members to the room

                member_account is a dict that contains members credentials
                Key - the same thing that is in room_members
                Value - tuple (jid, ejabber affiliation).
                Affiliation can be one of: owner, admin, member, outcast, none.
                owner > admin
                """
                members_accounts = {
                    "owner": (WEBSERVICE_NAME + "_backend@" + EJ_HOST, "owner"),
                    "client": (client, "member"),
                    "support": (support, "member"),
                    "provider": (provider, "member"),
                }
                data = {
                    "name": room_name,
                    "service": EJ_SERVICE,
                    "jid": WEBSERVICE_NAME + "_backend@" + EJ_HOST,
                    "affiliation": "owner"
                }
                # Add all neccessary member to the room
                for member in room_members:
                    data["jid"] = members_accounts[member][0]
                    data["affiliation"] = members_accounts[member][1]
                    res = ej_execute("set_room_affiliation", data)
                    if DEBUG: print(f"[ADDED USER]: {res.json()}")
        return instance