52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
from pathlib import Path
|
|
import sys
|
|
|
|
from sqlalchemy import select, text
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from app.database import SessionLocal, engine # noqa: E402
|
|
from app.models import AttendancePoint # noqa: E402
|
|
from app.services.misc_work import ensure_misc_work_product # noqa: E402
|
|
|
|
|
|
def _column_exists(conn, table_name: str, column_name: str) -> bool:
|
|
dialect = conn.dialect.name
|
|
if dialect == "mysql":
|
|
return bool(conn.execute(
|
|
text("""
|
|
SELECT COUNT(*)
|
|
FROM information_schema.columns
|
|
WHERE table_schema = DATABASE()
|
|
AND table_name = :table_name
|
|
AND column_name = :column_name
|
|
"""),
|
|
{"table_name": table_name, "column_name": column_name},
|
|
).scalar_one())
|
|
rows = conn.execute(text(f"PRAGMA table_info({table_name})")).all()
|
|
return any(row[1] == column_name for row in rows)
|
|
|
|
|
|
def main() -> None:
|
|
with engine.begin() as conn:
|
|
if not _column_exists(conn, "production_report_items", "remark"):
|
|
conn.execute(text("""
|
|
ALTER TABLE production_report_items
|
|
ADD COLUMN remark VARCHAR(500) NULL
|
|
"""))
|
|
|
|
with SessionLocal() as db:
|
|
points = db.scalars(
|
|
select(AttendancePoint).where(AttendancePoint.is_active.is_(True))
|
|
).all()
|
|
for point in points:
|
|
ensure_misc_work_product(db, point.name)
|
|
db.commit()
|
|
|
|
print(f"misc work migrated, attendance_points={len(points)}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|