Tuuli/based
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add filtering option

Browse source
This commit is contained in:
Andrew 2023-03-12 01:34:54 +07:00
parent c37c93d0d4
commit e25a5aae20
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
src/based/db.py
View file

@ -101,7 +101,6 @@ class DBConnector:
).format(Literal(table_name)) ).format(Literal(table_name))
with self.connection() as conn: with self.connection() as conn:
result = conn.execute(stmt).fetchone() result = conn.execute(stmt).fetchone()
print(result)
return False if result is None else result["exists"] return False if result is None else result["exists"]
def createTable( def createTable(
@ -149,6 +148,17 @@ class DBConnector:
with self.connection() as conn: with self.connection() as conn:
return conn.execute(stmt).fetchall() return conn.execute(stmt).fetchall()
def filterFromTable(
self, table_name: str, columns: list[str], where: list[ColumnCondition]
) -> list[dict[str, Any]]:
stmt = SQL("SELECT {} FROM {} WHERE {}").format(
SQL(", ").join(map(lambda c: Identifier(c), columns)),
Identifier(table_name),
SQL(", ").join(map(lambda cp: cp.sql(), where)),
)
with self.connection() as conn:
return conn.execute(stmt).fetchall()
def updateDataInTable( def updateDataInTable(
self, table_name: str, columns: list[ColumnUpdate], where: list[ColumnCondition] self, table_name: str, columns: list[ColumnUpdate], where: list[ColumnCondition]
): ):