From eaf89ff64ce5c4e39239aec50a973d1c58a40bee Mon Sep 17 00:00:00 2001 From: jdlugosz963 Date: Fri, 26 Nov 2021 18:44:38 +0100 Subject: add groups views --- chat_api/models.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) (limited to 'chat_api/models.py') diff --git a/chat_api/models.py b/chat_api/models.py index 71a8362..5076d0b 100644 --- a/chat_api/models.py +++ b/chat_api/models.py @@ -1,3 +1,27 @@ from django.db import models +from django.contrib.auth.models import User +from django.db.models.base import Model -# Create your models here. +class DmMessage(models.Model): + sender = models.ForeignKey(User, related_name="sender_dm_message", on_delete=models.CASCADE) + reciever = models.ForeignKey(User, related_name="receiver_dm_message", on_delete=models.CASCADE) + message = models.TextField(null=False) + create_date = models.DateTimeField(auto_now=True) + + def __str__(self): + return str(self.sender) + +class Group(models.Model): + name = models.CharField(max_length=16, null=False) + owner = models.ForeignKey(User, related_name="owner_group", on_delete=models.CASCADE) + users = models.ManyToManyField(User, related_name="users_group") + create_date = models.DateTimeField(auto_now=True) + +class GmMessage(models.Model): + sender = models.ForeignKey(User, related_name="sender_gm_message", on_delete=models.CASCADE) + reciever = models.ForeignKey(Group, related_name="receiver_gm_message", on_delete=models.CASCADE) + message = models.TextField(null=False) + create_date = models.DateTimeField(auto_now=True) + + def __str__(self): + return str(self.sender) \ No newline at end of file -- cgit v1.2.3