class ChatConsumer(AsyncWebsocketConsumer):
self.room_name = self.scope['url_route']['kwargs']['room_name']
self.room_group_name = f'chat_{self.room_name}'
# Authenticate user
if isinstance(user, AnonymousUser):
await self.close()
else:
self.user = user
# Join room group
await self.channel_layer.group_add(
self.room_group_name,
self.channel_name
)
await self.accept() # Accept the WebSocket connection
async def disconnect(self, close_code):
# Leave room group
await self.channel_layer.group_discard(
self.room_group_name,
self.channel_name
)
async def receive(self, text_data):
text_data_json = json.loads(text_data)
message = text_data_json['message']
print(f"Received message: {message} from user: {self.user.username}")
# saving message to the database
await save_message_to_db(self.room_name, self.user, message )
# Send message to room group
await self.channel_layer.group_send(
self.room_group_name,
{
'type': 'chat_message',
'message': message,
'sender': self.user.username
}
)
async def chat_message(self, event):
# Send message to WebSocket
await self.send(text_data=json.dumps({
'message': event['message'],
'sender': event['sender']
}))

Comments
Post a Comment