Files
TheGrapefruitsDuo/server/app/db/base_queries.py
2024-05-01 10:37:41 -07:00

33 lines
905 B
Python

from typing import Callable
from app.db.conn import connect_db
class BaseQueries:
from icecream import ic
def __init__(self) -> None:
self.table: str = None # type: ignore
self.connect_db: Callable = connect_db
async def select_all_series(self) -> list[dict]:
query = f"SELECT * FROM {self.table}"
db = connect_db()
cursor = db.cursor(dictionary=True)
cursor.execute(query)
data = cursor.fetchall()
cursor.close()
db.close()
return data # type: ignore
async def select_one_series_by_id(self, id: int) -> dict | None:
query = f"SELECT * FROM {self.table} WHERE id = %s"
db = self.connect_db()
cursor = db.cursor(dictionary=True)
cursor.execute(query, (id,))
data = cursor.fetchone()
cursor.close()
db.close()
return data # type: ignore