chats

Main/models/__init__.py

@@ -9,3 +9,5 @@ from Main.models.extrafile import ExtraFile
 from Main.models.progress import Progress
 from Main.models.solution_file import SolutionFile
 from Main.models.friendship import Friendship
+from Main.models.chat import Chat
+from Main.models.message import Message

Main/models/chat.py

@@ -0,0 +1,8 @@
+from django.contrib.auth.models import User
+from django.db import models
+
+
+class Chat(models.Model):
+    from_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="from_chat")
+    to_user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="to_chat")
+    user = None

Main/models/message.py

@@ -0,0 +1,13 @@
+from django.contrib.auth.models import User
+from django.db import models
+from django.utils import timezone
+
+from Main.models.chat import Chat
+
+
+class Message(models.Model):
+    user = models.ForeignKey(User, on_delete=models.CASCADE)
+    chat = models.ForeignKey(Chat, on_delete=models.CASCADE)
+    text = models.TextField()
+    read = models.BooleanField(default=False)
+    time_sent = models.DateTimeField(default=timezone.now)

Main/models/userinfo.py

@@ -26,6 +26,7 @@ class UserInfo(models.Model):
     telegram_chat_id = models.TextField(default="")
     notification_solution_result = models.BooleanField(default=False)
     notification_friends = models.BooleanField(default=False)
+    notification_messages = models.BooleanField(default=False)
     code = models.IntegerField(null=True)
     verified = models.BooleanField(default=False)