MangoFanFanw
b284c3c260
Python 后端（FastAPI + FastMCP + ...）的初始版本号设定为 0.1.0，这是 uv 在 pypriject.toml 里给我自动设置的，我觉得有道理。
87 lines
3.1 KiB
Python
87 lines
3.1 KiB
Python
from pathlib import Path
|
|
from secrets import token_urlsafe
|
|
|
|
from njupt_api.baselib import config, logger
|
|
from router import __version__
|
|
|
|
DATA_DIR = Path.cwd() / "data"
|
|
TEMP_DIR = Path.cwd() / "temp"
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
with open(file=Path.cwd() / "njupt_api" / "art.txt", mode="r", encoding="utf-8") as f:
|
|
print(f.read().format(__version__)) # noqa:T201
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
import uvicorn
|
|
|
|
logger.success("Ciallo~(∠·ω< )⌒★")
|
|
|
|
# 创建需要的工作目录
|
|
if DATA_DIR.exists() and DATA_DIR.is_dir():
|
|
logger.debug(f"工作目录 {DATA_DIR=!s} 已存在。")
|
|
else:
|
|
DATA_DIR.mkdir(parents=True, exist_ok=True)
|
|
logger.debug(f"工作目录 {DATA_DIR=!s} 已创建。")
|
|
|
|
if TEMP_DIR.exists() and TEMP_DIR.is_dir():
|
|
logger.debug(f"工作目录 {TEMP_DIR=!s} 已存在。")
|
|
# 清空 temp 工作目录中的已有文件
|
|
c = 0
|
|
for item in TEMP_DIR.iterdir():
|
|
item.unlink()
|
|
c += 1
|
|
logger.debug(f"清理了 temp 工作目录中的 {c} 个已有文件。")
|
|
else:
|
|
TEMP_DIR.mkdir(parents=True, exist_ok=True)
|
|
logger.debug(f"工作目录 {TEMP_DIR=!s} 已创建。")
|
|
|
|
# 如果没有 toml 配置文件就创建
|
|
try:
|
|
config.sync_load_json()
|
|
except FileNotFoundError:
|
|
config.init_config()
|
|
config.sync_create_json()
|
|
|
|
lines = [
|
|
"",
|
|
"🌐 WebUI 管理面板将运行在 /webui 端点下，登录需要令牌。",
|
|
"",
|
|
"============================================================",
|
|
"🔐 管理后端令牌",
|
|
"============================================================",
|
|
"",
|
|
]
|
|
try:
|
|
with open(file=DATA_DIR / "token.txt", mode="r", encoding="utf-8") as f:
|
|
token = f.readline().strip()
|
|
lines.insert(-2, f"🔐 使用已经存在的管理后端令牌 | {token}")
|
|
except FileNotFoundError:
|
|
token = token_urlsafe(32)
|
|
with open(file=DATA_DIR / "token.txt", mode="w", encoding="utf-8") as f:
|
|
f.write(token)
|
|
lines.insert(-2, f"🔐 新的管理后端令牌已生成 | {token}")
|
|
lines.insert(-2, "🔐 你需要此令牌以登录管理员面板并可视化地配置 NJUPT Suan API。")
|
|
logger.info("\n".join(lines))
|
|
|
|
logger.info("🥭 准备 uvicorn run ...")
|
|
host = config.get("system", "host", "0.0.0.0")
|
|
port = config.get("system", "port", 8000)
|
|
reload = config.get("system", "reload", True)
|
|
logger.debug(f"启动参数 - {host=} | {port=} | {reload=}")
|
|
logger.debug("这些参数无法自动热重载。如果你修改了他们，请 Ctrl + C 关闭并重新启动 Suan API。")
|
|
uvicorn.run(
|
|
"server:app",
|
|
host=host,
|
|
port=port,
|
|
reload=reload,
|
|
reload_dirs=["njupt_api", "router"],
|
|
access_log=False,
|
|
log_level="critical",
|
|
timeout_graceful_shutdown=2,
|
|
)
|
|
logger.debug("退出时未清理 temp 工作目录，将在 Suan API 下次启动时清理。")
|
|
logger.info("🥭 uvicorn run 已结束。")
|