From a2e7de0bab5b3035c55c1fc37f6cd389755fb597 Mon Sep 17 00:00:00 2001
From: Son NK <>
Date: Sat, 23 May 2020 18:50:36 +0200
Subject: [PATCH] Add Notification model

---
 app/models.py | 8 ++++++++
 server.py     | 5 +++++
 2 files changed, 13 insertions(+)

diff --git a/app/models.py b/app/models.py
index 6f75a724..68f473d4 100644
--- a/app/models.py
+++ b/app/models.py
@@ -1417,3 +1417,11 @@ class RecoveryCode(db.Model, ModelMixin):
         """Delete all recovery codes for user"""
         cls.query.filter_by(user_id=user.id).delete()
         db.session.commit()
+
+
+class Notification(db.Model, ModelMixin):
+    user_id = db.Column(db.ForeignKey(User.id, ondelete="cascade"), nullable=False)
+    message = db.Column(db.Text, nullable=False)
+
+    # whether user has marked the notification as read
+    read = db.Column(db.Boolean, nullable=False, default=False)
diff --git a/server.py b/server.py
index 47c48b22..562d391c 100644
--- a/server.py
+++ b/server.py
@@ -53,6 +53,7 @@ from app.models import (
     EmailLog,
     Referral,
     AliasMailbox,
+    Notification,
 )
 from app.monitor.base import monitor_bp
 from app.oauth.base import oauth_bp
@@ -247,6 +248,10 @@ def fake_data():
     referral = Referral.create(user_id=user.id, code="REFCODE", name="First referral")
     db.session.commit()
 
+    for i in range(6):
+        Notification.create(user_id=user.id, message=f"""Hey hey <b>{i}</b> """ * 10)
+    db.session.commit()
+
     User.create(
         email="winston@continental.com",
         name="Winston",