Internal Database Fix

This commit is contained in:
2026-08-26 11:39:31 -05:00
parent 18be8588f6
commit 6f271f8c30
+30 -4
View File
@@ -12,11 +12,11 @@ needs to change.
"""
from __future__ import annotations
from sqlalchemy import create_engine
from sqlalchemy import create_engine, inspect, text
from sqlalchemy.orm import sessionmaker, Session
from app import config
from app.models import Base
from app.models import Base, Order
_engine = None
_SessionLocal = None
@@ -42,8 +42,34 @@ def get_session_factory():
def init_db() -> None:
"""Create tables that don't exist yet. Safe to call every startup."""
Base.metadata.create_all(get_engine())
"""Create tables that don't exist yet, and patch existing ones with
any columns added to the model since they were created. Safe to call
every startup - this is a local cache rebuilt from JIRA/ShipStation,
not a system of record, so a simple additive auto-migration is
enough; we don't need a full migration framework for it."""
engine = get_engine()
Base.metadata.create_all(engine)
_add_missing_columns(engine)
def _add_missing_columns(engine) -> None:
inspector = inspect(engine)
if "orders" not in inspector.get_table_names():
return # create_all() just made it fresh - nothing to patch
existing_columns = {col["name"] for col in inspector.get_columns("orders")}
missing = [c for c in Order.__table__.columns if c.name not in existing_columns]
if not missing:
return
with engine.begin() as conn:
for column in missing:
col_type = column.type.compile(dialect=engine.dialect)
# Always added as nullable, regardless of what the model says -
# existing rows get backfilled with real values the next time
# they're touched by an import, and this avoids ALTER TABLE
# failing on a NOT NULL column with no default for old rows.
conn.execute(text(f'ALTER TABLE orders ADD COLUMN "{column.name}" {col_type}'))
def get_session() -> Session: