ft_transcendence/chat/models.py

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