Implement flask-migrate

open-cogsci · Jan 10, 2025 · 45e99d8 · 45e99d8
1 parent 0c459fa
commit 45e99d8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,6 +20,7 @@ dependencies = [
     "flask-sqlalchemy",
     "flask-wtf",
     "flask-cors",
+    "flask-migrate",
     "jinja2",
     "jq",
     "langchain",

diff --git a/sigmund/server.py b/sigmund/server.py
@@ -2,6 +2,7 @@
 os.environ['USE_FLASK_SQLALCHEMY'] = '1'
 from flask import Flask, Config, request
 from flask_login import LoginManager
+from flask_migrate import Migrate
 from . import config
 from .routes import api_blueprint, app_blueprint, User, subscribe_blueprint, \
     google_login_blueprint, public_blueprint
@@ -30,7 +31,8 @@ def create_app(config_class=SigmundConfig):
         db.create_all()
     # Initialize login manager
     login_manager = LoginManager()
-
+    # Set up database migration
+    migrate = Migrate(app, db)
     @login_manager.user_loader
     def load_user(user_id):
         return User(user_id)