352 lines
12 KiB
Python
352 lines
12 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
VWED任务系统集成启动器
|
|||
|
|
先显示配置界面，配置完成后直接启动服务
|
|||
|
|
支持后台运行模式
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
import threading
|
|||
|
|
import time
|
|||
|
|
import logging
|
|||
|
|
import argparse
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# 添加项目根目录到Python路径
|
|||
|
|
if getattr(sys, 'frozen', False):
|
|||
|
|
# 打包后的环境
|
|||
|
|
project_root = Path(sys.executable).parent
|
|||
|
|
else:
|
|||
|
|
# 开发环境
|
|||
|
|
project_root = Path(__file__).parent.parent
|
|||
|
|
|
|||
|
|
sys.path.insert(0, str(project_root))
|
|||
|
|
|
|||
|
|
# 导入配置相关模块
|
|||
|
|
try:
|
|||
|
|
from config_tool.startup_loader import load_startup_config, ensure_config_exists
|
|||
|
|
except ImportError:
|
|||
|
|
print("警告: 无法导入配置加载器")
|
|||
|
|
|
|||
|
|
class IntegratedLauncher:
|
|||
|
|
"""集成启动器"""
|
|||
|
|
|
|||
|
|
def __init__(self, headless=False):
|
|||
|
|
self.app = None
|
|||
|
|
self.service_started = False
|
|||
|
|
self.config_completed = False
|
|||
|
|
self.headless = headless
|
|||
|
|
|
|||
|
|
# 在无窗口模式下重定向标准流
|
|||
|
|
if self.headless:
|
|||
|
|
self._setup_headless_mode()
|
|||
|
|
|
|||
|
|
self.setup_logging()
|
|||
|
|
|
|||
|
|
def _setup_headless_mode(self):
|
|||
|
|
"""设置无窗口模式的标准流重定向"""
|
|||
|
|
try:
|
|||
|
|
# 创建空的文件对象来代替标准流
|
|||
|
|
import io
|
|||
|
|
null_file = io.StringIO()
|
|||
|
|
|
|||
|
|
# 重定向标准流到空文件，避免None对象引起的问题
|
|||
|
|
if sys.stdout is None:
|
|||
|
|
sys.stdout = null_file
|
|||
|
|
if sys.stderr is None:
|
|||
|
|
sys.stderr = null_file
|
|||
|
|
if sys.stdin is None:
|
|||
|
|
sys.stdin = io.StringIO("")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
# 如果重定向失败，记录错误但不中断程序
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def setup_logging(self):
|
|||
|
|
"""设置日志记录"""
|
|||
|
|
log_file = Path("vwed_service.log")
|
|||
|
|
|
|||
|
|
# 配置日志格式
|
|||
|
|
formatter = logging.Formatter(
|
|||
|
|
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 文件日志处理器
|
|||
|
|
file_handler = logging.FileHandler(log_file, encoding='utf-8')
|
|||
|
|
file_handler.setLevel(logging.INFO)
|
|||
|
|
file_handler.setFormatter(formatter)
|
|||
|
|
|
|||
|
|
# 获取根日志器
|
|||
|
|
self.logger = logging.getLogger('IntegratedLauncher')
|
|||
|
|
self.logger.setLevel(logging.INFO)
|
|||
|
|
self.logger.addHandler(file_handler)
|
|||
|
|
|
|||
|
|
# 如果不是无头模式，也添加控制台日志
|
|||
|
|
if not self.headless:
|
|||
|
|
console_handler = logging.StreamHandler()
|
|||
|
|
console_handler.setLevel(logging.INFO)
|
|||
|
|
console_handler.setFormatter(formatter)
|
|||
|
|
self.logger.addHandler(console_handler)
|
|||
|
|
|
|||
|
|
def log_and_print(self, message, level=logging.INFO):
|
|||
|
|
"""记录日志并在非无头模式下打印"""
|
|||
|
|
self.logger.log(level, message)
|
|||
|
|
if not self.headless:
|
|||
|
|
print(message)
|
|||
|
|
|
|||
|
|
def show_startup_message(self):
|
|||
|
|
"""显示启动信息"""
|
|||
|
|
self.log_and_print("=" * 60)
|
|||
|
|
self.log_and_print(" VWED任务系统启动器")
|
|||
|
|
self.log_and_print("=" * 60)
|
|||
|
|
self.log_and_print("正在初始化系统...")
|
|||
|
|
|
|||
|
|
# 确保配置文件存在
|
|||
|
|
try:
|
|||
|
|
ensure_config_exists()
|
|||
|
|
load_startup_config()
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log_and_print(f"配置初始化失败: {e}", logging.ERROR)
|
|||
|
|
|
|||
|
|
def start_configuration_gui(self):
|
|||
|
|
"""启动配置GUI"""
|
|||
|
|
# 检查是否存在配置文件且配置是否完整
|
|||
|
|
config_file = Path("config.ini")
|
|||
|
|
if config_file.exists() and self.check_config_complete():
|
|||
|
|
self.log_and_print("检测到完整的配置文件，跳过配置界面直接启动服务...")
|
|||
|
|
self.log_and_print("如需重新配置，请删除config.ini文件后重新启动")
|
|||
|
|
|
|||
|
|
# 模拟配置完成，直接启动服务
|
|||
|
|
self.on_config_completed({})
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 无头模式下跳过GUI
|
|||
|
|
if self.headless:
|
|||
|
|
self.log_and_print("无头模式运行，使用默认配置启动服务...")
|
|||
|
|
self.start_service()
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
from config_tool.integrated_config_gui import IntegratedConfigGUI
|
|||
|
|
|
|||
|
|
self.log_and_print("启动配置界面...")
|
|||
|
|
self.config_gui = IntegratedConfigGUI(self.on_config_completed)
|
|||
|
|
self.config_gui.run()
|
|||
|
|
|
|||
|
|
except ImportError as e:
|
|||
|
|
self.log_and_print(f"无法导入配置界面: {e}", logging.ERROR)
|
|||
|
|
# 如果GUI不可用，使用默认配置直接启动服务
|
|||
|
|
self.log_and_print("使用默认配置启动服务...")
|
|||
|
|
self.start_service()
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log_and_print(f"配置界面启动失败: {e}", logging.ERROR)
|
|||
|
|
self.start_service()
|
|||
|
|
|
|||
|
|
def check_config_complete(self):
|
|||
|
|
"""检查配置是否完整"""
|
|||
|
|
try:
|
|||
|
|
import configparser
|
|||
|
|
config = configparser.ConfigParser()
|
|||
|
|
config.read("config.ini", encoding='utf-8')
|
|||
|
|
|
|||
|
|
# 检查必要的配置项
|
|||
|
|
required_sections = {
|
|||
|
|
'database': ['username', 'password', 'host'],
|
|||
|
|
'api': ['tf_api_base_url']
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
for section, keys in required_sections.items():
|
|||
|
|
if section not in config:
|
|||
|
|
return False
|
|||
|
|
for key in keys:
|
|||
|
|
if key not in config[section] or not config[section][key].strip():
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log_and_print(f"检查配置文件失败: {e}", logging.ERROR)
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def on_config_completed(self, config_data):
|
|||
|
|
"""配置完成回调"""
|
|||
|
|
self.log_and_print("配置完成，准备启动服务...")
|
|||
|
|
self.config_completed = True
|
|||
|
|
|
|||
|
|
# 应用配置到环境变量
|
|||
|
|
if 'database' in config_data:
|
|||
|
|
os.environ.update({
|
|||
|
|
'DB_USER': config_data['database'].get('username', 'root'),
|
|||
|
|
'DB_PASSWORD': config_data['database'].get('password', 'root'),
|
|||
|
|
'DB_HOST': config_data['database'].get('host', 'localhost')
|
|||
|
|
})
|
|||
|
|
|
|||
|
|
if 'api' in config_data:
|
|||
|
|
os.environ['TF_API_BASE_URL'] = config_data['api'].get('tf_api_base_url', 'http://111.231.146.230:4080/jeecg-boot')
|
|||
|
|
|
|||
|
|
self.log_and_print("环境变量已设置，启动服务...")
|
|||
|
|
|
|||
|
|
# 直接启动服务，不使用守护线程
|
|||
|
|
self.start_service()
|
|||
|
|
|
|||
|
|
def start_service(self):
|
|||
|
|
"""启动VWED服务"""
|
|||
|
|
if self.service_started:
|
|||
|
|
self.log_and_print("服务已经在运行中...")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
self.log_and_print("\n" + "=" * 60)
|
|||
|
|
self.log_and_print(" 启动VWED任务系统服务")
|
|||
|
|
self.log_and_print("=" * 60)
|
|||
|
|
|
|||
|
|
# 写入PID文件用于后台管理
|
|||
|
|
pid_file = Path("vwed_service.pid")
|
|||
|
|
with open(pid_file, 'w') as f:
|
|||
|
|
f.write(str(os.getpid()))
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 导入主程序
|
|||
|
|
from app import app, settings
|
|||
|
|
from utils.logger import get_logger
|
|||
|
|
|
|||
|
|
logger = get_logger("launcher")
|
|||
|
|
|
|||
|
|
# 标记服务已启动
|
|||
|
|
self.service_started = True
|
|||
|
|
|
|||
|
|
port = 8000
|
|||
|
|
self.log_and_print(f"服务启动成功！")
|
|||
|
|
self.log_and_print(f"请在浏览器中访问: http://localhost:{port}")
|
|||
|
|
self.log_and_print(f"API文档地址: http://localhost:{port}/docs")
|
|||
|
|
if not self.headless:
|
|||
|
|
self.log_and_print("\n按 Ctrl+C 停止服务")
|
|||
|
|
|
|||
|
|
# 启动服务器
|
|||
|
|
import uvicorn
|
|||
|
|
|
|||
|
|
# 在无窗口模式下使用简化配置
|
|||
|
|
if self.headless:
|
|||
|
|
# 无窗口模式：禁用reload、单worker、简化日志
|
|||
|
|
uvicorn.run(
|
|||
|
|
app, # 直接传递app对象而不是字符串
|
|||
|
|
host="0.0.0.0",
|
|||
|
|
port=port,
|
|||
|
|
reload=False,
|
|||
|
|
workers=1,
|
|||
|
|
access_log=False, # 禁用访问日志避免格式化问题
|
|||
|
|
log_config=None # 禁用默认日志配置
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
# 前台模式：使用正常配置
|
|||
|
|
uvicorn.run(
|
|||
|
|
"app:app",
|
|||
|
|
host="0.0.0.0",
|
|||
|
|
port=port,
|
|||
|
|
reload=settings.SERVER_RELOAD,
|
|||
|
|
workers=settings.SERVER_WORKERS
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
except Exception as uvicorn_error:
|
|||
|
|
self.log_and_print(f"Uvicorn启动失败: {uvicorn_error}", logging.ERROR)
|
|||
|
|
|
|||
|
|
# 尝试使用最基本的方式启动
|
|||
|
|
self.log_and_print("尝试使用基本模式启动服务...", logging.WARNING)
|
|||
|
|
try:
|
|||
|
|
from app import app
|
|||
|
|
import uvicorn
|
|||
|
|
uvicorn.run(
|
|||
|
|
app,
|
|||
|
|
host="0.0.0.0",
|
|||
|
|
port=8000,
|
|||
|
|
log_level="error" # 只显示错误日志
|
|||
|
|
)
|
|||
|
|
except Exception as fallback_error:
|
|||
|
|
self.log_and_print(f"基本模式启动也失败: {fallback_error}", logging.ERROR)
|
|||
|
|
raise fallback_error
|
|||
|
|
|
|||
|
|
except ImportError as e:
|
|||
|
|
self.log_and_print(f"无法导入主程序模块: {e}", logging.ERROR)
|
|||
|
|
self.log_and_print("请检查项目依赖是否正确安装", logging.ERROR)
|
|||
|
|
except Exception as e:
|
|||
|
|
self.log_and_print(f"服务启动失败: {e}", logging.ERROR)
|
|||
|
|
import traceback
|
|||
|
|
self.log_and_print(traceback.format_exc(), logging.ERROR)
|
|||
|
|
finally:
|
|||
|
|
self.service_started = False
|
|||
|
|
# 清理PID文件
|
|||
|
|
pid_file = Path("vwed_service.pid")
|
|||
|
|
if pid_file.exists():
|
|||
|
|
pid_file.unlink()
|
|||
|
|
|
|||
|
|
def cleanup(self):
|
|||
|
|
"""清理资源"""
|
|||
|
|
try:
|
|||
|
|
self.log_and_print("清理资源中...")
|
|||
|
|
if hasattr(self, 'config_gui'):
|
|||
|
|
if hasattr(self.config_gui, 'root') and self.config_gui.root:
|
|||
|
|
self.config_gui.root.destroy()
|
|||
|
|
|
|||
|
|
# 清理PID文件
|
|||
|
|
pid_file = Path("vwed_service.pid")
|
|||
|
|
if pid_file.exists():
|
|||
|
|
pid_file.unlink()
|
|||
|
|
except:
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主函数"""
|
|||
|
|
# 解析命令行参数
|
|||
|
|
parser = argparse.ArgumentParser(description='VWED任务系统集成启动器')
|
|||
|
|
parser.add_argument('--headless', action='store_true',
|
|||
|
|
help='无头模式运行（后台运行，不显示GUI）')
|
|||
|
|
parser.add_argument('--daemon', action='store_true',
|
|||
|
|
help='守护进程模式运行')
|
|||
|
|
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
launcher = None
|
|||
|
|
try:
|
|||
|
|
launcher = IntegratedLauncher(headless=args.headless)
|
|||
|
|
|
|||
|
|
# 显示启动信息
|
|||
|
|
launcher.show_startup_message()
|
|||
|
|
|
|||
|
|
# 启动配置GUI
|
|||
|
|
launcher.start_configuration_gui()
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
if launcher:
|
|||
|
|
launcher.log_and_print("\n程序被用户中断")
|
|||
|
|
else:
|
|||
|
|
print("\n程序被用户中断")
|
|||
|
|
except Exception as e:
|
|||
|
|
if launcher:
|
|||
|
|
launcher.log_and_print(f"程序异常退出: {e}", logging.ERROR)
|
|||
|
|
import traceback
|
|||
|
|
launcher.log_and_print(traceback.format_exc(), logging.ERROR)
|
|||
|
|
else:
|
|||
|
|
print(f"程序异常退出: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
if not args.headless:
|
|||
|
|
input("按回车键退出...")
|
|||
|
|
finally:
|
|||
|
|
# 清理资源
|
|||
|
|
if launcher:
|
|||
|
|
launcher.cleanup()
|
|||
|
|
|
|||
|
|
def cleanup_and_exit():
|
|||
|
|
"""清理资源并退出"""
|
|||
|
|
print("正在清理资源...")
|
|||
|
|
# 强制退出所有线程
|
|||
|
|
os._exit(0)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
try:
|
|||
|
|
main()
|
|||
|
|
except:
|
|||
|
|
cleanup_and_exit()
|