blob: be1cf2659c168100db767e28ace3e66e82ec64da (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
|
import json
from channels.generic.websocket import WebsocketConsumer
from asgiref.sync import async_to_sync
class ChatConsumer(WebsocketConsumer):
def connect(self):
self.room_group_name = 'chat'
async_to_sync(self.channel_layer.group_add)(
self.room_group_name,
self.channel_name
)
self.accept()
def receive(self, text_data):
text_data_json = json.loads(text_data)
message = text_data_json['message']
username = text_data_json['username']
async_to_sync(self.channel_layer.group_send)(
self.room_group_name,
{
'type': 'chat',
'message': message,
'username': username
}
)
def chat(self, event):
message = event['message']
username = event['username']
self.send(text_data=json.dumps({
'message': message,
'username': username
}))
def disconnect(self, close_code):
async_to_sync(self.channel_layer.group_discard)(
self.room_group_name,
self.channel_name
)
|
