From a7101b7b0a1128a76e98909fb33ca53cbb9c6858 Mon Sep 17 00:00:00 2001
From: EstrellaXD <estrellaxd05@gmail.com>
Date: Tue, 16 May 2023 22:36:22 +0800
Subject: [PATCH] Universal Database orm

---
 src/module/api/bangumi.py        |  2 +-
 src/module/database/bangumi.py   | 28 ++++++++--------------------
 src/module/database/connector.py |  4 ++++
 3 files changed, 13 insertions(+), 21 deletions(-)

diff --git a/src/module/api/bangumi.py b/src/module/api/bangumi.py
index 13b7050e..5540078d 100644
--- a/src/module/api/bangumi.py
+++ b/src/module/api/bangumi.py
@@ -68,4 +68,4 @@ async def delete_rule(bangumi_id: str, file: bool = False):
 async def reset_all():
     with BangumiDatabase() as database:
         database.delete_all()
-        return {"status": "ok"}
\ No newline at end of file
+        return {"status": "ok"}
diff --git a/src/module/database/bangumi.py b/src/module/database/bangumi.py
index 6d48d5a7..34d174e7 100644
--- a/src/module/database/bangumi.py
+++ b/src/module/database/bangumi.py
@@ -9,11 +9,11 @@ logger = logging.getLogger(__name__)
 class BangumiDatabase(DataConnector):
     def __init__(self):
         super().__init__()
+        self.__table_name = "bangumi"
 
     def update_table(self):
-        table_name = "bangumi"
         db_data = self.__data_to_db(BangumiData())
-        self._update_table(table_name, db_data)
+        self._update_table(self.__table_name, db_data)
 
     @staticmethod
     def __data_to_db(data: BangumiData) -> dict:
@@ -45,19 +45,13 @@ class BangumiDatabase(DataConnector):
 
     def insert(self, data: BangumiData):
         db_data = self.__data_to_db(data)
-        columns = ", ".join(db_data.keys())
-        values = ", ".join([f":{key}" for key in db_data.keys()])
-        self._cursor.execute(f"INSERT INTO bangumi ({columns}) VALUES ({values})", db_data)
-        logger.debug(f"Add {data.official_title} into database.")
-        self._conn.commit()
+        self._insert(db_data=db_data, table_name=self.__table_name)
+        logger.debug(f"Insert {data.official_title} into database.")
 
     def insert_list(self, data: list[BangumiData]):
-        db_data = [self.__data_to_db(x) for x in data]
-        columns = ", ".join(db_data[0].keys())
-        values = ", ".join([f":{key}" for key in db_data[0].keys()])
-        self._cursor.executemany(f"INSERT INTO bangumi ({columns}) VALUES ({values})", db_data)
-        logger.debug(f"Add {len(data)} bangumi into database.")
-        self._conn.commit()
+        data_list = [self.__data_to_db(x) for x in data]
+        self._insert_list(data_list=data_list, table_name=self.__table_name)
+        logger.debug(f"Insert {len(data)} bangumi into database.")
 
     def update_one(self, data: BangumiData) -> bool:
         db_data = self.__data_to_db(data)
@@ -97,13 +91,7 @@ class BangumiDatabase(DataConnector):
         return self._cursor.rowcount == 1
 
     def delete_all(self):
-        self._cursor.execute(
-            """
-            DELETE FROM bangumi
-            """
-        )
-        self._conn.commit()
-        logger.debug("Delete all bangumi.")
+        self._delete_all(self.__table_name)
 
     def search_all(self) -> list[BangumiData]:
         self._cursor.execute(
diff --git a/src/module/database/connector.py b/src/module/database/connector.py
index 3807e32a..e68d9d6e 100644
--- a/src/module/database/connector.py
+++ b/src/module/database/connector.py
@@ -40,6 +40,10 @@ class DataConnector:
         self._cursor.executemany(f"INSERT INTO {table_name} ({columns}) VALUES ({values})", data_list)
         self._conn.commit()
 
+    def _delete_all(self, table_name: str):
+        self._cursor.execute(f"DELETE FROM {table_name}")
+        self._conn.commit()
+
     @staticmethod
     def __python_to_sqlite_type(value) -> str:
         if isinstance(value, int):