File size: 2,388 Bytes
727a40a | 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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 | import json
from channels.generic.websocket import AsyncWebsocketConsumer
from django.utils import timezone
from channels.db import database_sync_to_async
# Define an async function to save the message to the database
@database_sync_to_async
def save_message(sender_name, receiver_name, message):
from django.contrib.auth.models import User
from chat.models import Messages
# Get User instances for sender and receiver
sender = User.objects.get(username=sender_name)
receiver = User.objects.get(username=receiver_name)
# Create a new message
new_message = Messages(
description=message,
sender_name=sender,
receiver_name=receiver,
time=timezone.now().time(), # Store the current time
timestamp=timezone.now(), # Store the timestamp
seen=True
)
new_message.save()
class ChatConsumer(AsyncWebsocketConsumer):
async def connect(self):
self.room_name = self.scope['url_route']['kwargs']['room_name']
self.room_group_name = f'chat_{self.room_name}'
# print(self.room_name, self.room_group_name)
# Join room group
await self.channel_layer.group_add(
self.room_group_name,
self.channel_name
)
await self.accept()
async def disconnect(self, close_code):
# Leave room group
await self.channel_layer.group_discard(
self.room_group_name,
self.channel_name
)
# Receive message from WebSocket
async def receive(self, text_data):
text_data_json = json.loads(text_data)
message = text_data_json['message']
sender_name = text_data_json['sender_name']
receiver_name = text_data_json['receiver_name']
await save_message(sender_name, receiver_name, message)
# Send message to room group
await self.channel_layer.group_send(
self.room_group_name,
{
'type': 'chat_message',
'message': message,
'sender': sender_name
}
)
# Receive message from room group
async def chat_message(self, event):
message = event['message']
sender = event['sender']
# Send message to WebSocket
await self.send(text_data=json.dumps({
'message': message,
'sender': sender
}))
|