98 lines
2.6 KiB
Python
98 lines
2.6 KiB
Python
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
|
|
|
|
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}')
|
|
|
|
data = json.loads(text_data)
|
|
|
|
sender = data['sender']
|
|
receiver = data['receiver']
|
|
|
|
from .funcs import send_msg, update_chat
|
|
data.update({
|
|
'cur_user': sender,
|
|
'required_beep': False,
|
|
})
|
|
send_msg(data)
|
|
Dict = update_chat(data)
|
|
|
|
group_name = f'user_{sender}'
|
|
resDict = {
|
|
'type': 'update_chat',
|
|
'sender': sender,
|
|
'receiver': receiver,
|
|
'group_name': group_name,
|
|
|
|
}
|
|
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 = update_chat(data)
|
|
|
|
resDict = {
|
|
'type': 'update_chat',
|
|
'sender': receiver,
|
|
'receiver': sender,
|
|
'group_name': group_name,
|
|
}
|
|
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_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)) |