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

Selection filter

Browse source
This commit is contained in:
Andrew 2023-04-24 19:53:12 +07:00
parent 0e03594e54
commit b784422e69
3 changed files with 7 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

2
setup.cfg
View file

@ -1,6 +1,6 @@
[metadata] [metadata]
name = based name = based
version = 0.2.0 version = 0.2.1
description = A simple, fast (maybe), and reliable (I hope) PostgreSQL ORM oriented for dynamic models. description = A simple, fast (maybe), and reliable (I hope) PostgreSQL ORM oriented for dynamic models.
author = nuark author = nuark
author_email = me@nuark.xyz author_email = me@nuark.xyz

2
src/based/__init__.py
View file

@ -1 +1 @@
__version__ = "0.2.0" __version__ = "0.2.1"

6
src/based/db.py
View file

@ -212,7 +212,7 @@ class DBConnector:
conn.execute(stmt) conn.execute(stmt)
def selectFromTable( def selectFromTable(
self, table_name: str, columns: list[str] self, table_name: str, columns: list[str], where: list[ColumnCondition] = []
) -> list[dict[str, Any]]: ) -> list[dict[str, Any]]:
if len(columns) == 1 and columns[0] == "*": if len(columns) == 1 and columns[0] == "*":
stmt = SQL("SELECT * FROM {}").format(Identifier(table_name)) stmt = SQL("SELECT * FROM {}").format(Identifier(table_name))
@ -221,6 +221,10 @@ class DBConnector:
SQL(", ").join(map(lambda c: Identifier(c), columns)), SQL(", ").join(map(lambda c: Identifier(c), columns)),
Identifier(table_name), Identifier(table_name),
) )
if len(where) > 0:
stmt += SQL(" WHERE {}").format(
SQL(" AND ").join(map(lambda cp: cp.sql(), where))
)
with self.connection as conn: with self.connection as conn:
return conn.execute(stmt).fetchall() return conn.execute(stmt).fetchall()