Added for who the message is and another message type

Author: Seluj78

backend/PyMatcha/models/notification.py

@@ -20,6 +20,7 @@
 
 import logging
 from datetime import datetime
+from typing import Optional
 
 from PyMatcha.utils import create_notifications_table
 from PyMatcha.utils.orm import Field
@@ -30,6 +31,7 @@ class Notification(Model):
     table_name = "notifications"
 
     id = Field(int, modifiable=False)
+    user_id = Field(int)
     dt_received = Field(datetime, fmt="%Y-%m-%d %H:%M:%S")
     content = Field(str)
     type = Field(str)
@@ -38,9 +40,20 @@ class Notification(Model):
 
     @staticmethod
     def create(
-        content: str, type: str, link_to: str, is_seen: bool = False, dt_received: datetime = datetime.utcnow()
+        user_id: int,
+        content: str,
+        type: str,
+        link_to: Optional[str],
+        is_seen: bool = False,
+        dt_received: datetime = datetime.utcnow(),
     ) -> Notification:
-        new_notif = Notification(content=content, type=type, link_to=link_to, is_seen=is_seen, dt_received=dt_received)
+        if type not in ["match", "like", "superlike", "unlike", "view", "message", "message_like"]:
+            raise ValueError(
+                "type must be one of ['match', 'like', 'superlike', 'unlike', 'view', 'message', 'message_like']"
+            )
+        new_notif = Notification(
+            user_id=user_id, content=content, type=type, link_to=link_to, is_seen=is_seen, dt_received=dt_received
+        )
         new_notif.save()
         logging.debug("Creating new notification")
         return new_notif

backend/PyMatcha/utils/tables.py

@@ -228,9 +228,10 @@ def _create_notifications_table(db):
         CREATE TABLE IF NOT EXISTS notifications
         (
         id            INT auto_increment PRIMARY KEY,
+        user_id       INT NOT NULL,
         dt_received   DATETIME DEFAULT NOW(),
         content       LONGTEXT NOT NULL,
-        type          ENUM('match', 'like', 'superlike', 'unlike', 'view', 'message'),
+        type          ENUM('match', 'like', 'superlike', 'unlike', 'view', 'message', 'message_like'),
         is_seen       BOOLEAN DEFAULT FALSE,
         link_to       VARCHAR(256)
         ) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci ;