|
@@ -1,6 +1,10 @@
|
|
|
from flask import Flask, request, jsonify, send_from_directory
|
|
|
import os
|
|
|
import time
|
|
|
+from dotenv import load_dotenv
|
|
|
+
|
|
|
+# 加载环境变量
|
|
|
+load_dotenv()
|
|
|
|
|
|
# 导入自定义模块
|
|
|
from utils.logger import logger, reset_process_id
|
|
@@ -16,11 +20,11 @@ from flask_cors import CORS
|
|
|
CORS(app)
|
|
|
|
|
|
# 配置常量
|
|
|
-app.config['TEMPLATE_FOLDER'] = 'template' # 模板文件目录
|
|
|
-app.config['OUTPUT_FOLDER'] = 'outputs' # 处理后文件保存目录
|
|
|
+app.config['TEMPLATE_FOLDER'] = os.getenv('TEMPLATE_FOLDER', 'template') # 模板文件目录
|
|
|
+app.config['OUTPUT_FOLDER'] = os.getenv('OUTPUT_FOLDER', 'outputs') # 处理后文件保存目录
|
|
|
app.config['MAX_CONTENT_LENGTH'] = 16 * 1024 * 1024 # 限制上传文件大小为16MB
|
|
|
-app.config['DEBUG_MODE'] = False # 调试模式:True启用变量检查和验证,False禁用
|
|
|
-app.config['LOG_FOLDER'] = 'logs' # 日志保存目录
|
|
|
+app.config['DEBUG_MODE'] = os.getenv('DEBUG_MODE', 'False').lower() == 'true' # 调试模式:True启用变量检查和验证,False禁用
|
|
|
+app.config['LOG_FOLDER'] = os.getenv('LOG_FOLDER', 'logs') # 日志保存目录
|
|
|
|
|
|
# 确保模板、输出和日志目录存在
|
|
|
os.makedirs(app.config['TEMPLATE_FOLDER'], exist_ok=True)
|
|
@@ -182,5 +186,7 @@ if __name__ == '__main__':
|
|
|
init_scheduler(app.config['OUTPUT_FOLDER'])
|
|
|
logger.info("Flask应用程序开始运行")
|
|
|
# 使用host='0.0.0.0'使Flask监听所有网络接口,这样局域网内的其他设备可以访问
|
|
|
- # port=5000指定端口号
|
|
|
- app.run(host='0.0.0.0', port=5000, debug=True)
|
|
|
+ # 从环境变量获取端口号,默认为5000
|
|
|
+ port = int(os.getenv('PORT', 5000))
|
|
|
+ debug = os.getenv('FLASK_DEBUG', '0') == '1'
|
|
|
+ app.run(host='0.0.0.0', port=port, debug=debug)
|