Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

add board_id association to image

Browse Source
This commit is contained in:
maryhipp 2023-06-13 11:55:38 -07:00 committed by psychedelicious
parent 207602f425
commit a121e6b3a0
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/app/services/image_record_storage.py
View File

@ -260,6 +260,18 @@ class SqliteImageRecordStorage(ImageRecordStorageBase):
""", """,
(changes.is_intermediate, image_name), (changes.is_intermediate, image_name),
) )
# Change the image's `is_intermediate`` flag
if changes.board_id is not None:
self._cursor.execute(
f"""--sql
UPDATE images
SET board_id = ?
WHERE image_name = ?;
""",
(changes.board_id, image_name),
)
self._conn.commit() self._conn.commit()
except sqlite3.Error as e: except sqlite3.Error as e:
self._conn.rollback() self._conn.rollback()

4
invokeai/app/services/models/image_record.py
View File

@ -72,6 +72,10 @@ class ImageRecordChanges(BaseModel, extra=Extra.forbid):
default=None, description="The image's new `is_intermediate` flag." default=None, description="The image's new `is_intermediate` flag."
) )
"""The image's new `is_intermediate` flag.""" """The image's new `is_intermediate` flag."""
board_id: Optional[StrictStr] = Field(
default=None, description="The image's new board ID."
)
"""The image's new board ID."""
class ImageUrlsDTO(BaseModel): class ImageUrlsDTO(BaseModel):