From 833aa14e5cb342392f2dee7da50e2bafd901b2b9 Mon Sep 17 00:00:00 2001
From: MarkusJoe <markushammered@gmail.com>
Date: Fri, 25 Feb 2022 18:29:56 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E6=93=8D=E4=BD=9C=E6=95=B0?=
 =?UTF-8?q?=E6=8D=AE=E5=BA=93=E7=9A=84=E4=BB=A3=E7=A0=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bin/db/db.py | 124 ++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 123 insertions(+), 1 deletion(-)

diff --git a/bin/db/db.py b/bin/db/db.py
index 0b5349b..bc92a26 100644
--- a/bin/db/db.py
+++ b/bin/db/db.py
@@ -3,4 +3,126 @@
 # @Author: markushammered@gmail.com
 # @Development Tool: PyCharm
 # @Create Time: 2022/2/24
-# @File Name: db.py
\ No newline at end of file
+# @File Name: db.py
+
+
+import sqlite3
+from typing import Any
+
+
+class SQLite:
+    """操作SQLite数据库"""
+
+    def __init__(self, path: str = './bin/db/data.db'):
+        """
+        初始化SQLite类
+        :param path: 数据库路径
+        """
+        self.path = path
+
+    def insert_data(self, name: str) -> None:
+        """
+        插入数据
+        :param name:
+        :return:
+        """
+        conn = sqlite3.connect(self.path, check_same_thread=False)
+        cursor = conn.cursor()
+        try:
+            cursor.execute('insert into ReqCount values(?, ?)', (name, 1))
+            conn.commit()
+        finally:
+            cursor.close()
+            conn.close()
+
+    def fetch_data(self, name: str) -> int:
+        """
+        获取数据
+        :param name:
+        :return:
+        """
+        conn = sqlite3.connect(self.path, check_same_thread=False)
+        cursor = conn.cursor()
+        try:
+            cursor.execute('select * from ReqCount')
+            conn.commit()
+            data = cursor.fetchall()
+
+            temp_dict = {}
+            for k, v in data:  # 遍历数据将元组数据转换为字典类型
+                temp_dict.setdefault(k, []).append(v)
+            for i, c in zip(temp_dict.keys(), temp_dict.values()):
+                temp_dict[i] = c[0]
+
+            if name in temp_dict.keys():
+                count = temp_dict[name]  # 获取原数字 为 整型
+                self.update_data(name, count)
+                return count
+            else:
+                # 新建用户数据
+                self.insert_data(name)
+                return 0
+        finally:
+            cursor.close()
+            conn.close()
+
+    def update_data(self, name: str, times: int) -> None:
+        """
+        更新数据
+        :param name:
+        :param times:
+        :return:
+        """
+        conn = sqlite3.connect(self.path, check_same_thread=False)
+        cursor = conn.cursor()
+        try:
+            times += 1
+            cursor.execute('update ReqCount set times=? where name=?', (times, name))
+            conn.commit()
+        finally:
+            cursor.close()
+            conn.close()
+
+    def fetch_table(self) -> list:
+        """
+        返回已有主题列表
+        :return:
+        """
+        conn_temp = sqlite3.connect(self.path, check_same_thread=False)
+        cursor_temp = conn_temp.cursor()
+        try:
+            lst = []
+            cursor_temp.execute("select * from sqlite_master where type='table'")
+            for i in cursor_temp.fetchall():
+                lst.append(i[1])
+            return lst
+        finally:
+            cursor_temp.close()
+            conn_temp.close()
+
+    def fetch_style_data(self, style: str) -> list[dict[str, Any]]:
+        """
+        获取主题数据库内的数据
+        :param style:
+        :return:
+        """
+        conn = sqlite3.connect(self.path, check_same_thread=False)
+        cursor = conn.cursor()
+        try:
+            cursor.execute('select * from %(style_name)s' % {'style_name': style})
+            data = cursor.fetchall()
+            data_set = []
+            for i in data:
+                data_set.append({'index': i[0], 'base64': i[1], 'width': i[2], 'height': i[3]})
+            return data_set
+        finally:
+            cursor.close()
+            conn.close()
+
+
+if __name__ == '__main__':
+    print('MySQL数据库并未开发完成暂时只能使用SQLite数据库\n'
+          '如果你想帮助开发MySQL的支持你可以在./bin/tests内找到MySQL_.py并继续开发\n'
+          '开发进度: 已经完成 删, 改, 增, 查, 以及查询已有的表\n'
+          '获取配置文件直接使用from bin.utils.settings import Settings\n'
+          '可以直接实例化并使用此类, 调用Settings内的属性可以直接获取对应的信息')