dekun
823aeda42a
Wire bootstrap_deploy_secrets into setup_env.sh (one-time HUB_BRIDGE_TOKEN, FLASK_SECRET_KEY, HUB_SESSION_SECRET). Remove AI section from instance env UI; hub saves OPENAI settings to all four .env files. SSO unchanged. Co-authored-by: Cursor <cursoragent@cursor.com>
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
"""中控 .env 读写与 PM2 重启。"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from lib.env.env_file_lib import apply_env_updates, load_env_file_into_environ
|
|
from lib.env.shared_env_lib import (
|
|
apply_ai_env_to_all,
|
|
build_ai_env_payload,
|
|
restart_hub_and_instances_pm2,
|
|
)
|
|
|
|
HUB_DIR = Path(__file__).resolve().parent
|
|
|
|
|
|
def hub_env_path() -> str:
|
|
return str(HUB_DIR / ".env")
|
|
|
|
|
|
def update_hub_credentials(*, new_password: str, new_username: str | None = None) -> list[str]:
|
|
updates: dict[str, str] = {"HUB_PASSWORD": new_password}
|
|
if new_username:
|
|
updates["HUB_USERNAME"] = new_username
|
|
path = hub_env_path()
|
|
changed = apply_env_updates(path, updates)
|
|
if changed:
|
|
load_env_file_into_environ(path)
|
|
return changed
|
|
|
|
|
|
def get_hub_ai_env_payload() -> dict[str, Any]:
|
|
return build_ai_env_payload(hub_env_path())
|
|
|
|
|
|
def save_hub_ai_env(updates: dict[str, str]) -> dict[str, Any]:
|
|
return apply_ai_env_to_all(updates)
|
|
|
|
|
|
def restart_all_pm2() -> dict[str, Any]:
|
|
return restart_hub_and_instances_pm2()
|
|
|
|
|
|
def restart_hub_pm2() -> dict[str, Any]:
|
|
app_name = (os.getenv("PM2_APP_NAME") or "").strip() or "manual-trading-hub"
|
|
if not sys.platform.startswith("linux"):
|
|
return {"ok": False, "msg": "仅 Linux 服务器支持 PM2 重启", "app": app_name}
|
|
try:
|
|
proc = subprocess.run(
|
|
["pm2", "restart", app_name, "--update-env"],
|
|
capture_output=True,
|
|
text=True,
|
|
timeout=120,
|
|
)
|
|
ok = proc.returncode == 0
|
|
return {
|
|
"ok": ok,
|
|
"app": app_name,
|
|
"msg": (proc.stdout or proc.stderr or "").strip()[:500],
|
|
"returncode": proc.returncode,
|
|
}
|
|
except FileNotFoundError:
|
|
return {"ok": False, "msg": "未找到 pm2 命令", "app": app_name}
|
|
except subprocess.TimeoutExpired:
|
|
return {"ok": False, "msg": "pm2 restart 超时", "app": app_name}
|
|
except Exception as e:
|
|
return {"ok": False, "msg": str(e), "app": app_name}
|