Портал технической службы: наряды и чек-листы, задачи, отказы и неисправности, анализ надёжности, карты, склады, приложение для Android

app/migrate.py

@@ -0,0 +1,35 @@
+"""Мини-миграция: create_all не добавляет колонки в существующие таблицы — добавляем недостающие."""
+import logging
+
+from sqlalchemy import inspect, text, Boolean, Integer, Float
+
+from . import models  # noqa: F401  (регистрация таблиц)
+from . import db as _db
+from .db import Base
+
+log = logging.getLogger("migrate")
+
+
+def ensure_schema(engine=None):
+    engine = engine or _db.engine
+    Base.metadata.create_all(engine)
+    insp = inspect(engine)
+    with engine.begin() as conn:
+        for table in Base.metadata.sorted_tables:
+            have = {c["name"] for c in insp.get_columns(table.name)}
+            for col in table.columns:
+                if col.name in have:
+                    continue
+                ddl = col.type.compile(engine.dialect)
+                default = ""
+                if not col.nullable or col.default is not None:
+                    arg = getattr(col.default, "arg", None)
+                    if isinstance(col.type, Boolean):
+                        default = " DEFAULT " + ("true" if arg else "false") if engine.dialect.name == "postgresql" \
+                            else f" DEFAULT {1 if arg else 0}"
+                    elif isinstance(col.type, (Integer, Float)) and isinstance(arg, (int, float)):
+                        default = f" DEFAULT {arg}"
+                    elif isinstance(arg, str):
+                        default = " DEFAULT '" + arg.replace("'", "''") + "'"
+                log.warning("Добавляю колонку %s.%s", table.name, col.name)
+                conn.execute(text(f'ALTER TABLE {table.name} ADD COLUMN {col.name} {ddl}{default}'))