from flask import redirect, url_for

from Util.CommonUtil import *

app = Flask(__name__)

# 默认页面
@app.before_request
def redirect_to_default_static_file():
    # 检查请求的URL是否符合默认路由的模式
    if request.path == '/':
        # 重定向到静态文件，例如 'index.html'
        return redirect(url_for('static', filename='index.html'))


# 根据姓名获取相关信息
# 测试样例 ：http://127.0.0.1:5000/XingMing?xm=%E9%BB%84%E6%B5%B7
@app.route('/XingMing', methods=['GET'])
def XingMing():
    xm = request.args.get('xm')
    if len(xm) < 2:
        return jsonify({'success': False, 'message': '姓名输入错误'}), 201

    return getXingMing(xm)


# 根据姓氏和性别超名
# 测试样例 ：http://127.0.0.1:5000/QingMing?gsname=黄&qimingsex=1
@app.route('/QingMing', methods=['GET'])
def QingMing():
    gsname = request.args.get('gsname')
    qimingsex = request.args.get('qimingsex')

    if gsname not in extract_chinese_surnames():
        return jsonify({'success': False, 'message': '姓氏输入错误'}), 201

    if qimingsex not in ['1', '0']:
        return jsonify({'success': False, 'message': '性别输入错误'}), 201

    return getQingMing(gsname, qimingsex)


if __name__ == '__main__':
    # CPU序列号
    cpu_serial = get_cpu_serial()
    # 生成授权码
    authorization_code = generate_authorization_code(cpu_serial)
    print(f"Authorization Code: {authorization_code}")
    # 启动Flask
    app.run(port=5000)  # 指定端口为5000