token-tracker 0.4.6__tar.gz → 0.4.7__tar.gz

{token_tracker-0.4.6/src/token_tracker.egg-info → token_tracker-0.4.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.4.6
+Version: 0.4.7
 Summary: Track token usage across local AI agents (Claude Code, Codex)
 Author-email: stormzhang <stormzhang.dev@gmail.com>
 License: MIT License
@@ -42,7 +42,7 @@ Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: rich>=13.7
-Requires-Dist: questionary>=2.0
+Requires-Dist: questionary<3,>=2.0
 Provides-Extra: test
 Requires-Dist: pytest>=8.0; extra == "test"
 Provides-Extra: dev
@@ -83,6 +83,8 @@ Dynamic: license-file
 
 基于 Claude Code 官方自定义 StatusLine 接口，**数据完全来自本地 Claude，无任何推测**。
 
+> 状态栏接管是**可选**的：已有自定义 statusLine 时默认保留你的配置（向导里也可随时选 No），报表命令完全不受影响。注意：不接管时 `tt status` 的 CC 订阅额度段将没有数据来源（CC 配额只经状态栏脚本落盘）。
+
 ![Claude Code StatusLine](assets/screenshot-statusline-cc.png)
 
 <details>
@@ -159,7 +161,7 @@ curl -sSL https://raw.githubusercontent.com/stormzhang/token-tracker/main/instal
 ## 使用
 
 ```bash
-tt setup          # 交互配置向导（终端：上下键选语言 / 主题 / 各组件）；非 tty 环境自动全装
+tt setup          # 交互配置向导（终端：上下键选语言 / 主题 / 各组件）；非 tty 环境按推荐默认自动配置
 tt                # 过去一年 token 热力图 + 顶部三段概览（= tt daily）
 tt daily          # 同上（tt 无参即进 daily）
 tt status         # 过去 5h 实时面板
@@ -171,6 +173,8 @@ tt unsetup        # 卸载并恢复安装前的配置
 tt --version      # 查看版本（-v / -V 同义）
 ```
 
+> 多 agent 环境下想只看某一个 agent 的报表，加 `--claude` 或 `--codex` 即可（互斥），对 `status` / `daily` / `weekly` / `monthly` / `sessions` 均生效。例如 `tt daily --codex` 只显示 Codex 的热力图。会话内的 `daily` / `weekly` 默认已自动跟随当前会话的 agent，显式 flag 会覆盖该行为。
+
 > 💡 `tt daily` 是 GitHub 风格的 token 贡献热力图（深浅绿方格）。在 Claude Code 会话里输入 `!tt daily` 即可看到彩色热力图 —— 用户主动用 `!` 执行的命令，Claude Code 会渲染其 24-bit 真彩色输出。
 
 ## 配色主题
@@ -204,9 +208,10 @@ tt monthly --theme nord  # 任意报表临时换主题渲染（不持久化、
 
 1. **选语言** — 中文 / English（落 `~/.config/token-tracker/config.json`）
 2. **选配色主题** — 6 套主题上下键选择，每个选项右侧内联色板预览
-3. **启用 Codex 伪 statusline** — Yes/No（仅检测到 Codex 时）
+3. **接管 Claude Code 状态栏** — Yes/No（仅检测到 Claude Code 时；已有自定义 statusLine 会先备份、选 No 完全不碰）
+4. **启用 Codex 伪 statusline** — Yes/No（仅检测到 Codex 时）
 
-CI / 非 tty 环境（Docker / 脚本 / `curl|bash`）自动按默认全装：**语言跟随系统设置**、主题 mocha、组件全开。装好后想改任何一项，再跑一次 `tt setup` 即可。
+CI / 非 tty 环境（Docker / 脚本 / `curl|bash`）自动按**推荐默认**配置：语言跟随系统设置、主题 mocha、组件默认开启但**不替换已有自定义 statusLine**。装好后想改任何一项，再跑一次 `tt setup` 即可。
 
 ### 报告排序
 

{token_tracker-0.4.6 → token_tracker-0.4.7}/README.md

@@ -30,6 +30,8 @@
 
 基于 Claude Code 官方自定义 StatusLine 接口，**数据完全来自本地 Claude，无任何推测**。
 
+> 状态栏接管是**可选**的：已有自定义 statusLine 时默认保留你的配置（向导里也可随时选 No），报表命令完全不受影响。注意：不接管时 `tt status` 的 CC 订阅额度段将没有数据来源（CC 配额只经状态栏脚本落盘）。
+
 ![Claude Code StatusLine](assets/screenshot-statusline-cc.png)
 
 <details>
@@ -106,7 +108,7 @@ curl -sSL https://raw.githubusercontent.com/stormzhang/token-tracker/main/instal
 ## 使用
 
 ```bash
-tt setup          # 交互配置向导（终端：上下键选语言 / 主题 / 各组件）；非 tty 环境自动全装
+tt setup          # 交互配置向导（终端：上下键选语言 / 主题 / 各组件）；非 tty 环境按推荐默认自动配置
 tt                # 过去一年 token 热力图 + 顶部三段概览（= tt daily）
 tt daily          # 同上（tt 无参即进 daily）
 tt status         # 过去 5h 实时面板
@@ -118,6 +120,8 @@ tt unsetup        # 卸载并恢复安装前的配置
 tt --version      # 查看版本（-v / -V 同义）
 ```
 
+> 多 agent 环境下想只看某一个 agent 的报表，加 `--claude` 或 `--codex` 即可（互斥），对 `status` / `daily` / `weekly` / `monthly` / `sessions` 均生效。例如 `tt daily --codex` 只显示 Codex 的热力图。会话内的 `daily` / `weekly` 默认已自动跟随当前会话的 agent，显式 flag 会覆盖该行为。
+
 > 💡 `tt daily` 是 GitHub 风格的 token 贡献热力图（深浅绿方格）。在 Claude Code 会话里输入 `!tt daily` 即可看到彩色热力图 —— 用户主动用 `!` 执行的命令，Claude Code 会渲染其 24-bit 真彩色输出。
 
 ## 配色主题
@@ -151,9 +155,10 @@ tt monthly --theme nord  # 任意报表临时换主题渲染（不持久化、
 
 1. **选语言** — 中文 / English（落 `~/.config/token-tracker/config.json`）
 2. **选配色主题** — 6 套主题上下键选择，每个选项右侧内联色板预览
-3. **启用 Codex 伪 statusline** — Yes/No（仅检测到 Codex 时）
+3. **接管 Claude Code 状态栏** — Yes/No（仅检测到 Claude Code 时；已有自定义 statusLine 会先备份、选 No 完全不碰）
+4. **启用 Codex 伪 statusline** — Yes/No（仅检测到 Codex 时）
 
-CI / 非 tty 环境（Docker / 脚本 / `curl|bash`）自动按默认全装：**语言跟随系统设置**、主题 mocha、组件全开。装好后想改任何一项，再跑一次 `tt setup` 即可。
+CI / 非 tty 环境（Docker / 脚本 / `curl|bash`）自动按**推荐默认**配置：语言跟随系统设置、主题 mocha、组件默认开启但**不替换已有自定义 statusLine**。装好后想改任何一项，再跑一次 `tt setup` 即可。
 
 ### 报告排序
 

{token_tracker-0.4.6 → token_tracker-0.4.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "token-tracker"
-version = "0.4.6"
+version = "0.4.7"
 description = "Track token usage across local AI agents (Claude Code, Codex)"
 readme = "README.md"
 requires-python = ">=3.11"
@@ -26,7 +26,8 @@ classifiers = [
 ]
 dependencies = [
     "rich>=13.7",
-    "questionary>=2.0",
+    # wizard 补丁挂在 questionary 私有内部（InquirerControl._get_choice_tokens 等），锁大版本防破坏
+    "questionary>=2.0,<3",
 ]
 
 [project.optional-dependencies]
@@ -58,6 +59,12 @@ target-version = "py311"
 # E501(行长)故意不开：状态栏 ANSI 色码串本就长，启用只产生噪音
 select = ["E4", "E7", "E9", "F", "W", "I", "UP", "B"]
 
+[tool.ruff.lint.per-file-ignores]
+# 状态栏脚本模板：__STATUSLINE_*__ 占位符是烘焙时注入的（F821 未定义名是模板本性）；
+# E701/E401/I001/UP015/UP017 是脚本既有风格（紧凑 import、timezone.utc 兼容手改老 python）——
+# 豁免以保持渲染产物与历史版本逐字节一致（模板内容 = 落盘脚本内容）
+"src/token_tracker/templates/*" = ["F821", "E701", "E401", "I001", "UP015", "UP017"]
+
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 addopts = "-q"
@@ -67,3 +74,8 @@ python_version = "3.11"
 ignore_missing_imports = true
 # 宽松起步，后续逐步收紧
 check_untyped_defs = true
+
+[[tool.mypy.overrides]]
+# 状态栏脚本模板含烘焙占位符（__STATUSLINE_*__），不是可独立类型检查的模块
+module = "token_tracker.templates.*"
+ignore_errors = true

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/adapters/rate_limits.py

@@ -2,11 +2,9 @@ import json
 import os
 from datetime import UTC, datetime
 
+from ..config import STATUS_FILE
 from .types import RateLimits, normalize_pct
 
-# 与 hooks.STATUS_FILE 一致：tt 自己的产物集中放 ~/.config/token-tracker（XDG）
-STATUS_FILE = os.path.join(os.path.expanduser("~/.config/token-tracker"), "tt-status.json")
-
 
 def load_rate_limits() -> RateLimits | None:
     if not os.path.exists(STATUS_FILE):

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/adapters/types.py

@@ -129,29 +129,6 @@ class RateLimits:
     context_window: int | None = None
 
 
-@dataclass
-class P90Limits:
-    token_limit: int = 0
-    cost_limit: float = 0.0
-    message_limit: int = 0
-
-
-@dataclass
-class SessionBlock:
-    start_time: datetime
-    end_time: datetime
-    entries: list[UsageEntry] = field(default_factory=list)
-    input_tokens: int = 0
-    output_tokens: int = 0
-    cache_creation_tokens: int = 0
-    cache_read_tokens: int = 0
-    total_tokens: int = 0
-    cost_usd: float = 0.0
-    is_active: bool = False
-    burn_rate: float = 0.0
-    is_gap: bool = False
-
-
 @dataclass
 class StatusSummary:
     """tt status 头图面板：当天多 agent 合并的消耗汇总（add_token_fields 累加用）。"""

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/analyzer/aggregator.py

@@ -1,14 +1,15 @@
 from collections import defaultdict
 from collections.abc import Callable
-from datetime import timedelta
+from datetime import date, timedelta
 from typing import Any
 
 from ..adapters.types import DailyStats, MonthlyStats, SessionStats, UsageEntry, WeeklyStats
+from ..tz import system_tz
 from .cost import calculate_cost
 
 
 def add_token_fields(target: Any, e: UsageEntry, cost: float) -> None:
-    """把一条 UsageEntry 的 6 个通用 token/成本字段累加到任意 *Stats / SessionBlock 上。"""
+    """把一条 UsageEntry 的 6 个通用 token/成本字段累加到任意 *Stats / StatusSummary 上。"""
     target.input_tokens += e.input_tokens
     target.output_tokens += e.output_tokens
     target.cache_creation_tokens += e.cache_creation_tokens
@@ -47,31 +48,39 @@ def _aggregate_by_key(
     return sorted(by_key.values(), key=lambda s: getattr(s, sort_key))
 
 
+# daily/weekly/monthly 一律按**系统时区**的日历日分桶（entries 时间戳是 UTC；
+# 不转换会把北京时间 00:00-08:00 的用量记到前一天，且与 tt status 的「今天」口径打架）。
+# tz 在函数入口取一次（system_tz 每次调用要 readlink），闭包复用。
+
 def aggregate_daily(entries: list[UsageEntry]) -> list[DailyStats]:
+    tz = system_tz()
     return _aggregate_by_key(
         entries,
-        lambda e: e.timestamp.strftime("%Y-%m-%d"),
+        lambda e: e.timestamp.astimezone(tz).strftime("%Y-%m-%d"),
         lambda k, e: DailyStats(date=k),
         "date",
     )
 
 
 def aggregate_monthly(entries: list[UsageEntry]) -> list[MonthlyStats]:
+    tz = system_tz()
     return _aggregate_by_key(
         entries,
-        lambda e: e.timestamp.strftime("%Y-%m"),
+        lambda e: e.timestamp.astimezone(tz).strftime("%Y-%m"),
         lambda k, e: MonthlyStats(month=k),
         "month",
     )
 
 
 def aggregate_weekly(entries: list[UsageEntry]) -> list[WeeklyStats]:
+    tz = system_tz()
+
     def _week_key(e: UsageEntry) -> str:
-        monday = e.timestamp.date() - timedelta(days=e.timestamp.weekday())
-        return monday.isoformat()
+        d = e.timestamp.astimezone(tz).date()
+        return (d - timedelta(days=d.weekday())).isoformat()
 
     def _factory(k: str, e: UsageEntry) -> WeeklyStats:
-        monday = e.timestamp.date() - timedelta(days=e.timestamp.weekday())
+        monday = date.fromisoformat(k)
         sunday = monday + timedelta(days=6)
         return WeeklyStats(week=k, week_start=monday.strftime("%m-%d"), week_end=sunday.strftime("%m-%d"))
 
@@ -102,6 +111,10 @@ def aggregate_sessions(entries: list[UsageEntry]) -> list[SessionStats]:
             gap = (b.timestamp - a.timestamp).total_seconds() / 60
             if gap <= ACTIVE_GAP_CAP_MIN:
                 active += gap
+        # codex 会话是单条 entry（段内无相邻间隔可累计），活跃时长退化为整段跨度——
+        # 否则恒为 0，被 status 的 active_minutes>=5 过滤，codex 会话永远进不了当天列表
+        if len(session_entries) == 1 and first.session_end:
+            active = duration
 
         # 代表模型按 output_tokens（真实生成量）选，output 持平时用 total 兜底；
         # 不用 total 直接选，避免后台小模型（如 Haiku）读了大量上下文、cache_read 撑高 total 被误判为主力。

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/analyzer/cost.py

@@ -24,7 +24,7 @@ _warned_insecure = False
 # codex- 兜底覆盖 Codex 内部虚拟 model（如 codex-auto-review，stop-time auto-review gate 用）
 _FAMILY_FALLBACK = (
     ("claude-opus", "claude-opus-4-8"),
-    ("claude-sonnet", "claude-sonnet-4-6"),
+    ("claude-sonnet", "claude-sonnet-5"),
     ("claude-haiku", "claude-haiku-4-5-20251001"),
     ("claude-fable", "claude-fable-5"),
     ("codex-", "gpt-5.5"),
@@ -48,6 +48,10 @@ _FAMILY_FALLBACK = (
 # 解析不到定价的模型只提示一次，避免聚合时每条 entry 刷屏
 _warned_unknown_models: set[str] = set()
 
+# model → 解析出的定价 key。非精确命中要线性扫全表（litellm 数千 key），逐 entry 调用必须记忆化。
+# 命中后还校验 key 仍在当前 pricing 里（测试会整表替换 _pricing），失效则重算。
+_model_key_cache: dict[str, str | None] = {}
+
 
 def get_pricing() -> dict:
     global _pricing
@@ -85,6 +89,16 @@ def calculate_cost(entry: UsageEntry) -> float:
 def _resolve_model_key(model: str, pricing: dict) -> str | None:
     if not model:
         return None
+    if model in _model_key_cache:
+        cached = _model_key_cache[model]
+        if cached is None or cached in pricing:
+            return cached
+    key = _resolve_model_key_uncached(model, pricing)
+    _model_key_cache[model] = key
+    return key
+
+
+def _resolve_model_key_uncached(model: str, pricing: dict) -> str | None:
     if model in pricing:
         return model
 
@@ -243,6 +257,14 @@ def _fallback_pricing() -> dict:
         "claude-opus-4-7": _OPUS_PRICING,
         "claude-opus-4-6": _OPUS_PRICING,
         "claude-opus-4-5": _OPUS_PRICING,
+        # Sonnet 5 当前为导入价（$2 / $10 / $2.50 cache-write / $0.20 cache-read），截止 2026-08-31；
+        # 9-1 起标准价 $3 / $15（与 Sonnet 4.6 一致）。到期后需切价（注意：litellm 若已收录、以在线价为准）。
+        "claude-sonnet-5": {
+            "input_cost_per_token": 2e-6,
+            "output_cost_per_token": 10e-6,
+            "cache_creation_input_token_cost": 2.5e-6,
+            "cache_read_input_token_cost": 0.2e-6,
+        },
         "claude-sonnet-4-6": {
             "input_cost_per_token": 3e-6,
             "output_cost_per_token": 15e-6,

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/cli.py

@@ -1,5 +1,6 @@
 import os
 import sys
+from collections.abc import Callable
 from datetime import datetime
 
 from rich.text import Text
@@ -19,9 +20,9 @@ from .analyzer.aggregator import (
 from .analyzer.cost import calculate_cost
 from .hooks import is_setup, needs_update, setup, unsetup, update_hook
 from .i18n import t
+from .tz import system_tz
 from .ui import theme, themes
 from .ui.console import forced_color_console, get_console
-from .ui.format import system_tz
 from .ui.heatmap import render_daily_heatmap
 from .ui.status import render_sessions_view, render_status
 from .ui.tables import (
@@ -47,13 +48,13 @@ VALID_SORT_KEYS = (*SORT_ATTRS.keys(), "time")
 
 # 数据报表命令分发表：命令 → (聚合函数, 渲染函数, time 排序的属性, 无 --sort 时的默认属性, 默认降序)
 # time_attr 与 no_sort_attr 仅 daily 不同（默认按 token 排，--sort time 才按日期）
-_REPORT_COMMANDS = {
+_REPORT_COMMANDS: dict[str, tuple[Callable, Callable | None, str, str, bool]] = {
     "daily": (aggregate_daily, render_daily_heatmap, "date", "total_tokens", True),
     "weekly": (aggregate_weekly, render_weekly, "week", "week", True),
     "monthly": (aggregate_monthly, render_monthly, "month", "month", False),
-    # sessions 走专门分支调 render_sessions_view（顶部+底部仿 status、无额度段）；render_fn 槽
-    # 用 render_monthly 占位（仅为保持表项 callable、不经通用 render_fn 调用）；默认按 cost 倒序
-    "sessions": (aggregate_sessions, render_monthly, "start_time", "cost_usd", True),
+    # sessions 走专门分支调 render_sessions_view（顶部+底部仿 status、无额度段），
+    # 不经通用 render_fn（None 占位）；默认按 cost 倒序
+    "sessions": (aggregate_sessions, None, "start_time", "cost_usd", True),
 }
 
 
@@ -66,6 +67,24 @@ def _parse_limit(args: list[str], default: int) -> int:
     return default
 
 
+def _extract_agent_arg(args: list[str]) -> tuple[list[str], str | None]:
+    """提取 `--claude` / `--codex`，返回 (剩余 args, agent_id | None)。互斥（同时给报错退出）；
+    未给返回 None（走默认合并 + 会话自动识别）。用于多 agent 环境按需只看一个 agent（issue #19）。"""
+    remaining: list[str] = []
+    agent_id: str | None = None
+    _FLAG_TO_ID = {"--claude": "claude-code", "--codex": "codex"}
+    for a in args:
+        target = _FLAG_TO_ID.get(a)
+        if target is None:
+            remaining.append(a)
+            continue
+        if agent_id is not None and agent_id != target:
+            get_console().print(f"[red]{t('agent_filter_conflict')}[/red]")
+            sys.exit(1)
+        agent_id = target
+    return remaining, agent_id
+
+
 def _extract_theme_arg(args: list[str]) -> tuple[list[str], str | None]:
     """提取 --theme NAME，返回 (剩余 args, theme_name)；未给则 name=None。用于报表临时切主题、不落配置。"""
     remaining: list[str] = []
@@ -81,11 +100,12 @@ def _extract_theme_arg(args: list[str]) -> tuple[list[str], str | None]:
     return remaining, name
 
 
-def _parse_sort_args(args: list[str]) -> tuple[list[str], str | None, bool]:
-    """Extract --sort KEY and --asc from args, return (remaining, sort_key, descending)."""
+def _parse_sort_args(args: list[str]) -> tuple[list[str], str | None, bool | None]:
+    """Extract --sort KEY and --asc/--desc from args, return (remaining, sort_key, descending).
+    descending=None 表示用户没显式给方向（让 _apply_sort 落各命令默认方向）。"""
     remaining = []
     sort_key = None
-    descending = True
+    descending: bool | None = None
     i = 0
     while i < len(args):
         if args[i] == "--sort" and i + 1 < len(args):
@@ -103,9 +123,13 @@ def _parse_sort_args(args: list[str]) -> tuple[list[str], str | None, bool]:
     return remaining, sort_key, descending
 
 
-def _apply_sort(stats, sort_key: str | None, descending: bool, default_attr: str, default_reverse: bool):
+def _apply_sort(stats, sort_key: str | None, descending: bool | None,
+                default_attr: str, default_reverse: bool):
+    # 用户显式给了 --asc/--desc 就尊重（即便没配 --sort，如 `tt sessions --asc`）；
+    # 没给方向时：带 --sort 默认降序、无 --sort 落各命令默认方向
     if sort_key is None:
-        stats.sort(key=lambda s: getattr(s, default_attr), reverse=default_reverse)
+        reverse = default_reverse if descending is None else descending
+        stats.sort(key=lambda s: getattr(s, default_attr), reverse=reverse)
         return
     if sort_key not in VALID_SORT_KEYS:
         valid = ", ".join(VALID_SORT_KEYS)
@@ -114,7 +138,7 @@ def _apply_sort(stats, sort_key: str | None, descending: bool, default_attr: str
         return
     # "time" 不在 SORT_ATTRS → 退回 default_attr（各命令的时间字段）
     attr = SORT_ATTRS.get(sort_key, default_attr)
-    stats.sort(key=lambda s: getattr(s, attr), reverse=descending)
+    stats.sort(key=lambda s: getattr(s, attr), reverse=True if descending is None else descending)
 
 
 def _load_entries(agent_id: str, hours_back: int = 0):
@@ -122,10 +146,15 @@ def _load_entries(agent_id: str, hours_back: int = 0):
     return loader.load_entries(hours_back=hours_back) if loader else []
 
 
-def _aggregate_per_agent(agents, agg_fn):
+def _load_per_agent(agents) -> list[tuple]:
+    """(agent, 全量 entries) 列表：每个 agent 只从磁盘解析一次 JSONL（全量扫描很重），
+    daily/weekly/monthly 多种聚合复用同一份——避免 tt daily 这类命令重复扫 2~3 遍。"""
+    return [(a, _load_entries(a.id)) for a in agents]
+
+
+def _aggregate_per_agent(loaded, agg_fn):
     stats = []
-    for a in agents:
-        entries = _load_entries(a.id)
+    for a, entries in loaded:
         for s in agg_fn(entries):
             s.agent_id = a.id
             stats.append(s)
@@ -188,10 +217,12 @@ def _summary_from_sessions(sessions) -> StatusSummary:
 
 
 def _current_session_agent() -> str | None:
-    """识别当前所在的 agent 会话（靠环境变量）：Codex / Claude Code；独立终端返回 None。"""
+    """识别当前所在的 agent 会话（靠**会话内才有**的环境变量）：Codex / Claude Code；独立终端返回 None。
+    不能用 CLAUDE_CONFIG_DIR 判断——那是用户级配置变量（可长期 export 在 shell profile 里挪配置目录，
+    tt 自己也支持它），拿它当会话信号会让独立终端被误判成会话内（报表被过滤、首次运行进不了 wizard）。"""
     if os.environ.get("CODEX_THREAD_ID") or os.environ.get("CODEX_SANDBOX"):
         return "codex"
-    if os.environ.get("CLAUDE_CONFIG_DIR") or os.environ.get("CLAUDECODE"):
+    if os.environ.get("CLAUDECODE"):
         return "claude-code"
     return None
 
@@ -241,8 +272,9 @@ def _run_setup_flow() -> None:
 
 
 def _auto_setup() -> None:
-    """非交互环境（非 tty / CI / 会话内）：默认全装——语言跟随**系统设置**（绕过 CLI LANG）、
-    主题 mocha、组件全开。仅当用户从未配置过语言/主题时落默认（不覆盖已有选择）。
+    """非交互环境（非 tty / CI / 会话内）：语言跟随**系统设置**（绕过 CLI LANG）、主题 mocha、
+    组件按推荐默认（hooks.recommended_components：已有意图优先、不替换已有自定义 statusLine）。
+    仅当用户从未配置过语言/主题时落默认（不覆盖已有选择）。
     agent 守卫在 _run_setup_flow 已做，这里假设至少有一个 agent。"""
     if config.resolve_lang() is None:
         sys_lang = i18n._detect_system_lang()
@@ -250,7 +282,7 @@ def _auto_setup() -> None:
         i18n.set_lang(sys_lang)
     if not config.load_config().get("theme"):
         config.save_theme("mocha")
-    setup(auto=True)  # 组件默认全开
+    setup(auto=True)  # 组件走推荐默认（setup 内部 components=None → recommended_components）
     get_console().print(f"[dim]{t('auto_setup_hint')}[/dim]")
 
 
@@ -360,6 +392,8 @@ def main():
         _load_local_mock()
 
     args = sys.argv[1:]
+    # --claude / --codex：多 agent 环境按需只看一个 agent 的报表（issue #19），互斥；不给则走默认
+    args, filter_agent = _extract_agent_arg(args)
     # --theme NAME：临时覆盖主题（仅本次进程、不落配置/不重烘焙状态栏），对所有报表 + status 生效
     args, theme_override = _extract_theme_arg(args)
     if theme_override is not None:
@@ -411,6 +445,14 @@ def main():
             sys.exit(1)
 
     agents = detect_agents()
+    if filter_agent is not None:
+        # 显式指定 agent：收窄 agents 列表，所有报表命令自动跟随；未装 / 未检测到 → 报错退出
+        matched = [a for a in agents if a.id == filter_agent]
+        if not matched:
+            flag = "--claude" if filter_agent == "claude-code" else "--codex"
+            get_console().print(f"[red]{t('agent_not_detected', flag=flag)}[/red]")
+            sys.exit(1)
+        agents = matched
     agent_ids = {a.id for a in agents}
 
     if command in ("status", "dashboard"):
@@ -429,41 +471,44 @@ def main():
         sys.exit(1)
 
     # daily / weekly 跟随当前会话：CC 会话只看 CC、Codex 会话只看 Codex；
-    # 独立终端（识别不到会话）保持合并所有 agent。
+    # 独立终端（识别不到会话）保持合并所有 agent。显式 --claude / --codex 已在上面收窄 agents、优先级最高。
     report_agents = agents
-    if command in ("daily", "weekly"):
+    if filter_agent is None and command in ("daily", "weekly"):
         session_agent = _current_session_agent()
         if session_agent and session_agent in agent_ids:
             report_agents = [a for a in agents if a.id == session_agent]
     agent_names = [a.name for a in report_agents]
 
     agg_fn, render_fn, time_attr, no_sort_attr, default_reverse = _REPORT_COMMANDS[command]
-    stats = _aggregate_per_agent(report_agents, agg_fn)
+    loaded = _load_per_agent(report_agents)
+    stats = _aggregate_per_agent(loaded, agg_fn)
     default_attr = time_attr if sort_key == "time" else no_sort_attr
 
     if command == "sessions":
-        # sessions 看「最近的会话」：先过滤掉跨度 <5min 的碎片会话，再按时间取最近 N 条
-        #（否则史上高 cost 会话恒久霸榜、新会话和低成本 agent 永远进不了榜），这 N 条再按 cost（或 --sort）展示
-        kept = [s for s in stats if s.duration_minutes >= 5]
+        # sessions 看「最近的会话」：先过滤掉活跃 <5min 的碎片会话（与 tt status 同口径，
+        # 见 types.SessionStats.active_minutes），再按时间取最近 N 条（否则史上高 cost 会话
+        # 恒久霸榜、新会话和低成本 agent 永远进不了榜），这 N 条再按 cost（或 --sort）展示
+        kept = [s for s in stats if s.active_minutes >= 5]
         kept.sort(key=lambda s: s.start_time, reverse=True)
         shown = kept[:_parse_limit(rest_args, default=20)]
         _apply_sort(shown, sort_key, sort_desc, default_attr, default_reverse)
         render_sessions_view(_summary_from_sessions(shown), shown, agent_names)
         return
 
+    assert render_fn is not None  # sessions（render_fn=None）已在上面 return，其余命令都有渲染函数
     _apply_sort(stats, sort_key, sort_desc, default_attr, default_reverse)
     if command == "daily":
         # daily 顶部三卡片：Last 12 months + This Month + This Week，跟 weekly/monthly 同款样式、
-        # 复用 _render_month_summary / _render_week_summary。多算两份 weekly/monthly 聚合传进去。
-        render_fn(stats, agents=agent_names,  # type: ignore[operator]
-                  weekly=_aggregate_per_agent(report_agents, aggregate_weekly),
-                  monthly=_aggregate_per_agent(report_agents, aggregate_monthly))
+        # 复用 _render_month_summary / _render_week_summary。weekly/monthly 聚合复用同一份 entries。
+        render_fn(stats, agents=agent_names,
+                  weekly=_aggregate_per_agent(loaded, aggregate_weekly),
+                  monthly=_aggregate_per_agent(loaded, aggregate_monthly))
     elif command == "weekly":
-        render_weekly(stats, agents=agent_names, daily=_aggregate_per_agent(report_agents, aggregate_daily))
+        render_weekly(stats, agents=agent_names, daily=_aggregate_per_agent(loaded, aggregate_daily))
     elif command == "monthly":
         render_monthly(stats, agents=agent_names,
-                       daily=_aggregate_per_agent(report_agents, aggregate_daily),
-                       weekly=_aggregate_per_agent(report_agents, aggregate_weekly))
+                       daily=_aggregate_per_agent(loaded, aggregate_daily),
+                       weekly=_aggregate_per_agent(loaded, aggregate_weekly))
     else:
         render_fn(stats, agents=agent_names)
 

{token_tracker-0.4.6 → token_tracker-0.4.7}/src/token_tracker/config.py

@@ -1,4 +1,5 @@
-"""统一配置存储：~/.config/token-tracker/config.json（XDG，不绑 Claude Code）。
+"""统一配置存储：~/.config/token-tracker/config.json（固定路径、不读 XDG_CONFIG_HOME——
+跨平台行为一致 + statusline 模板内硬编码同路径；不绑 Claude Code）。
 
 包含所有 tt 自身偏好——主题、语言、组件意图。schema_version 用于未来格式迁移。
 
@@ -16,12 +17,17 @@ from .ui import themes
 
 CONFIG_DIR = os.path.expanduser("~/.config/token-tracker")
 CONFIG_PATH = os.path.join(CONFIG_DIR, "config.json")
+# CC statusline 缓存的单一权威路径（statusline 脚本写、tt status / adapters.rate_limits 读）。
+# 注意 templates/claude_statusline.py 因脚本独立运行硬编码了同一路径——改这里必须同步改模板。
+STATUS_FILE = os.path.join(CONFIG_DIR, "tt-status.json")
 SCHEMA_VERSION = 1
 
 # 引导版本：每次新增"值得让老用户重新走一遍 setup"的配置项时手动 +1（只能整数、一次 +1）。
 # 老用户 config 里没这字段 / 旧版本号 < 当前 → 触发重新引导（真终端弹 wizard、非 tty 静默 _auto_setup）。
 # 跟 SCHEMA_VERSION 解耦：那是数据格式版本，这是用户引导版本，bump 节奏完全不同。
 # 2（0.4.2）：强制所有现存用户（0.3.8/0.4.0=无字段=0、0.4.1=1，全 < 2）升级后重走一遍 setup。
+# 注：CC statusLine 变可选组件（issue #16/#17）时决定**不 bump**——小众需求不打断存量用户，
+# 存量 tt 用户 intent 缺失时由 is_setup 按「statusLine 已是 tt 的」推断为已配，想改的手动 tt setup。
 SETUP_VERSION = 2
 
 # 旧位置（独立 theme.json / lang.json），老用户首次读 config.json 不存在时自动合并迁移
@@ -157,6 +163,17 @@ def codex_faux_statusline_intent() -> bool | None:
     return val if isinstance(val, bool) else None
 
 
+def save_cc_statusline(enabled: bool) -> None:
+    """wizard 选完后写入意图（CC statusLine 是否由 tt 接管）。"""
+    _save_field("cc_statusline", bool(enabled))
+
+
+def cc_statusline_intent() -> bool | None:
+    """读用户对 CC statusLine 接管的意图。严格 bool；非 bool / 缺字段 → None（视为没表达）。"""
+    val = load_config().get("cc_statusline")
+    return val if isinstance(val, bool) else None
+
+
 # --- setup 引导版本（老用户升级后重新引导一次的判定依据） ---