1

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: