Only run migrations when needed

le717 · Oct 28, 2021 · 3702d0e · 3702d0e
1 parent 6116f9a
commit 3702d0e
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/src/app_factory.py b/src/app_factory.py
@@ -2,6 +2,8 @@
 from importlib import import_module
 import logging
 
+from alembic import command
+from alembic.config import Config
 from flask import Flask
 from flask_cors import CORS
 from flask_smorest import Api
@@ -46,13 +48,17 @@ def create_app():
     # Create a database connection
     db_path = sys_vars.get_path("DB_PATH").resolve()
     with app.app_context():
-        app.config["SQLALCHEMY_DATABASE_URI"] = f"sqlite:///{fspath(db_path)}"
+        app.config["SQLALCHEMY_DATABASE_URI"] = f"sqlite:///{db_path.as_posix()}"
         app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
-    db.init_app(app)
+        db.init_app(app)
 
-    # Create the database if needed
-    if not db_path.exists():
-        with app.app_context():
+        # Create the database if needed
+        if not db_path.exists():
             db.create_all()
 
+            # Tell Alembic this is a new database and we don't need
+            # to update it to a newer schema
+            alembic_cfg = Config("alembic.ini")
+            command.stamp(alembic_cfg, "head")
+
     return app