120 lines
4.0 KiB
Python
120 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
测试配置。从项目 soul-api/.env* 或 scripts/test/.env.test 或环境变量读取。
|
||
必须明确指定测试环境，避免误测正式库。
|
||
"""
|
||
import os
|
||
from pathlib import Path
|
||
|
||
# 项目根目录（scripts/test 的上级的上级）
|
||
PROJECT_ROOT = Path(__file__).resolve().parent.parent.parent
|
||
SOUL_API_ENV = PROJECT_ROOT / "soul-api"
|
||
TEST_DIR = Path(__file__).resolve().parent
|
||
|
||
|
||
def _apply_env_file(path: Path) -> None:
|
||
"""将 .env 文件中的变量加载到 os.environ（仅当未设置时）"""
|
||
if not path.exists():
|
||
return
|
||
for line in path.read_text(encoding="utf-8", errors="ignore").splitlines():
|
||
line = line.strip()
|
||
if not line or line.startswith("#") or "=" not in line:
|
||
continue
|
||
k, v = line.split("=", 1)
|
||
k = k.strip()
|
||
v = v.strip().strip('"').strip("'")
|
||
if k and k not in os.environ:
|
||
os.environ[k] = v
|
||
|
||
|
||
# 优先加载 scripts/test/.env.test（本地覆盖）
|
||
_apply_env_file(TEST_DIR / ".env.test")
|
||
|
||
# 环境与 API 地址映射（与 miniprogram/app.js、soul-api/.env* 一致）
|
||
ENV_PROFILES = {
|
||
"local": "http://localhost:8080",
|
||
"souldev": "https://souldev.quwanzhi.com",
|
||
"soulapi": "https://soulapi.quwanzhi.com",
|
||
}
|
||
|
||
# 环境中文名（用于提示）
|
||
ENV_LABELS = {
|
||
"local": "本地",
|
||
"souldev": "测试",
|
||
"soulapi": "正式",
|
||
}
|
||
|
||
|
||
def _load_env_file(path: Path) -> dict:
|
||
"""解析 .env 文件为 dict"""
|
||
out = {}
|
||
if not path.exists():
|
||
return out
|
||
for line in path.read_text(encoding="utf-8", errors="ignore").splitlines():
|
||
line = line.strip()
|
||
if not line or line.startswith("#") or "=" not in line:
|
||
continue
|
||
k, v = line.split("=", 1)
|
||
out[k.strip()] = v.strip().strip('"').strip("'")
|
||
return out
|
||
|
||
|
||
def _resolve_api_base() -> tuple[str, str]:
|
||
"""
|
||
解析 API 地址与当前环境标签。
|
||
优先级：SOUL_TEST_ENV > SOUL_API_BASE > 从 soul-api/.env 读取 > 默认 local
|
||
"""
|
||
env_val = os.environ.get("SOUL_TEST_ENV", "").strip().lower()
|
||
explicit_base = os.environ.get("SOUL_API_BASE", "").strip().rstrip("/")
|
||
|
||
if explicit_base:
|
||
# 显式指定了地址，根据地址推断环境标签
|
||
label = "自定义"
|
||
for k, v in ENV_PROFILES.items():
|
||
if v.rstrip("/") == explicit_base:
|
||
label = ENV_LABELS.get(k, k)
|
||
break
|
||
return explicit_base, label
|
||
|
||
if env_val in ENV_PROFILES:
|
||
return ENV_PROFILES[env_val], ENV_LABELS.get(env_val, env_val)
|
||
|
||
# 尝试从 soul-api/.env 读取 API_BASE_URL
|
||
env_path = SOUL_API_ENV / ".env"
|
||
env_dev = SOUL_API_ENV / ".env.development"
|
||
env_prod = SOUL_API_ENV / ".env.production"
|
||
for p in [env_path, env_dev, env_prod]:
|
||
loaded = _load_env_file(p)
|
||
if loaded.get("API_BASE_URL"):
|
||
base = loaded["API_BASE_URL"].rstrip("/")
|
||
for k, v in ENV_PROFILES.items():
|
||
if v.rstrip("/") == base:
|
||
return base, ENV_LABELS.get(k, k)
|
||
return base, "项目配置"
|
||
|
||
# 默认本地，并提示未显式指定
|
||
return ENV_PROFILES["local"], ENV_LABELS["local"]
|
||
|
||
|
||
API_BASE, ENV_LABEL = _resolve_api_base()
|
||
|
||
# 管理端登录（与 scripts/本地启动.sh 一致；不同环境账号可能不同）
|
||
ADMIN_USERNAME = os.environ.get("SOUL_ADMIN_USERNAME", "admin")
|
||
ADMIN_PASSWORD = os.environ.get("SOUL_ADMIN_PASSWORD", "admin123")
|
||
|
||
# 小程序开发环境登录用 userId（仅 local/souldev 且 APP_ENV=development 时可用）
|
||
MINIAPP_DEV_USER_ID = os.environ.get("SOUL_MINIPROGRAM_DEV_USER_ID", "")
|
||
|
||
|
||
def get_env_banner() -> str:
|
||
"""返回测试环境提示横幅"""
|
||
return (
|
||
"\n"
|
||
"========================================\n"
|
||
f" 测试环境: {ENV_LABEL} ({API_BASE})\n"
|
||
"========================================\n"
|
||
" 若需切换，请设置: SOUL_TEST_ENV=local|souldev|soulapi\n"
|
||
" 或: SOUL_API_BASE=<完整 API 地址>\n"
|
||
"========================================\n"
|
||
)
|