webServer.py

import flask
from flask import request

from __init__ import *
from main import process


def is_version_acceptable(user_agent):
    """
    判断客户端版本是否符合要求
    :param user_agent:
    :return: 0 不符合要求 1 最新版本 2 符合要求但可升级的版本
    """
    # 正则表达式用于匹配版本号
    match = re.search(r"NextTrace v([\d.]+)/", user_agent)
    if match:
        user_version = match.group(1)
        try:
            # 使用 packaging.version 比较版本号
            if version.parse(user_version) >= version.parse(accept_version):
                if version.parse(user_version) >= version.parse(latest_version):
                    return 1
                else:
                    return 2
            else:
                return 0
        except ValueError:
            # 如果无法解析版本号，视为不符合要求
            return 0
    else:
        match = re.search(r"NextTrace ([\d.]+)/", user_agent)  # 给 HomeBrew 版本的不规范版本擦屁股
        if match:
            user_version = match.group(1)
            try:
                # 使用 packaging.version 比较版本号
                if version.parse(user_version) >= version.parse(accept_version):
                    if version.parse(user_version) >= version.parse(latest_version):
                        return 1
                    else:
                        return 2
                else:
                    return 0
            except ValueError:
                # 如果无法解析版本号，视为不符合要求
                return 0
        else:
            # 如果无法解析版本号，视为不符合要求
            return 0


def html(filename):
    return flask.send_from_directory('html', filename)


def favicon():
    return flask.send_file('favicon.ico', mimetype='image/vnd.microsoft.icon')


def api():
    data = json.loads(request.data.decode("utf-8"))

    uName = str(uuid.uuid5(uuid.NAMESPACE_DNS, request.get_data().decode()))
    json_str = json.dumps(data, ensure_ascii=False)
    data = json.loads(json_str)
    with open('log/' + uName + '.json', 'w', encoding='utf-8') as f:
        f.write(json_str)
        logging.info("Saved log to log/" + uName + ".json")
    try:
        filename = process(data, filename=uName + '.html')
        logging.info("Saved html to " + filename)
    except Exception as e:
        logging.error("uuid:", uName)
        logging.error(e)
        print(traceback.format_exc())
        return "", 500
    # 根据UA判断版本
    version_result = is_version_acceptable(request.headers.get('User-Agent'))
    if version_result == 1:
        return filename, 200
    elif version_result == 2:
        return filename, 200
    else:
        return filename + '\n' + '您正在使用的版本，将在一个月内停止支持，请及时升级您的客户端', 406


class WebServer:
    def __init__(self):
        self.app = flask.Flask(__name__)
        if 'GUNICORN_CMD_ARGS' in os.environ:
            gunicorn_logger = logging.getLogger('gunicorn.error')
            self.app.logger.handlers = gunicorn_logger.handlers
            self.app.logger.setLevel(gunicorn_logger.level)
        else:
            logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')

        self.app.config['JSON_AS_ASCII'] = False

        self.urlPrefix = "http://localhost:8888/html/"

        self.app.route('/api', methods=['post'])(api)
        self.app.route('/html/<filename>', methods=['get'])(html)
        self.app.route('/favicon.ico', methods=['get'])(favicon)

    def run(self, host="0.0.0.0", port=18888, debug=True):
        self.app.run(host=host, port=port, debug=debug)


server = WebServer()
APP = server.app
if __name__ == '__main__':
    server.run()