27 lines
857 B
Python
27 lines
857 B
Python
from __future__ import annotations
|
|
|
|
from sqlalchemy import Engine, text
|
|
|
|
from app.db.models import Base
|
|
|
|
|
|
def _has_column(conn, table: str, col: str) -> bool:
|
|
rows = conn.execute(text(f"PRAGMA table_info({table})")).fetchall()
|
|
return any(r[1] == col for r in rows) # PRAGMA columns: (cid, name, type, notnull, dflt_value, pk)
|
|
|
|
|
|
def ensure_schema(engine: Engine) -> None:
|
|
"""
|
|
SQLite 轻量自升级：
|
|
- create_all 不会更新既有表结构，因此用 PRAGMA + ALTER TABLE 补列
|
|
- 必须保证任何失败都不影响主流程（上层可选择忽略异常）
|
|
"""
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
with engine.begin() as conn:
|
|
# job_logs.run_log
|
|
if not _has_column(conn, "job_logs", "run_log"):
|
|
conn.execute(text("ALTER TABLE job_logs ADD COLUMN run_log TEXT NOT NULL DEFAULT ''"))
|
|
|
|
|