fix up db fetch and insert
This commit is contained in:
1 parent
4dfe50db93
commit
9804b6d673
1 file changed
+12
-12
@@ -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
|
||||
Reference in new issue
Block a user