from django.db import models from django.db.models import IntegerField from django.contrib.auth.models import User from django.contrib import admin # Create your models here. class ChatChannelModel(models.Model): def create(self, users_id: [int]): self.save() for user_id in users_id: ChatMemberModel(channel_id = self.pk, member_id = user_id).save() def get_members_id(self): return [member_channel.member_id for member_channel in ChatMemberModel.objects.filter(channel_id = self.pk)] class ChatMemberModel(models.Model): member_id = IntegerField(primary_key=False) channel_id = IntegerField(primary_key=False) def __str__(self): return "member_id: " + str(self.member_id) + ", channel_id: " + str(self.channel_id) class ChatMessageModel(models.Model): channel_id = IntegerField(primary_key=False) author_id = IntegerField(primary_key=False) content = models.CharField(max_length=255) time = IntegerField(primary_key=False) def __str__(self): return "author_id: " + str(self.author_id) + ", channel_id: " + str(self.channel_id) + ", content: " + self.content