ruff fix

Author: terapyon

scripts/fix_event_names.py

@@ -41,9 +41,9 @@ def _fetchone_scalar(conn: duckdb.DuckDBPyConnection, sql: str) -> int:
 
 def apply_fixes(db_path: str, *, dry_run: bool) -> None:
     label = "DRY RUN" if dry_run else "APPLY"
-    print(f"\n{'='*60}")
+    print(f"\n{'=' * 60}")
     print(f"[{label}] Database: {db_path}")
-    print(f"{'='*60}")
+    print(f"{'=' * 60}")
 
     conn = duckdb.connect(str(db_path), read_only=dry_run)
 
@@ -53,46 +53,36 @@ def apply_fixes(db_path: str, *, dry_run: bool) -> None:
     if dry_run:
         for old_name, new_name, new_year in FIXES:
             if new_year is not None:
-                sql = ("UPDATE images SET event_name = ?, "
-                       "event_year = ? WHERE event_name = ?")
+                sql = "UPDATE images SET event_name = ?, event_year = ? WHERE event_name = ?"
                 params = [new_name, new_year, old_name]
             else:
-                sql = ("UPDATE images SET event_name = ? "
-                       "WHERE event_name = ?")
+                sql = "UPDATE images SET event_name = ? WHERE event_name = ?"
                 params = [new_name, old_name]
             print(f"  [DRY RUN] {sql}  params={params}")
     else:
         # Temporarily drop FK constraint by backing up
         # and recreating image_embeddings
-        has_embeddings = _fetchone_scalar(
-            conn, "SELECT COUNT(*) FROM image_embeddings"
-        ) > 0
+        has_embeddings = _fetchone_scalar(conn, "SELECT COUNT(*) FROM image_embeddings") > 0
 
         print("  Backing up image_embeddings...")
-        conn.execute(
-            "CREATE TEMPORARY TABLE _emb_backup "
-            "AS SELECT * FROM image_embeddings"
-        )
+        conn.execute("CREATE TEMPORARY TABLE _emb_backup AS SELECT * FROM image_embeddings")
         conn.execute("DROP TABLE image_embeddings")
 
         # Now we can UPDATE images freely
         for old_name, new_name, new_year in FIXES:
             if new_year is not None:
                 row = conn.execute(
-                    "UPDATE images SET event_name = ?, "
-                    "event_year = ? WHERE event_name = ?",
+                    "UPDATE images SET event_name = ?, event_year = ? WHERE event_name = ?",
                     [new_name, new_year, old_name],
                 ).fetchone()
             else:
                 row = conn.execute(
-                    "UPDATE images SET event_name = ? "
-                    "WHERE event_name = ?",
+                    "UPDATE images SET event_name = ? WHERE event_name = ?",
                     [new_name, old_name],
                 ).fetchone()
             cnt = int(row[0]) if row else 0
             suffix = f" (year={new_year})" if new_year is not None else ""
-            print(f"  Updated {cnt} rows: '{old_name}' -> '{new_name}'"
-                  + suffix)
+            print(f"  Updated {cnt} rows: '{old_name}' -> '{new_name}'" + suffix)
 
         # Restore image_embeddings with FK
         print("  Restoring image_embeddings...")
@@ -106,21 +96,14 @@ def apply_fixes(db_path: str, *, dry_run: bool) -> None:
                 FOREIGN KEY (image_id) REFERENCES images(id)
             )
         """)
+        conn.execute("INSERT INTO image_embeddings SELECT * FROM _emb_backup")
         conn.execute(
-            "INSERT INTO image_embeddings "
-            "SELECT * FROM _emb_backup"
-        )
-        conn.execute(
-            "CREATE INDEX IF NOT EXISTS idx_embeddings_model "
-            "ON image_embeddings(model_name)"
+            "CREATE INDEX IF NOT EXISTS idx_embeddings_model ON image_embeddings(model_name)"
         )
         conn.execute("DROP TABLE _emb_backup")
 
-        restored = _fetchone_scalar(
-            conn, "SELECT COUNT(*) FROM image_embeddings"
-        )
-        print(f"  Restored {restored} embeddings"
-              f" (had_data={has_embeddings})")
+        restored = _fetchone_scalar(conn, "SELECT COUNT(*) FROM image_embeddings")
+        print(f"  Restored {restored} embeddings (had_data={has_embeddings})")
 
         print("\n--- After ---")
         show_events(conn)