Upload 67 files

app/services/bootstrap.py

@@ -38,11 +38,11 @@ def submission_indexes() -> tuple[set[str], set[str]]:
     return unique_names, index_names
 
 
-def task_columns() -> set[str]:
+def task_column_details() -> dict[str, dict]:
     inspector = inspect(engine)
     if "tasks" not in inspector.get_table_names():
-        return set()
-    return {column["name"] for column in inspector.get_columns("tasks")}
+        return {}
+    return {column["name"]: column for column in inspector.get_columns("tasks")}
 
 
 def ensure_submission_constraints() -> None:
@@ -74,12 +74,16 @@ def ensure_submission_constraints() -> None:
 
 
 def migrate_task_media_to_files() -> None:
-    columns = task_columns()
-    if not columns:
+    column_details = task_column_details()
+    if not column_details:
         return
 
-    has_image_data = "image_data" in columns
-    has_clue_image_data = "clue_image_data" in columns
+    has_image_data = "image_data" in column_details
+    has_clue_image_data = "clue_image_data" in column_details
+    can_clear_image_data = bool(has_image_data and column_details["image_data"].get("nullable", True))
+    can_clear_clue_image_data = bool(
+        has_clue_image_data and column_details["clue_image_data"].get("nullable", True)
+    )
     if not has_image_data and not has_clue_image_data:
         return
 
@@ -116,7 +120,7 @@ def migrate_task_media_to_files() -> None:
                 )
                 updates.append("image_path = :image_path")
                 params["image_path"] = image_path
-                if has_image_data:
+                if can_clear_image_data:
                     updates.append("image_data = NULL")
 
             current_clue_path = row.get("clue_image_path")
@@ -131,7 +135,7 @@ def migrate_task_media_to_files() -> None:
                 )
                 updates.append("clue_image_path = :clue_image_path")
                 params["clue_image_path"] = clue_image_path
-                if has_clue_image_data:
+                if can_clear_clue_image_data:
                     updates.append("clue_image_data = NULL")
 
             if updates:
@@ -223,3 +227,7 @@ def seed_super_admin(db: Session) -> None:
         db.add(admin)
         db.commit()
 
+
+
+
+