Files
tick-stock-panel/backend/app/config.py
T
shy3130 41be5096dc feat: 桌面客户端打包 (pywebview + PyInstaller) + 系统通知
- 桌面入口: desktop.py (单实例锁 + 端口探测 + uvicorn + pywebview 窗口)
- 配置层: config.py 加 sys.frozen 检测, frozen 下 data_dir 指向用户目录
- 系统通知: notify_adapter.py 三平台 (Win winotify / mac osascript / Linux notify-send)
  接入 quote_service 告警出口, preferences 开关控制, 前端监控设置加 toggle
- 前端: 监控设置加系统通知开关, 系统设置加检查更新入口
- 打包: packaging/tickflow.spec (onedir, collect-all 原生库, 排除 vectorbt)
- CI: .github/workflows/release.yml 手动触发 + 平台可选 (windows/macos/linux)
- 依赖: platformdirs/winotify/plyer 入主依赖, pywebview 入 desktop extras
- README 加桌面客户端下载章节
- bump 版本号到 0.1.33
2026-06-23 21:47:29 +08:00

106 lines
3.8 KiB
Python

"""全局配置 — 从环境变量 / .env 读取。"""
from __future__ import annotations
import sys
from pathlib import Path
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:
"""桌面版用户数据根目录 (跨平台持久可写)。
Windows: %LOCALAPPDATA%/TickFlowStockPanel/TickFlowStockPanel
macOS: ~/Library/Application Support/TickFlowStockPanel
Linux: ~/.local/share/TickFlowStockPanel
注意: platformdirs 已含应用名, 切勿再拼一层。
"""
try:
from platformdirs import user_data_dir
return Path(user_data_dir("TickFlowStockPanel"))
except Exception: # noqa: BLE001
# platformdirs 不可用时兜底: 可执行文件旁的 data/
return Path(sys.executable).resolve().parent / "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=str(_RESOURCE_ROOT / ".env") if not _IS_FROZEN else ".env",
env_file_encoding="utf-8",
extra="ignore",
)
# TickFlow
tickflow_api_key: str = Field(default="", description="留空启用 free 模式")
# AI
ai_provider: str = "openai_compat"
ai_base_url: str = "https://api.alysc.top"
ai_api_key: str = ""
ai_model: str = "gpt-5.5"
ai_daily_token_budget: int = 5_000_000
# Server
host: str = "0.0.0.0"
port: int = 3018
log_level: str = "INFO"
backtest_range_guard: bool = False
# Data — frozen: 用户数据目录; 非 frozen: 项目根目录的 data/ (可被 DATA_DIR 覆盖)
data_dir: Path = _user_data_root() if _IS_FROZEN else (_PROJECT_ROOT / "data")
# 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:
"""是否走 Free 模式。优先看 secrets.json,其次看 .env。"""
from app import secrets_store
return not secrets_store.get_tickflow_key()
settings = Settings()