mirror of
https://github.com/NovaOSS/nova-api.git
synced 2024-11-25 22:43:57 +01:00
37 lines
833 B
Python
37 lines
833 B
Python
|
"""Database helper."""
|
||
|
|
|
||
|
|
import os
|
||
|
|
import aiosqlite
|
||
|
|
|
||
|
|
from dotenv import load_dotenv
|
||
|
|
|
||
|
|
load_dotenv()
|
||
|
|
|
||
|
|
async def row_to_dict(row, cursor):
|
||
|
|
"""Converts a database row to into a <dict>."""
|
||
|
|
|
||
|
|
return {col[0]: row[idx] for idx, col in enumerate(cursor.description)}
|
||
|
|
|
||
|
|
async def connect(name: str):
|
||
|
|
"""Creates a connection to the database"""
|
||
|
|
|
||
|
|
return await aiosqlite.connect(f'{os.getenv("API_DB_PATH")}{name}.db')
|
||
|
|
|
||
|
|
async def insert_dict(dict_: dict, table: str, db) -> dict:
|
||
|
|
"""Adds a dictionary to a table, safely."""
|
||
|
|
|
||
|
|
if isinstance(db, str):
|
||
|
|
db = await connect(db)
|
||
|
|
|
||
|
|
sep = ', '
|
||
|
|
query = f"""
|
||
|
|
INSERT INTO {table} ({sep.join(dict_.keys())})
|
||
|
|
VALUES ({sep.join([f':{key}' for key in dict_.keys()])})
|
||
|
|
"""
|
||
|
|
|
||
|
|
await db.execute(query, dict_)
|
||
|
|
await db.commit()
|
||
|
|
|
||
|
|
return dict_
|
||
|
|
|