Added notifications table

Author: Seluj78

backend/PyMatcha/models/notification.py

@@ -0,0 +1,50 @@
+"""
+    PyMatcha - A Python Dating Website
+    Copyright (C) 2018-2019 jlasne/gmorer
+    <jlasne@student.42.fr> - <lauris.skraucis@gmail.com>
+
+    This program is free software: you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation, either version 3 of the License, or
+    (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program.  If not, see <https://www.gnu.org/licenses/>.
+"""
+from __future__ import annotations
+
+import logging
+from datetime import datetime
+
+from PyMatcha.utils import create_notifications_table
+from PyMatcha.utils.orm import Field
+from PyMatcha.utils.orm import Model
+
+
+class Notification(Model):
+    table_name = "notifications"
+
+    id = Field(int, modifiable=False)
+    dt_received = Field(datetime, fmt="%Y-%m-%d %H:%M:%S")
+    content = Field(str)
+    type = Field(str)
+    is_seen = Field(bool)
+    link_to = Field(str)
+
+    @staticmethod
+    def create(
+        content: str, type: str, link_to: 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)
+        new_notif.save()
+        logging.debug("Creating new notification")
+        return new_notif
+
+    @classmethod
+    def create_table(cls):
+        create_notifications_table(cls.db)

backend/PyMatcha/utils/__init__.py

@@ -4,6 +4,7 @@
 from PyMatcha.utils.tables import _create_likes_table
 from PyMatcha.utils.tables import _create_matches_table
 from PyMatcha.utils.tables import _create_messages_table
+from PyMatcha.utils.tables import _create_notifications_table
 from PyMatcha.utils.tables import _create_reports_table
 from PyMatcha.utils.tables import _create_tags_table
 from PyMatcha.utils.tables import _create_user_table
@@ -19,6 +20,7 @@
 create_messages_table = _create_messages_table
 create_images_table = _create_images_table
 create_blocks_table = _create_blocks_table
+create_notifications_table = _create_notifications_table
 
 __all__ = [
     "hash_password",
@@ -32,4 +34,5 @@
     "create_messages_table",
     "create_images_table",
     "create_blocks_table",
+    "create_notifications_table",
 ]

backend/PyMatcha/utils/tables.py

@@ -192,7 +192,7 @@ def _create_images_table(db):
         user_id        INT NOT NULL,
         timestamp      TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
         link           VARCHAR(256) NOT NULL,
-        is_primary       BOOLEAN DEFAULT FALSE
+        is_primary     BOOLEAN DEFAULT FALSE
         ) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci ;
         """
         )
@@ -219,6 +219,27 @@ def _create_blocks_table(db):
         c.close()
 
 
+def _create_notifications_table(db):
+    with db.cursor() as c:
+        logging.info("Creating table notifications.")
+        c.execute(DISABLE_SQL_NOTES)
+        c.execute(
+            """
+        CREATE TABLE IF NOT EXISTS notifications
+        (
+        id            INT auto_increment PRIMARY KEY,
+        dt_received   DATETIME DEFAULT NOW(),
+        content       LONGTEXT NOT NULL,
+        type          ENUM('match', 'like', 'superlike', 'unlike', 'view', 'message'),
+        is_seen       BOOLEAN DEFAULT FALSE,
+        link_to       VARCHAR(256)
+        ) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci ;
+        """
+        )
+        c.execute(ENABLE_SQL_NOTES)
+        c.close()
+
+
 def create_tables(db):
     _create_user_table(db)
     _create_tags_table(db)
@@ -229,3 +250,4 @@ def create_tables(db):
     _create_messages_table(db)
     _create_images_table(db)
     _create_blocks_table(db)
+    _create_notifications_table(db)