update bare except

Wrapped image migration with a general exception
2024-08-30 20:32:17 +00:00 · 2023-12-13 12:06:28 -05:00 · 2023-12-13 11:18:21 -05:00
1 changed files with 7 additions and 5 deletions
--- a/invokeai/app/services/shared/sqlite_migrator/migrations/migration_2.py
+++ b/invokeai/app/services/shared/sqlite_migrator/migrations/migration_2.py
@ -1,8 +1,6 @@
 import sqlite3
 from logging import Logger
-
 from tqdm import tqdm
-
 from invokeai.app.services.image_files.image_files_base import ImageFileStorageBase
 from invokeai.app.services.shared.sqlite_migrator.sqlite_migrator_common import Migration

@ -149,9 +147,13 @@ class Migration2Callback:
        pbar = tqdm(image_names)
        for idx, image_name in enumerate(pbar):
            pbar.set_description(f"Checking image {idx + 1}/{total_image_names} for workflow")
+            try:
                pil_image = self._image_files.get(image_name)
                if "invokeai_workflow" in pil_image.info:
                    to_migrate.append((True, image_name))
+            except KeyError:
+                self._logger.warning(f"Image '{image_name}' not found. Skipping.")
+            continue

        self._logger.info(f"Adding {len(to_migrate)} embedded workflows to database")
        cursor.executemany("UPDATE images SET has_workflow = ? WHERE image_name = ?", to_migrate)
Author	SHA1	Message	Date
Kent Keirsey	7d65a18667	update bare except	2023-12-13 12:06:28 -05:00
Kent Keirsey	7bada22ded	Wrapped image migration with a general exception	2023-12-13 11:18:21 -05:00