add groups views

1 files changed, 25 insertions, 1 deletions

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