Каналы Django добавляют объект в неправильную модель

Я делаю приложение чата с каналами django и у него есть 2 потребителя. Один для командного чата, а другой для личного. Я добавляю историю для чата. Есть две модели. "Chat" и "Chat2". Chat предназначен для командного чата, а Chat2 - для личного. История работает с командным чатом, и сообщения добавляются в модель "Чат". Но в личном чате сообщения добавляются в "Чат", а не в "Чат2".

Вот модели:

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)

А вот два потребителя (потребители находятся в одном файле, но я добавляю их отдельно для более чистого кода):

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()

Другой потребитель:

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)

И 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()),
]

Вернуться на верх