from django.db import models from accounts.models import IAmPrincipal # Create your models here. class ChatGroupManager(models.Manager): def get_or_create_group(self, game_id): group_name = f"game_{game_id}" chat_group, created = self.get_or_create(name=group_name) return chat_group class ChatGroup(models.Model): name = models.CharField(max_length=255, unique=True) objects = ChatGroupManager() def __str__(self): return self.name class ChatMessage(models.Model): group = models.ForeignKey(ChatGroup, on_delete=models.CASCADE) message = models.TextField() timestamp = models.DateTimeField(auto_now_add=True) user = models.ForeignKey( IAmPrincipal, related_name="PrincipalChat", on_delete=models.CASCADE ) def __str__(self): return f"{self.user} - {self.message}"