From ba3a1f7cb5ff4b14fc777adea6d37f3b5b1d092a Mon Sep 17 00:00:00 2001
From: MarkusJoe <markushammered@gmail.com>
Date: Sun, 11 Sep 2022 10:31:21 +0800
Subject: [PATCH] =?UTF-8?q?update:=20=E5=B0=86=E6=95=B0=E6=8D=AE=E5=BA=93?=
 =?UTF-8?q?=E6=93=8D=E4=BD=9C=E7=A7=BB=E5=8A=A8=E8=87=B3db.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/db/__init__.py | 55 ++++++----------------------------------------
 1 file changed, 7 insertions(+), 48 deletions(-)

diff --git a/src/db/__init__.py b/src/db/__init__.py
index c605b06..74c7bcf 100644
--- a/src/db/__init__.py
+++ b/src/db/__init__.py
@@ -6,53 +6,12 @@
 # @File Name: __init__.py.py
 
 
-import os
-import sqlite3
-from src.utils.t_download import download
+from src.config import Config
+
+if Config.database == 'sqlite':
+    from src.db.db import SQLite as Database
+else:
+    from src.db.db import MySQL as Database
 
 
-class SQLite:
-    def __init__(self):
-        self.conn = sqlite3.connect('./src/db/data.sqlite')
-        # self.conn = sqlite3.connect('./data.sqlite')
-        self.cursor = self.conn.cursor()
-
-    def __del__(self):
-        self.conn.commit()
-        self.cursor.close()
-        self.conn.close()
-
-    def query(self, _id: str) -> tuple:
-        self.cursor.execute('select * from data where id="%s"' % _id)
-        result = self.cursor.fetchone()
-        if result is None:
-            self.insert(_id)
-            return tuple([_id, 0])
-        self.update(_id, result[1])
-        return result
-
-    def insert(self, _id: str) -> bool:
-        self.cursor.execute('insert into data (id, times) values ("%s", 1)' % _id)
-        return True
-
-    def update(self, _id: str, times: int) -> bool:
-        times += 1
-        self.cursor.execute('update data set times=%s where id="%s"' % (times, _id))
-        return True
-
-    def query_all(self) -> list:
-        self.cursor.execute('select * from data')
-        result = self.cursor.fetchall()
-        return result
-
-    def query_image(self, _id: str) -> list:
-        self.cursor.execute('select * from image where id="%s"' % _id)
-        result = self.cursor.fetchall()
-        return result
-
-
-if __name__ != '__main__':
-    if not os.path.exists('./src/db/data.sqlite'):
-        print('database file not exists, start downloading...')
-        download('https://pac.rtst.tech/static_file_hosting/static/counter/data.sqlite')
-        print('Done.')
+__all__ = [Database]