diff --git a/backend/repair_class_memberships.py b/backend/repair_class_memberships.py index 8e0322b..f51225e 100644 --- a/backend/repair_class_memberships.py +++ b/backend/repair_class_memberships.py @@ -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: python repair_class_memberships.py @@ -21,6 +21,7 @@ from app.db.models import ( Assignment, AssignmentSubmission, ClassMembership, + EmailVerificationCode, FundRecord, Resource, Timeline, @@ -31,7 +32,13 @@ from app.db.models import ( async def ensure_class_memberships_table() -> None: 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: