61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
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() |