RequestCounter/app/api/views.py

#!/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