重置项目
This commit is contained in:
+95
-18
@@ -1,27 +1,93 @@
|
||||
"""全局配置 — 硬编码,个人工具不依赖 .env。"""
|
||||
"""全局配置 — 从环境变量 / .env 读取。"""
|
||||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
from pydantic import Field
|
||||
from pydantic import Field, model_validator
|
||||
from pydantic_settings import BaseSettings, SettingsConfigDict
|
||||
|
||||
# ── 运行环境检测 ──────────────────────────────────────────
|
||||
# PyInstaller 打包后: __file__ 指向临时解压目录 _MEIPASS, 不能作为路径基准。
|
||||
# 此时:
|
||||
# - 只读资源 (tiers.yaml / 前端 dist) 放在 _MEIPASS 内
|
||||
# - 可写用户数据 (data_dir) 放在可执行文件旁的用户目录
|
||||
# 非 frozen 模式 (开发/Docker): 保持原有 __file__ 推导, 行为完全不变。
|
||||
_IS_FROZEN = getattr(sys, "frozen", False)
|
||||
|
||||
|
||||
def _user_data_root() -> Path:
|
||||
"""桌面版用户数据根目录。
|
||||
|
||||
定位策略 (按优先级):
|
||||
1. 环境变量 DATA_DIR (pydantic-settings 自动注入到 settings.data_dir, 不在此处理)
|
||||
2. 打包桌面版: exe 同级的 data/ 子目录 (<安装目录>/data/)
|
||||
—— 与程序同处一个总目录 (用户选择的安装目录), 视觉直观, 便于备份/迁移。
|
||||
3. 非 frozen (开发模式): 项目根 data/
|
||||
|
||||
为什么不用 platformdirs 默认 (%LOCALAPPDATA%) 作为主路径:
|
||||
- 落在 C 盘系统目录, 用户不易察觉, 占系统盘空间
|
||||
- 用户期望「数据跟随程序」(便于备份/迁移)
|
||||
为什么放 {app}/data (exe 旁的 data/) 而非 {app} 外的兄弟目录:
|
||||
- 用户体验: 用户选了安装目录, 自然期望「程序和数据都在这」, 单一总目录更直观。
|
||||
- 数据安全: Inno Setup 覆盖安装(升级)时只往 {app} 写新程序文件, 不会清空
|
||||
目录里不在安装清单上的运行时文件 (data/ 即此类), 故覆盖安装不丢数据。
|
||||
(注意: 卸载时需在 .iss 中豁免 data/, 见 packaging/tickflow.iss 的 [UninstallDelete]。)
|
||||
旧版本数据迁移: 见 DataStore._migrate_legacy_data_dir(), 老用户首次启动自动搬迁。
|
||||
"""
|
||||
# 打包桌面版: exe 同级的 data/ 子目录 (与程序同一总目录, 覆盖安装不丢数据)
|
||||
if _IS_FROZEN:
|
||||
exe_dir = Path(sys.executable).resolve().parent
|
||||
return exe_dir / "data"
|
||||
|
||||
# 开发模式: 项目根 data/
|
||||
return _PROJECT_ROOT / "data"
|
||||
|
||||
|
||||
def _resource_root() -> Path:
|
||||
"""只读资源根目录。
|
||||
|
||||
frozen: PyInstaller 解压目录 (_MEIPASS)
|
||||
非 frozen: 项目根目录 (源码树)
|
||||
"""
|
||||
if _IS_FROZEN:
|
||||
# sys._MEIPASS 是 PyInstaller 注入的解压根
|
||||
return Path(getattr(sys, "_MEIPASS", Path(sys.executable).resolve().parent))
|
||||
return Path(__file__).resolve().parent.parent.parent
|
||||
|
||||
|
||||
def _project_root() -> Path:
|
||||
"""项目根目录 (非 frozen 用)。"""
|
||||
return Path(__file__).resolve().parent.parent.parent
|
||||
|
||||
|
||||
_PROJECT_ROOT = _project_root()
|
||||
_RESOURCE_ROOT = _resource_root()
|
||||
|
||||
|
||||
class Settings(BaseSettings):
|
||||
model_config = SettingsConfigDict(
|
||||
env_file=None, # 不读取 .env
|
||||
env_file=str(_RESOURCE_ROOT / ".env") if not _IS_FROZEN else ".env",
|
||||
env_file_encoding="utf-8",
|
||||
extra="ignore",
|
||||
)
|
||||
|
||||
# 数据源 API Key(个人工具直接写死;如分享代码请改为空字符串或从环境变量注入)
|
||||
tickflow_api_key: str = "tk_94a20304993f45b5b0e376b9767597cc"
|
||||
# TickFlow
|
||||
tickflow_api_key: str = Field(default="", description="留空启用 free 模式")
|
||||
|
||||
# AI(可选,留空即关闭)
|
||||
# AI
|
||||
ai_provider: str = "openai_compat"
|
||||
ai_base_url: str = "https://api.deepseek.com/v1"
|
||||
ai_base_url: str = "https://api.alysc.top"
|
||||
ai_api_key: str = ""
|
||||
ai_model: str = "deepseek-chat"
|
||||
ai_daily_token_budget: int = 500_000
|
||||
ai_model: str = "gpt-5.5"
|
||||
ai_codex_command: str = "codex"
|
||||
# 默认浏览器风格 UA,绕过 Cloudflare 等 CDN/WAF 的 Bot 拦截(Issue #8)。
|
||||
# 用户可在 AI 设置页按需修改。
|
||||
ai_user_agent: str = (
|
||||
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
|
||||
"AppleWebKit/537.36 (KHTML, like Gecko) "
|
||||
"Chrome/131.0.0.0 Safari/537.36"
|
||||
)
|
||||
|
||||
# Server
|
||||
host: str = "0.0.0.0"
|
||||
@@ -29,16 +95,27 @@ class Settings(BaseSettings):
|
||||
log_level: str = "INFO"
|
||||
backtest_range_guard: bool = False
|
||||
|
||||
# 访问门控:留空则不启用,部署时通过 ACCESS_UUID 环境变量注入
|
||||
# 优先级:ADMIN_TOKEN > 动态 UUID > ACCESS_UUID
|
||||
access_uuid: str = ""
|
||||
# 管理员初始令牌,硬编码以便开箱即用;如需更安全可改为空字符串并从环境变量 ADMIN_TOKEN 注入
|
||||
admin_token: str = "admin7226132"
|
||||
# Auth — 首次启动时预置访问密码(明文, 仅用于初始化, 详见 services/auth.bootstrap_from_env)
|
||||
# 公网服务器部署时免去 SSH 端口转发设密码的麻烦。写入 auth.json(哈希)后即不再读取。
|
||||
auth_password: str = ""
|
||||
|
||||
# 路径 — 硬编码为 Docker 容器内路径,确保数据持久化
|
||||
data_dir: Path = Path("/app/data")
|
||||
tiers_yaml: Path = Path("/app/tiers.yaml")
|
||||
static_dir: Path = Path("/app/static")
|
||||
# Data — frozen: exe 同级 data/ 子目录; 非 frozen: 项目根 data/
|
||||
# (均可被环境变量 DATA_DIR 覆盖, pydantic-settings 自动注入)
|
||||
data_dir: Path = _user_data_root()
|
||||
|
||||
# tiers.yaml 路径 — frozen: 资源目录内; 非 frozen: 项目根目录
|
||||
tiers_yaml: Path = _RESOURCE_ROOT / "tiers.yaml" if _IS_FROZEN else _PROJECT_ROOT / "tiers.yaml"
|
||||
|
||||
# 静态文件(前端 dist) — frozen: 资源目录的 static/; 非 frozen: frontend/dist
|
||||
static_dir: Path = _RESOURCE_ROOT / "static" if _IS_FROZEN else (_PROJECT_ROOT / "frontend" / "dist")
|
||||
|
||||
@model_validator(mode="after")
|
||||
def _resolve_paths(self) -> Settings:
|
||||
"""确保 data_dir 是绝对路径（环境变量传入的相对路径基于项目根目录解析）。"""
|
||||
if not self.data_dir.is_absolute():
|
||||
# 相对路径基于项目根目录解析，而非 CWD
|
||||
self.data_dir = (_PROJECT_ROOT / self.data_dir).resolve()
|
||||
return self
|
||||
|
||||
@property
|
||||
def use_free_mode(self) -> bool:
|
||||
|
||||
Reference in New Issue
Block a user