nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-04-24 02:20:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add database test

Browse Source
This commit is contained in:
EstrellaXD
2023-06-15 14:18:57 +08:00
parent 22b3d4caf7
commit e4d34f0e42
3 changed files with 66 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
backend/src/module/database/orm/select.py
View File

@@ -31,8 +31,9 @@ class Select:
combine_operator: str = "AND",
):
if keys is None:
keys = ["*"]
columns = ", ".join(keys)
columns = "*"
else:
columns = ", ".join(keys)
condition_sql = self.__select_condition(conditions, combine_operator)
self._connector.execute(
f"""
@@ -81,9 +82,13 @@ class Select:
if not conditions:
raise ValueError("No conditions provided.")
if combine_operator not in ["AND", "OR", "INSTR"]:
raise ValueError("Invalid combine_operator, must be 'AND' or 'OR'.")
raise ValueError(
"Invalid combine_operator, must be 'AND' or 'OR' or 'INSTR'."
)
if combine_operator == "INSTR":
condition_sql = f" {combine_operator} {' AND '.join([f'({key} = :{key})' for key in conditions.keys()])}"
condition_sql = f" AND ".join(
[f"INSTR({key}, :{key})" for key in conditions.keys()]
)
else:
condition_sql = f" {combine_operator} ".join(
[f"{key} = :{key}" for key in conditions.keys()]