tripwithbonus/ChatServiceApp/websocket_views.py

import json
from channels.generic.websocket import WebsocketConsumer, AsyncWebsocketConsumer, JsonWebsocketConsumer
from asgiref.sync import async_to_sync, sync_to_async
# from channels.auth import channel_session_user, channel_session_user_from_http
from channels.layers import get_channel_layer
from django.template.loader import render_to_string

def get_tickets_wo_manager_html(ticket, user, data):
    from .funcs import get_tickets_wo_manager

    # если не менеджер - возвращаем None
    if not ticket or ticket.manager.id != user.id:
        return None

    Dict = {
        'ticket': ticket,
        'tickets_wo_manager': get_tickets_wo_manager()
    }
    tickets_wo_manager_html = render_to_string('widgets/w_tickets_wo_manager.html', Dict)

    return tickets_wo_manager_html


def get_tickets_w_manager_html(ticket, user, data):
    from .funcs import get_tickets_for_manager

    # если не менеджер - возвращаем None
    if not ticket or ticket.manager.id != user.id:
        return None

    Dict = {
        'ticket': ticket,
        'tickets_w_manager': get_tickets_for_manager(user)
    }
    tickets_w_manager_html = render_to_string('widgets/w_tickets_w_manager.html', Dict)

    return tickets_w_manager_html


class ChatConsumer(WebsocketConsumer):

    # def __init__(self, *args, **kwargs):
    #     super().__init__(args, kwargs)
    #     self.room_name = None

    def connect(self):
        print('ws connect')
        self.room_group_name = 'test'

        async_to_sync(self.channel_layer.group_add)(
            f'user_{self.scope["user"].id}',
            self.channel_name
        )
        print(f'created group user_{self.scope["user"].id}')

        if self.scope['user'].is_staff:
            async_to_sync(self.channel_layer.group_add)(
                f'support_managers',
                self.channel_name
            )
            print(f'add user {self.scope["user"].id} to group support_managers')

        self.accept()




    def receive(self, text_data):
        print(f'ws receive text_data = {text_data}')

        from AuthApp.models import User
        from .models import Message, MsgGroup


        data = json.loads(text_data)

        sender = data['sender']
        receiver = data['receiver']
        sender_obj = User.objects.get(id=sender)
        receiver_obj = User.objects.get(id=receiver)



        from .funcs import send_msg, get_update_chat_Dict
        data.update({
            'cur_user': sender,
            'required_beep': False,
        })
        send_msg(data)

        ticket = None

        if 'ticket_id' in data and data['ticket_id']:
            ticket = MsgGroup.objects.get(id=data['ticket_id'])
            # receiver_obj = User.objects.get(id=receiver)
            msgs = Message.objects.filter(
                receiver__id=sender, group__id=data['ticket_id']
            )
            msgs.update(status='seen')

        Dict = get_update_chat_Dict(data)



        group_name = f'user_{sender}'
        resDict = {
            'type': 'update_chat',
            'sender': sender,
            'receiver': receiver,
            'group_name': group_name,

        }

        if ticket:
            tickets_wo_manager_html = get_tickets_wo_manager_html(ticket, sender_obj, data)
            if tickets_wo_manager_html:
                resDict.update({'tickets_wo_manager_html': tickets_wo_manager_html})

            tickets_w_manager_html = get_tickets_w_manager_html(ticket, sender_obj, data)
            if tickets_w_manager_html:
                resDict.update({'tickets_w_manager_html': tickets_w_manager_html})


        resDict.update(Dict)
        async_to_sync(self.channel_layer.group_send)(
            group_name,
            resDict
        )

        group_name = f'user_{receiver}'
        # if group_name in self.channel_layer.groups.keys():
        data.update({
            'cur_user': receiver,
            'required_beep': True,
        })
        Dict = get_update_chat_Dict(data)

        if 'support_chat_html' in Dict:
            msg_type = 'update_support_chat'
        else:
            msg_type = 'update_chat'

        resDict = {
            'type': msg_type,
            'sender': receiver,
            'receiver': sender,
            'group_name': group_name,
        }

        if ticket:
            tickets_wo_manager_html = get_tickets_wo_manager_html(ticket, receiver_obj, data)
            if tickets_wo_manager_html:
                resDict.update({'tickets_wo_manager_html': tickets_wo_manager_html})

            tickets_w_manager_html = get_tickets_w_manager_html(ticket, receiver_obj, data)
            if tickets_w_manager_html:
                resDict.update({'tickets_w_manager_html': tickets_w_manager_html})

        resDict.update(Dict)
        async_to_sync(self.channel_layer.group_send)(
            group_name,
            resDict
        )



    def echo(self, data):
        print('ws echo')

        self.send(text_data=json.dumps(data))

    def update_support_chat(self, data):
        print(f'ws update_support_chat {data["group_name"]}')

        self.send(text_data=json.dumps(data))

    def update_chat(self, data):
        print(f'ws update_chat {data["group_name"]}')

        self.send(text_data=json.dumps(data))

    def ws_send_msg(self, data):
        print('ws ws_send_msg')

        self.send(text_data=json.dumps(data))