90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
|
|
# ===== Main.py 顶部放置（所有 import 之前）=====
|
|
import os
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from Utils.LogManager import LogManager
|
|
|
|
if "IOSAI_PYTHON" not in os.environ:
|
|
base_path = Path(sys.argv[0]).resolve()
|
|
base_dir = base_path.parent if base_path.is_file() else base_path
|
|
sidecar = base_dir / "python-rt" / ("python.exe" if os.name == "nt" else "python")
|
|
if sidecar.exists():
|
|
os.environ["IOSAI_PYTHON"] = str(sidecar)
|
|
# ==============================================
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
from Module.DeviceInfo import DeviceInfo
|
|
from Module.FlaskSubprocessManager import FlaskSubprocessManager
|
|
from Utils.DevDiskImageDeployer import DevDiskImageDeployer
|
|
|
|
# 确定 exe 或 py 文件所在目录
|
|
BASE = Path(getattr(sys, 'frozen', False) and sys.executable or __file__).resolve().parent
|
|
LOG_DIR = BASE / "log"
|
|
LOG_DIR.mkdir(exist_ok=True) # 确保 log 目录存在
|
|
|
|
print(f"日志目录: {LOG_DIR}")
|
|
|
|
def _run_flask_role():
|
|
from Module import FlaskService
|
|
port = int(os.getenv("FLASK_COMM_PORT", "34566")) # 固定端口的兜底仍是 34567
|
|
app_factory = getattr(FlaskService, "create_app", None)
|
|
app = app_factory() if callable(app_factory) else FlaskService.app
|
|
app.run(host="0.0.0.0", port=port + 1, debug=False, use_reloader=False)
|
|
|
|
if "--role=flask" in sys.argv:
|
|
_run_flask_role()
|
|
sys.exit(0)
|
|
|
|
# 启动锁
|
|
def main(arg):
|
|
if len(arg) != 2 or arg[1] != "iosai":
|
|
sys.exit(0)
|
|
|
|
# 项目入口
|
|
if __name__ == "__main__":
|
|
|
|
# 清空日志
|
|
LogManager.clearLogs()
|
|
|
|
# main(sys.argv)
|
|
|
|
# 添加iOS开发包到电脑上
|
|
deployer = DevDiskImageDeployer(verbose=True)
|
|
deployer.deploy_all()
|
|
|
|
# 启动 Flask 子进程
|
|
manager = FlaskSubprocessManager.get_instance()
|
|
manager.start()
|
|
|
|
# 设备监听（即使失败/很快返回，也不会导致主进程退出）
|
|
try:
|
|
info = DeviceInfo()
|
|
info.listen()
|
|
except Exception as e:
|
|
print("[WARN] Device listener not running:", e)
|
|
|
|
# === 保活：阻塞主线程，直到收到 Ctrl+C/关闭 ===
|
|
import threading, time, signal
|
|
stop = threading.Event()
|
|
|
|
def _handle(_sig, _frm):
|
|
stop.set()
|
|
|
|
# Windows 上 SIGINT/SIGTERM 都可以拦到
|
|
try:
|
|
signal.signal(signal.SIGINT, _handle)
|
|
signal.signal(signal.SIGTERM, _handle)
|
|
except Exception:
|
|
pass # 某些环境可能不支持，忽略
|
|
|
|
try:
|
|
while not stop.is_set():
|
|
time.sleep(1)
|
|
finally:
|
|
# 进程退出前记得把子进程关掉
|
|
manager.stop()
|