重写了app内的主函数, 并支持了MySQL数据库, 将所有关于数据库操作的代码全部重写

bin/tests/SQLite_.py

@@ -0,0 +1,141 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/2/27
+# @File Name: SQLite_.py
+
+
+import sqlite3
+
+
+class SQLite:
+    """操作SQLite数据库"""
+
+    def __init__(self,
+                 path: str = './bin/db/data.db'
+                 ) -> None:
+        """
+        初始化SQLite对象
+        :param path: 数据库路径
+        """
+        self.__path = path
+        self.__conn = sqlite3.connect(self.__path)
+        self.__cursor = self.__conn.cursor()
+
+    def __del__(self) -> None:
+        """
+        自动提交
+        自动关闭连接
+        :return:
+        """
+        self.__conn.commit()
+        self.__cursor.close()
+        self.__conn.close()
+
+    @property
+    def show_tables(self) -> list:
+        """
+        列出所有在数据库内的表
+        :return:
+        """
+        self.__cursor.execute('select * from sqlite_master where type="table"')
+        tables = self.__cursor.fetchall()
+        lst = []
+        for table in tables:
+            lst.append(table[1])
+        del tables
+        return lst
+
+    def exist_name(self, name: str) -> bool:
+        """
+        判断当前名称是否在数据库内
+        调用fetch方法判断
+        :param name:
+        :return:
+        """
+        status = self.fetch(name)
+        if status == ():
+            return False
+        else:
+            return True
+
+    def exist_table(self, table: str) -> bool:
+        """
+        判断表是否在数据库内
+        :param table:
+        :return:
+        """
+        tables = self.show_tables
+        if table in tables:
+            return True
+        else:
+            return False
+
+    def insert(self, name: str, times: int) -> bool:
+        """
+        插入数据
+        :param name:
+        :param times:
+        :return:
+        """
+        self.__cursor.execute(
+            'insert into reqcount values(name="%(name)s", times=%(times)s)' % {'name': name, 'times': times})
+        return True
+
+    def delete(self, name: str) -> bool:
+        """
+        删除数据
+        :param name:
+        :return:
+        """
+        self.__cursor.execute('delete from reqcount where name="%(name)s"' % {'name': name})
+        return True
+
+    def fetch(self, name: str) -> tuple:
+        """
+        抓取数据
+        :param name:
+        :return:
+        """
+        self.__cursor.execute('select * from reqcount where name="%(name)s"' % {'name': name})
+        data = self.__cursor.fetchall()
+        if len(data):
+            return data[0]
+        else:
+            return tuple(data)
+
+    def update(self, name: str, times: int) -> bool:
+        """
+        更新数据
+        :param name:
+        :param times:
+        :return:
+        """
+        self.__cursor.execute(
+            'update reqcount set times=%(times)s where name="%(name)s"' % {'times': times, 'name': name})
+        return True
+
+    def fetching_table(self, table: str) -> list[dict]:
+        """
+        抓取主题表的数据
+        :param table:
+        :return:
+        """
+        status = self.exist_table(table)
+        if status:
+            tables = self.__cursor.execute('select * from %(table)s' % {'table': table})
+            lst = []
+            for data in tables:
+                lst.append({'index': data[0],
+                            'base64': data[1],
+                            'width': data[2],
+                            'height': data[3]})
+                del tables
+                return lst
+        else:
+            return []
+
+
+sqlite = SQLite('../db/data.db')
+print(sqlite.fetching_table('gelbooru'))