From 8bce234542f135d913f6a5b0142e55b35663f480 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 20 Jun 2023 22:49:14 +1000 Subject: [PATCH] feat(db): update image-board relationships on add Functionally, `add_image_to_board()` now moves images between boards. --- invokeai/app/services/board_image_record_storage.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/invokeai/app/services/board_image_record_storage.py b/invokeai/app/services/board_image_record_storage.py index 2f1603be82..73fad13406 100644 --- a/invokeai/app/services/board_image_record_storage.py +++ b/invokeai/app/services/board_image_record_storage.py @@ -139,9 +139,10 @@ class SqliteBoardImageRecordStorage(BoardImageRecordStorageBase): self._cursor.execute( """--sql INSERT INTO board_images (board_id, image_name) - VALUES (?, ?); + VALUES (?, ?) + ON CONFLICT (image_name) DO UPDATE SET board_id = ?; """, - (board_id, image_name), + (board_id, image_name, board_id), ) self._conn.commit() except sqlite3.Error as e: