This commit is contained in:
aaron 2026-05-01 22:02:03 +08:00
parent b343d33466
commit 3b5ce47aed

View File

@ -100,10 +100,16 @@ def upgrade() -> None:
user_columns = {column["name"] for column in inspector.get_columns("users")}
if "class_id" in user_columns:
committee_expr = (
"committee_role" if "committee_role" in user_columns else "NULL AS committee_role"
)
permissions_expr = (
"class_permissions" if "class_permissions" in user_columns else "NULL AS class_permissions"
)
legacy_rows = bind.execute(
sa.text(
"""
SELECT id, class_id, role, committee_role, class_permissions, created_at, updated_at
f"""
SELECT id, class_id, role, {committee_expr}, {permissions_expr}, created_at, updated_at
FROM users
WHERE class_id IS NOT NULL
"""