Add notif page

migrations/add_notifs_table.py

@@ -0,0 +1,61 @@
+import os
+import sys
+from pathlib import Path
+
+# Add the parent directory to Python path so we can import from root
+sys.path.append(str(Path(__file__).parent.parent))
+
+from flask import Flask
+from flask_sqlalchemy import SQLAlchemy
+from extensions import db
+from sqlalchemy import text
+
+def upgrade():
+    # Create notifs table
+    with db.engine.connect() as conn:
+        conn.execute(text('''
+            CREATE TABLE IF NOT EXISTS notifs (
+                id SERIAL PRIMARY KEY,
+                notif_type VARCHAR(50) NOT NULL,
+                user_id INTEGER NOT NULL REFERENCES "user" (id),
+                sender_id INTEGER REFERENCES "user" (id),
+                timestamp TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+                read BOOLEAN NOT NULL DEFAULT FALSE,
+                details JSONB
+            );
+            
+            -- Create indexes for faster queries
+            CREATE INDEX IF NOT EXISTS idx_notifs_notif_type ON notifs(notif_type);
+            CREATE INDEX IF NOT EXISTS idx_notifs_timestamp ON notifs(timestamp);
+            CREATE INDEX IF NOT EXISTS idx_notifs_user_id ON notifs(user_id);
+            CREATE INDEX IF NOT EXISTS idx_notifs_sender_id ON notifs(sender_id);
+            CREATE INDEX IF NOT EXISTS idx_notifs_read ON notifs(read);
+        '''))
+        conn.commit()
+
+def downgrade():
+    # Drop notifs table and its indexes
+    with db.engine.connect() as conn:
+        conn.execute(text('''
+            DROP INDEX IF EXISTS idx_notifs_notif_type;
+            DROP INDEX IF EXISTS idx_notifs_timestamp;
+            DROP INDEX IF EXISTS idx_notifs_user_id;
+            DROP INDEX IF EXISTS idx_notifs_sender_id;
+            DROP INDEX IF EXISTS idx_notifs_read;
+            DROP TABLE IF EXISTS notifs;
+        '''))
+        conn.commit()
+
+if __name__ == '__main__':
+    app = Flask(__name__)
+    
+    # Use the same database configuration as in app.py
+    app.config['SQLALCHEMY_DATABASE_URI'] = os.getenv('DATABASE_URL', 'postgresql://postgres:1253@localhost:5432/docupulse')
+    app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+    
+    print("Connecting to database...")
+    
+    db.init_app(app)
+    
+    with app.app_context():
+        upgrade()