This commit is contained in:
aaron 2026-04-27 22:43:19 +08:00
parent 82b786f513
commit 105bde0d3c

View File

@ -1,5 +1,5 @@
""" """
Repair missing class_memberships data for databases that were partially migrated. Repair required tables and missing class_memberships data for databases that were partially migrated.
Usage: Usage:
python repair_class_memberships.py python repair_class_memberships.py
@ -21,6 +21,7 @@ from app.db.models import (
Assignment, Assignment,
AssignmentSubmission, AssignmentSubmission,
ClassMembership, ClassMembership,
EmailVerificationCode,
FundRecord, FundRecord,
Resource, Resource,
Timeline, Timeline,
@ -31,7 +32,13 @@ from app.db.models import (
async def ensure_class_memberships_table() -> None: async def ensure_class_memberships_table() -> None:
async with engine.begin() as conn: async with engine.begin() as conn:
await conn.run_sync(Base.metadata.create_all, tables=[ClassMembership.__table__]) await conn.run_sync(
Base.metadata.create_all,
tables=[
ClassMembership.__table__,
EmailVerificationCode.__table__,
],
)
async def repair_memberships() -> None: async def repair_memberships() -> None: