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

feat(nodes): make board <> images a one-to-many relationship

Browse Source 
we can extend this to many-to-many in the future if desired.
This commit is contained in:
psychedelicious 2023-06-15 14:41:36 +10:00
parent e06c43adc8
commit 4b32322a58
1 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
invokeai/app/services/board_image_record_storage.py
View File

@ -12,7 +12,7 @@ from invokeai.app.services.models.image_record import (
class BoardImageRecordStorageBase(ABC): class BoardImageRecordStorageBase(ABC):
"""Abstract base class for board-image relationship record storage.""" """Abstract base class for the one-to-many board-image relationship record storage."""
@abstractmethod @abstractmethod
def add_image_to_board( def add_image_to_board(
@ -45,7 +45,7 @@ class BoardImageRecordStorageBase(ABC):
self, self,
board_id: str, board_id: str,
) -> OffsetPaginatedResults[BoardRecord]: ) -> OffsetPaginatedResults[BoardRecord]:
"""Gets images for a board.""" """Gets boards for an image."""
pass pass
@abstractmethod @abstractmethod
@ -93,7 +93,9 @@ class SqliteBoardImageRecordStorage(BoardImageRecordStorageBase):
created_at DATETIME NOT NULL DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW')), created_at DATETIME NOT NULL DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW')),
-- updated via trigger -- updated via trigger
updated_at DATETIME NOT NULL DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW')), updated_at DATETIME NOT NULL DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW')),
PRIMARY KEY (board_id, image_name), -- enforce one-to-many relationship between boards and images using PK
-- (we can extend this to many-to-many later)
PRIMARY KEY (image_name),
FOREIGN KEY (board_id) REFERENCES boards (board_id) ON DELETE CASCADE, FOREIGN KEY (board_id) REFERENCES boards (board_id) ON DELETE CASCADE,
FOREIGN KEY (image_name) REFERENCES images (image_name) ON DELETE CASCADE FOREIGN KEY (image_name) REFERENCES images (image_name) ON DELETE CASCADE
); );
@ -184,7 +186,7 @@ class SqliteBoardImageRecordStorage(BoardImageRecordStorageBase):
SELECT COUNT(*) FROM images WHERE 1=1; SELECT COUNT(*) FROM images WHERE 1=1;
""" """
) )
count = self._cursor.fetchone()[0] count = cast(int, self._cursor.fetchone()[0])
except sqlite3.Error as e: except sqlite3.Error as e:
self._conn.rollback() self._conn.rollback()
@ -222,7 +224,7 @@ class SqliteBoardImageRecordStorage(BoardImageRecordStorageBase):
SELECT COUNT(*) FROM boards WHERE 1=1; SELECT COUNT(*) FROM boards WHERE 1=1;
""" """
) )
count = self._cursor.fetchone()[0] count = cast(int, self._cursor.fetchone()[0])
except sqlite3.Error as e: except sqlite3.Error as e:
self._conn.rollback() self._conn.rollback()
@ -243,11 +245,10 @@ class SqliteBoardImageRecordStorage(BoardImageRecordStorageBase):
""", """,
(board_id,), (board_id,),
) )
count = self._cursor.fetchone()[0] count = cast(int, self._cursor.fetchone()[0])
return count
except sqlite3.Error as e: except sqlite3.Error as e:
self._conn.rollback() self._conn.rollback()
raise e raise e
finally: finally:
self._lock.release() self._lock.release()
return count