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

bin/utils/error.py

@@ -9,74 +9,54 @@
 from typing import Optional
 from flask import jsonify
 from flask import Response
-from bin.db.db import SQLite
 from bin.utils.logger import logger
 
 
 class ErrorProcess:
     """处理错误的页面"""
-    def __init__(self) -> None:
-        self.msg_template = {'code': -2,
-                             'msg': Optional[str],
-                             'data': Optional[list]}
 
-    def get_theme_list(self) -> Response:
+    def __init__(self, db) -> None:
         """
-        直接获取可选主题
-        :return:
+        初始化
+        :param db:
         """
-        logger.debug('获取主题')
-        table_list = SQLite().fetch_table()
-        self.msg_template['code'] = 200
-        self.msg_template['msg'] = '当前已保存到数据库的主题如下'
-        self.msg_template['data'] = table_list
-        return jsonify(self.msg_template)
+        self.db = db  # 数据库对象
+        self.response = {'code': Optional[int],
+                         'msg': Optional[str],
+                         'data': Optional[list]}
 
-    def error_theme(self, theme: str) -> Response:
+    def theme_error(self, theme: str) -> Response:
         """
         错误的主题
         :param theme:
         :return:
         """
         logger.debug('错误的主题')
-        table_list = SQLite().fetch_table()
-        self.msg_template['code'] = -2
-        self.msg_template['msg'] = f'错误的主题: {theme}. 以下是已保存的主题'
-        self.msg_template['data'] = table_list
-        return jsonify(self.msg_template)
+        table_list = self.db().show_tables
+        self.response['code'] = -2
+        self.response['msg'] = f'错误的主题: {theme}. 以下是已保存的主题'
+        self.response['data'] = table_list
+        return jsonify(self.response)
 
-    def error_length(self, length: int or str) -> Response:
+    def length_error(self, length: int or str) -> Response:
         """
         数值太长显示此页面
         :param length:
         :return:
         """
-        logger.debug('错误的长度')
-        self.msg_template['code'] = -2
-        self.msg_template['msg'] = f'错误的长度: {length}'
-        self.msg_template['data'] = None
-        return jsonify(self.msg_template)
+        self.response['code'] = -2
+        self.response['msg'] = f'错误的长度: {length}'
+        self.response['data'] = []
+        return jsonify(self.response)
 
-    def too_lang_to_count(self, name: str) -> Response:
+    def count_error(self, name: str) -> Response:
         """
         数据过长重置数据
         :param name:
         :return:
         """
-        logger.debug('超出计数器最大计数限制')
-        SQLite().update_data(name, 0)
-        self.msg_template['code'] = 200
-        self.msg_template['msg'] = '当前长度已超过最大可计数范围. 已将此名称的计数器重置为零'
-        self.msg_template['data'] = None
-        return jsonify(self.msg_template)
-
-    def arg_not_be_full(self) -> Response:
-        """
-        参数不完整
-        :return:
-        """
-        logger.debug('参数不完整')
-        self.msg_template['code'] = -2
-        self.msg_template['msg'] = '参数填写错误或填写不完整'
-        self.msg_template['data'] = None
-        return jsonify(self.msg_template)
+        self.db().update(name, 0)
+        self.response['code'] = 200
+        self.response['msg'] = '当前长度已超过最大可计数范围. 已将此名称的计数器重置为零'
+        self.response['data'] = []
+        return jsonify(self.response)