28 lines
838 B
Python
28 lines
838 B
Python
from __future__ import annotations
|
||
|
||
from sqlalchemy import Engine, inspect, text
|
||
|
||
from app.db.models import Base
|
||
|
||
|
||
def _has_column(engine: Engine, table: str, col: str) -> bool:
|
||
insp = inspect(engine)
|
||
cols = insp.get_columns(table)
|
||
return any(c.get("name") == col for c in cols)
|
||
|
||
|
||
def ensure_schema(engine: Engine) -> None:
|
||
"""
|
||
轻量自升级（跨 SQLite/PostgreSQL）：
|
||
- create_all 不会更新既有表结构，因此用 inspector + ALTER TABLE 补列
|
||
- 必须保证任何失败都不影响主流程（上层可选择忽略异常）
|
||
"""
|
||
Base.metadata.create_all(bind=engine)
|
||
|
||
with engine.begin() as conn:
|
||
# job_logs.run_log
|
||
if not _has_column(engine, "job_logs", "run_log"):
|
||
conn.execute(text("ALTER TABLE job_logs ADD COLUMN run_log TEXT NOT NULL DEFAULT ''"))
|
||
|
||
|