删除历史版本, 初始化仓库

app/api/__init__.py

@@ -0,0 +1,19 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/4/9
+# @File Name: __init__.py.py
+
+
+from flask import Blueprint
+
+api = Blueprint('api', __name__)
+
+from . import views, errors
+
+__all__ = [
+    'api',
+    'views',
+    'errors'
+]

app/api/auth.py

@@ -0,0 +1,26 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: Pycharm
+# @Create Time: 2022/5/2
+# @File Name: auth.py
+
+
+import os
+from flask import request
+from flask_httpauth import HTTPTokenAuth
+
+auth = HTTPTokenAuth()
+
+
+@auth.verify_token
+def verify_token(token):
+    secret_key = os.getenv('ACCESS_KEY')
+    key = request.args.get('key')
+    if token or key == secret_key:
+        return True
+    elif (token == 'unittest' or
+          key == 'unittest') and \
+            'system' in request.path.split('/'):  # 限定在system这个接口使用unittest密钥获取数据
+        return True
+    return False

app/api/errors.py

@@ -0,0 +1,44 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/4/9
+# @File Name: errors.py
+
+from . import api
+from .auth import auth
+from flask import jsonify
+
+
+@api.errorhandler(403)
+@auth.error_handler
+def forbidden(e):
+    return jsonify(
+        {
+            'code': 403,
+            'msg': 'API: Permission Denied.',
+            'data': None
+        }
+    ), 403
+
+
+@api.errorhandler(404)
+def page_not_found(e):
+    return jsonify(
+        {
+            'code': 404,
+            'msg': 'API: Page Not Found.',
+            'data': None
+        }
+    ), 404
+
+
+@api.errorhandler(500)
+def internal_server_error(e):
+    return jsonify(
+        {
+            'code': 500,
+            'msg': 'API: Internal Server Error.',
+            'data': None
+        }
+    ), 500

app/api/views.py

@@ -0,0 +1,172 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/4/9
+# @File Name: views.py
+
+
+import sys
+import psutil
+import platform
+from flask import jsonify
+from flask import request
+from flask import send_file
+from ..db.db import SQLite as db
+from . import api
+from .auth import auth
+
+
+@api.route('/test/', methods=['GET', 'POST'])
+def test_api():
+    """
+    测试专用接口
+    :return:
+    """
+    data = db().fetch('test-example')
+    return str(data[1])
+
+
+@api.route('/overall/', methods=['GET', 'POST'])
+def overall():
+    """
+    查询ReqCount表内所有的源数据
+    :return:
+    """
+    limit = request.args.get('limit', type=int)
+    try:
+        data = db().exec('select * from reqcount;')
+        format_data = []
+        for i in data:
+            format_data.append({'name': i[0], 'times': i[1]})
+        return jsonify(
+            {
+                'code': 200,
+                'msg': 'Success. Total: %s row(s)' % len(data),
+                'data': format_data[:limit]
+            }
+        ), 200
+    except Exception as e:
+        return jsonify(
+            {
+                'code': -200,
+                'msg': 'Error',
+                'data': e
+            }
+        ), 500
+
+
+@api.route('/query/', methods=['GET', 'POST'])
+def query():
+    """
+    查询指定名称的计数数据
+    :return:
+    """
+    name = request.args.get('name', type=str)
+    nochange = request.args.get('nochange', type=bool)
+    if nochange:
+        nochange = True
+    if not db().exists_name(name):
+        return jsonify(
+            {
+                'code': -200,
+                'msg': 'Failed. No such name',
+                'data': None
+            }
+        ), 404
+    else:
+        data = db().fetch(name, nochange)
+        return jsonify(
+            {
+                'code': 200,
+                'msg': 'Success',
+                'data': data
+            }
+        ), 200
+
+
+@api.route('/theme/', methods=['GET', 'POST'])
+def theme():
+    """
+    查询数据库内主题的源数据
+    :return:
+    """
+    _theme = request.args.get('name', type=str)
+    if not db().exists_table(_theme):
+        return jsonify(
+            {
+                'code': 404,
+                'msg': 'Failed. No such theme',
+                'data': None
+            }
+        ), 404
+    else:
+        data = db().fetching_table(_theme)
+        return jsonify(
+            {
+                'code': 200,
+                'msg': 'Success',
+                'data': data
+            }
+        ), 200
+
+
+@api.route('/alltables/', methods=['GET', 'POST'])
+def all_tables():
+    """
+    查询所有已有表
+    :return:
+    """
+    data = db().show_tables
+    return jsonify(
+        {
+            'code': 200,
+            'msg': 'Success',
+            'data': data
+        }
+    ), 200
+
+
+@api.route('/export/', methods=['GET', 'POST'])
+@auth.login_required
+def export():
+    """
+    导出数据库文件
+    :return:
+    """
+    return send_file('./db/data.sqlite', as_attachment=True), 200
+
+
+@api.route('/system/', methods=['GET', 'POST'])
+@auth.login_required
+def system_info():
+    """
+    获取当前部署服务器的系统状态信息
+    :return:
+    """
+    _platform = platform.system()
+    _cpu_count = psutil.cpu_count()
+    _memory_size = psutil.virtual_memory()
+    _total_memory_size = round(_memory_size[0] / 1024 / 1024 / 1024, 2)
+    _free_memory_size = round(_memory_size[4] / 1024 / 1024 / 1024, 2)
+    _used_memory_size = round(_memory_size[3] / 1024 / 1024 / 1024, 2)
+    _percent = _memory_size[2]
+    _py_version = sys.version
+
+    return jsonify(
+        {
+            'code': 200,
+            'msg': 'Success',
+            'data': {
+                'Platform': _platform,
+                'CPU-Count': _cpu_count,
+                'Memory-Size(GigaBytes)': {
+                    'Total': _total_memory_size,
+                    'used': _used_memory_size,
+                    'Free': _free_memory_size,
+                    'Percent': _percent
+                },
+                'Python': _py_version
+            }
+        }
+    ), 200