diff --git a/main.py b/main.py index 059b9b8..1051dbc 100644 --- a/main.py +++ b/main.py @@ -4,6 +4,7 @@ from flask import Flask, request, jsonify import threading import queue import time +import argparse from ffmpeg_utils import process_ffmpeg from sanjuuni_utils import process_sanjuuni from file_cleanup import start_cleanup_thread @@ -122,7 +123,22 @@ def download_file_endpoint(file_id, filename): logging.error(f"下载文件时出错: {e}") return jsonify({'status': 'error', 'error': str(e)}), 500 + + + + +@app.route('/health', methods=['GET']) +def health_check(): + """健康检查接口,直接返回 'ok'""" + return "OK" + + if __name__ == '__main__': - logging.basicConfig(level=logging.INFO) + # 配置命令行参数解析 + parser = argparse.ArgumentParser(description='启动FFmpeg处理服务') + parser.add_argument('--port', type=int, default=5000, help='服务监听的端口号(默认: 5000)') + args = parser.parse_args() + + logging.info(f"启动应用程序,端口: {args.port}...") start_cleanup_thread() - app.run(debug=True) \ No newline at end of file + app.run(host='0.0.0.0', port=args.port, threaded=True) \ No newline at end of file