修改操作数据库的代码

bin/db/db.py

@@ -3,4 +3,126 @@
 # @Author: markushammered@gmail.com
 # @Development Tool: PyCharm
 # @Create Time: 2022/2/24
-# @File Name: db.py
+# @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内的属性可以直接获取对应的信息')