nas/Auto_Bangumi
1
0
Fork 0
mirror of https://github.com/EstrellaXD/Auto_Bangumi.git synced 2026-05-12 02:56:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make dataconnctor a orm class

Browse Source
This commit is contained in:
EstrellaXD
2023-05-13 21:05:04 +08:00
parent ba8ef03cdf
commit 5fb5489abd
3 changed files with 110 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/module/database/connector.py
View File

@@ -28,6 +28,18 @@ class DataConnector:
self._conn.commit()
logger.debug(f"Create / Update table {table_name}.")
def _insert(self, table_name: str, db_data: dict):
columns = ", ".join(db_data.keys())
values = ", ".join([f":{key}" for key in db_data.keys()])
self._cursor.execute(f"INSERT INTO {table_name} ({columns}) VALUES ({values})", db_data)
self._conn.commit()
def _insert_list(self, table_name: str, data_list: list[dict]):
columns = ", ".join(data_list[0].keys())
values = ", ".join([f":{key}" for key in data_list[0].keys()])
self._cursor.executemany(f"INSERT INTO {table_name} ({columns}) VALUES ({values})", data_list)
self._conn.commit()
@staticmethod
def __python_to_sqlite_type(value) -> str:
if isinstance(value, int):