from flask import Flask, request, jsonify
import os
import handler
import logging

app = Flask(__name__)

# 从环境变量获取日志级别,默认为'WARNING'
log_level_name = os.environ.get("LOGLEVEL", "WARNING")
# 获取logging模块对应的日志级别对象
log_level = getattr(logging, log_level_name.upper(), None)

if not isinstance(log_level, int):
    raise ValueError(f'Invalid log level: {log_level_name}')

# 设置Flask应用的日志级别
app.logger.setLevel(log_level)

# 尝试设置Werkzeug日志记录器的日志级别
werkzeug_logger = logging.getLogger('werkzeug')
werkzeug_logger.setLevel(log_level)

config = {
    "inputFile": "input.txt",
    "outputFile": "output.txt"
}

init_query = {"itemKeyArray": ["1"]}


@app.route('/process', methods=['POST'])
def process():
    """处理请求的接口"""
    data = request.get_json()
    if data and 'itemKeyArray' in data:
        input_file = config["inputFile"]
        output_file = config["outputFile"]
        result = handler.process_data(data, input_file, output_file)
        return jsonify(result)
    else:
        return jsonify({"status": "fail", "message": "Invalid data"}), 400


if __name__ == '__main__':
    nodePort = os.environ.get('innerPort', 5000)

    app.run(host="0.0.0.0", port=nodePort)