Django channels add object at wrong model - django

Question

I am making a chat app with django channels and it has 2 consumers. One is for the team chat and the other for personal. I am adding history for the chat. There are two models. "Chat" and "Chat2". Chat is for team and Chat2 for personal. The history is working with the team chat and the messages are added to the "Chat" model. But with personal it is added to "Chat" instead of "Chat2".
Here are the models:
class Chat(models.Model):
team_id = models.ForeignKey(Team,on_delete=models.CASCADE,null=True)
username = models.CharField(max_length=200,null=True)
message = models.CharField(max_length=200,null=True)
profile_pic = models.ImageField(upload_to='chat/',null=True)
class Chat2(models.Model):
username = models.CharField(max_length=200,null=True)
username2 = models.CharField(max_length=200,null=True)
message = models.CharField(max_length=200,null=True)
profile_pic = models.ImageField(upload_to='chat/',null=True)
room_name1 = models.CharField(max_length=200,null=True)
And here are the two consumers(The consumers are in the same file but I add it separately for cleaner code):
class ChatConsumer(AsyncWebsocketConsumer):
#sync_to_async
def get_profile(self):
return self.scope['user'].profile
async def connect(self):
self.room_name = self.scope['url_route']['kwargs']['room_name']
self.room_group_name = 'chat_%s' % self.room_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']
username = text_data_json['profile_name']
team = text_data_json['team']
profile = await self.get_profile()
profile_pic = profile.profile_pic.url
await self.save_messages(message, team, username, profile_pic)
# Send message to room group
await self.channel_layer.group_send(
self.room_group_name,
{
'type': 'chat_message',
'message': message,
'username':username,
'team':team,
'profile_pic':profile_pic
}
)
# Receive message from room group
async def chat_message(self, event):
message = event['message']
team = event['team']
username = event['username']
profile_pic = event['profile_pic']
# Send message to WebSocket
await self.send(text_data=json.dumps({
'message': message,
'team':team,
'username':username,
'profile_pic':profile_pic,
'type':'chat'
}))
#sync_to_async
def save_messages(self,message,team,username,profile_pic):
team1 = Team.objects.filter(team_name=team).first()
chat = Chat(team_id=team1,message=message,username=username,profile_pic=profile_pic)
chat.save()
The other consumer:
class Chat2Consumer(AsyncWebsocketConsumer):
#sync_to_async
def get_profile(self):
return self.scope['user'].profile
async def connect(self):
self.room_name = self.scope['url_route']['kwargs']['room_name']
self.room_group_name = 'chat_%s' % self.room_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']
username = text_data_json['profile_name']
room_name = text_data_json['room_name']
profile = await self.get_profile()
profile_pic = profile.profile_pic.url
await self.save_messages1(message,username,profile_pic)
# Send message to room group
await self.channel_layer.group_send(
self.room_group_name,
{
'type': 'chat_message',
'message': message,
'username':username,
'profile_pic':profile_pic,
}
)
# Receive message from room group
async def chat_message(self, event):
message = event['message']
username = event['username']
profile_pic = event['profile_pic']
# Send message to WebSocket
await self.send(text_data=json.dumps({
'message': message,
'username':username,
'profile_pic':profile_pic,
'type':'chat'
}))
#sync_to_async
def save_messages(self,message,username,profile_pic,username2):
Chat2.objects.create(message=message,username=username,profile_pic=profile_pic)
And routing.py
websocket_urlpatterns = [
re_path(r'ws/chat/(?P<room_name>\w+)/$', consumers.ChatConsumer.as_asgi()),
re_path(r'ws/chat2/(?P<room_name>\w+)/$', consumers.Chat2Consumer.as_asgi()),
]