main: run db migrations on startup

We set an extra attribute on the alembic config to prevent it from
overriding the existing logger config in WAS. Without this, alembic
overwrites the root logger and WAS appears to hang after the migration
has completed, while it is just logging that is no longer correctly
configured.

app/const.py

@@ -1,3 +1,4 @@
+ALEMBIC_CONFIG = '/app/alembic.ini'
 DB_URL = 'sqlite:////app/storage/was.db'
 DIR_ASSET = '/app/storage/asset'
 DIR_OTA = '/app/storage/ota'

app/db/main.py

@@ -14,10 +14,6 @@
 engine = create_engine(DB_URL, echo=True, connect_args=connect_args)
 
 
-def create_db_and_tables():
-    SQLModel.metadata.create_all(engine)
-
-
 def get_config_db(config_type):
     config = WillowConfig()
 

app/main.py

@@ -1,6 +1,9 @@
 import asyncio
 import json
 import os
+
+import alembic
+import alembic.config
 from fastapi import (
     FastAPI,
     Header,
@@ -18,11 +21,12 @@
 from fastapi.middleware.cors import CORSMiddleware
 
 from app.const import (
+    ALEMBIC_CONFIG,
     DIR_OTA,
     STORAGE_USER_CONFIG,
 )
 
-from app.db.main import create_db_and_tables, migrate_user_config
+from app.db.main import migrate_user_config
 from app.internal.command_endpoints import (
     CommandEndpointResponse,
     CommandEndpointResult,
@@ -59,9 +63,16 @@
     pass
 
 
+def db_migrations():
+    cfg = alembic.config.Config(ALEMBIC_CONFIG)
+    cfg.attributes['logger'] = log
+    alembic.command.upgrade(cfg, "head")
+
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    create_db_and_tables()
+    # database schema migrations
+    db_migrations()
 
     migrate_user_files()
     get_tz_config(refresh=True)

migrations/env.py

@@ -14,7 +14,8 @@
 
 # Interpret the config file for Python logging.
 # This line sets up loggers basically.
-if config.config_file_name is not None:
+# We must skip this when config has a logger attribute to prevent the WAS logger config from being overwritten
+if config.config_file_name is not None and config.attributes.get('logger', None) is None:
     fileConfig(config.config_file_name)
 
 # add your model's MetaData object here