mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
fix(db): return board records from CRUD methods
This commit is contained in:
@ -1,5 +1,4 @@
|
|||||||
from abc import ABC, abstractmethod
|
from abc import ABC, abstractmethod
|
||||||
from datetime import datetime
|
|
||||||
from typing import Optional, cast
|
from typing import Optional, cast
|
||||||
import sqlite3
|
import sqlite3
|
||||||
import threading
|
import threading
|
||||||
@ -181,23 +180,12 @@ class SqliteBoardRecordStorage(BoardRecordStorageBase):
|
|||||||
(board_id, board_name),
|
(board_id, board_name),
|
||||||
)
|
)
|
||||||
self._conn.commit()
|
self._conn.commit()
|
||||||
|
|
||||||
self._cursor.execute(
|
|
||||||
"""--sql
|
|
||||||
SELECT *
|
|
||||||
FROM boards
|
|
||||||
WHERE board_id = ?;
|
|
||||||
""",
|
|
||||||
(board_id,),
|
|
||||||
)
|
|
||||||
|
|
||||||
result = self._cursor.fetchone()
|
|
||||||
return BoardRecord(**result)
|
|
||||||
except sqlite3.Error as e:
|
except sqlite3.Error as e:
|
||||||
self._conn.rollback()
|
self._conn.rollback()
|
||||||
raise BoardRecordSaveException from e
|
raise BoardRecordSaveException from e
|
||||||
finally:
|
finally:
|
||||||
self._lock.release()
|
self._lock.release()
|
||||||
|
return self.get(board_id)
|
||||||
|
|
||||||
def get(
|
def get(
|
||||||
self,
|
self,
|
||||||
@ -228,7 +216,7 @@ class SqliteBoardRecordStorage(BoardRecordStorageBase):
|
|||||||
self,
|
self,
|
||||||
board_id: str,
|
board_id: str,
|
||||||
changes: BoardChanges,
|
changes: BoardChanges,
|
||||||
) -> None:
|
) -> BoardRecord:
|
||||||
try:
|
try:
|
||||||
self._lock.acquire()
|
self._lock.acquire()
|
||||||
|
|
||||||
@ -260,6 +248,7 @@ class SqliteBoardRecordStorage(BoardRecordStorageBase):
|
|||||||
raise BoardRecordSaveException from e
|
raise BoardRecordSaveException from e
|
||||||
finally:
|
finally:
|
||||||
self._lock.release()
|
self._lock.release()
|
||||||
|
return self.get(board_id)
|
||||||
|
|
||||||
def get_many(
|
def get_many(
|
||||||
self,
|
self,
|
||||||
|
Reference in New Issue
Block a user