75cc105ebc
- .gitignore: 忽略 docker/offline 离线部署包（镜像/运行时等大文件） - 安全加固: 新增 compute/api/security.py 及各端安全测试，补充 docs/security-hardening.md - 数据库: 新增完整初始化 SQL 与 docs/database-config.md - 数据转换与评测: 修复类型检查、增强校验并补充测试 - Docker 配置与环境变量更新 Co-Authored-By: Claude <noreply@anthropic.com>
30 lines
1023 B
Python
30 lines
1023 B
Python
"""计算节点 API 安全配置：Swagger / ReDoc / OpenAPI 文档路由开关。"""
|
||
from __future__ import annotations
|
||
|
||
import os
|
||
from typing import Any
|
||
|
||
|
||
def docs_enabled() -> bool:
|
||
"""判断 FastAPI 文档路由（/docs、/redoc、/openapi.json）是否开放。
|
||
|
||
显式配置 ENABLE_DOCS 时以之为准；否则仅在关闭 token 鉴权
|
||
（COMPUTE_AUTH_ENABLED=false，本地开发）时开放，生产环境默认关闭，
|
||
避免未授权访问泄露 API 结构。
|
||
"""
|
||
raw = os.getenv("ENABLE_DOCS", "").strip().lower()
|
||
if raw in {"true", "false"}:
|
||
return raw == "true"
|
||
auth_enabled = os.getenv("COMPUTE_AUTH_ENABLED", "true").lower() == "true"
|
||
return not auth_enabled
|
||
|
||
|
||
def docs_kwargs() -> dict[str, Any]:
|
||
"""返回传入 FastAPI 的文档路由参数。
|
||
|
||
关闭时 FastAPI 不注册 /docs、/redoc、/openapi.json，访问一律返回 404。
|
||
"""
|
||
if docs_enabled():
|
||
return {}
|
||
return {"docs_url": None, "redoc_url": None, "openapi_url": None}
|