From 105bde0d3c8863b3a60f6b3e355ef4a325b1cfa3 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Mon, 27 Apr 2026 22:43:19 +0800
Subject: [PATCH] 1

---
 backend/repair_class_memberships.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

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: