重写&重构项目, 优化项目结构

app/main/__init__.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/3/25
+# @File Name: __init__.py.py
+
+
+from flask import Blueprint
+
+main = Blueprint('main', __name__)
+
+from . import views, errors

app/main/errors.py

@@ -0,0 +1,29 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/3/25
+# @File Name: errors.py
+
+from . import main
+
+
+@main.app_errorhandler(404)
+def page_not_found(e):
+    """
+    404页面未找到
+    :param e:
+    :return:
+    """
+    return {'code': 404, 'msg': '页面未找到', 'data': []}, 404
+
+
+@main.app_errorhandler(500)
+def internal_server_error(e):
+    """
+    500服务器内部错误
+    :param e:
+    :return:
+    """
+    return {'code': 500, 'msg': '服务器内部错误', 'data': []}, 500
+

app/main/views.py

@@ -0,0 +1,62 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/3/25
+# @File Name: views.py
+
+
+from flask import abort
+from flask import jsonify
+from flask import request
+from flask import Response
+from flask import make_response
+from flask import render_template
+from . import main
+from ..db.db import SQLite as db
+from ..utils.view import view
+
+
+@main.route('/', methods=['GET', 'POST'])
+def index():
+    """
+    主页
+    :return:
+    """
+    return render_template('index.html', remote_address=request.remote_addr)
+
+
+@main.route('/api/v1/<string:name>', methods=['GET', 'POST'])
+@main.route('/count/<string:name>', methods=['GET', 'POST'])
+def api_v1(name: str):
+    length = request.args.get('length', type=int)
+    theme = request.args.get('theme', type=str)
+    if not bool(length):
+        length = 7
+    if not bool(theme):
+        theme = 'lewd'
+    if not db().exists_name(name):
+        db().insert(name)
+    if not db().exists_table(theme):
+        abort(500)
+    if 7 <= length <= 10:
+        data = db().fetch(name)
+        number = '0' * (length - len(str(data[-1]))) + str(data[-1])
+        # 将设置的长度减去数据库内已有的数据库字符串长度剩下的结果为[0]的个数
+        # 再将这个字符串后拼接上数据库内已有次数的字符串, 就可以得到最终的结果
+        response = make_response(view(theme, int(length), name, number))
+        response.headers['Content-Type'] = 'image/svg+xml; charset=utf-8'
+        response.headers['cache-control'] = 'max-age=0, no-cache, no-store, must-revalidate'
+        return response
+    return abort(500)
+
+
+@main.route('/exists-table', methods=['GET', 'POST'])
+@main.route('/exists', methods=['GET', 'POST'])
+def show_tables() -> Response:
+    """
+    返回已有表
+    :return:
+    """
+    tables = db().show_tables
+    return jsonify({'code': 200, 'msg': '已有表', 'data': [tables]})