Python 后端提交

Python 后端（FastAPI + FastMCP + ...）的初始版本号设定为 0.1.0，这是 uv 在 pypriject.toml 里给我自动设置的，我觉得有道理。
2026-04-21 13:38:46 +08:00
parent 14eadaab86
commit b284c3c260
27 changed files with 1691 additions and 0 deletions
--- a/main.py
+++ b/main.py
@@ -0,0 +1,86 @@
+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 已结束。")