fix(db): return board records from CRUD methods

This commit is contained in:
psychedelicious
2023-06-16 11:00:40 +10:00
parent 4a0a718b96
commit e4893e4031

View File

@ -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,