fix up db fetch and insert

This commit is contained in:
C committed 2025-10-11 13:56:13 -07:00
1 parent 04cfd42b4b
commit 67891e5222
1 file changed
+12 -12
+12 -12
View File
@@ -3,7 +3,11 @@ from time import time
time = int(time()) # we dont need no stinkin floats
# db = await aiosqlite.connect("edderkop.db") # type: ignore
async def fetch(sqobj):
async def close(db):
await db.close()
async def fetch(table, sqobj):
"""
Asynchronously fetch a record from the REVIEW table by sq_object.
@@ -13,21 +17,17 @@ async def fetch(sqobj):
dict or None: The fetched record as a dictionary, or None if not found.
"""
async with aiosqlite.connect("edderkop.db") as db: # type: ignore
db.row_factory = aiosqlite.Row # type: ignore
async with db.execute(
"SELECT * FROM REVIEW WHERE sq_object = ?", (sqobj,)
) as cursor:
row = await cursor.fetchone()
if row:
return dict(row)
return None
async with db.execute(f"SELECT * FROM {table} WHERE sq_object = {sqobj}") as cursor:
async for row in cursor:
return row
async def upsert(table: str, data: dict, unique_key: str) -> None:
return None
async def upsert(table: str, data: dict, unique_key: str) -> bool:
"""
Asynchronously insert or update a record in the specified table.
Args:
sq_ib (aiosqlite.Connection): The SQLite database connection.
table (str): The name of the table to upsert into.
data (dict): A dictionary of column-value pairs to insert or update.
unique_key (str): The column name that serves as the unique key for upsert.
@@ -50,4 +50,4 @@ async def upsert(table: str, data: dict, unique_key: str) -> None:
async with aiosqlite.connect("edderkop.db") as db: # type: ignore
async with db.execute(sql, tuple(data.values())) as cursor:
await db.commit()
#await db.close()
return True