token-tracker 0.3.2__tar.gz → 0.3.3__tar.gz

{token_tracker-0.3.2 → token_tracker-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.3.2
+Version: 0.3.3
 Summary: Track token usage across local AI agents (Claude Code, Codex)
 Requires-Python: >=3.11
 Requires-Dist: rich>=13.7

{token_tracker-0.3.2 → token_tracker-0.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "token-tracker"
-version = "0.3.2"
+version = "0.3.3"
 description = "Track token usage across local AI agents (Claude Code, Codex)"
 requires-python = ">=3.11"
 dependencies = [

{token_tracker-0.3.2 → token_tracker-0.3.3}/src/hooks.py

@@ -2,6 +2,7 @@ import json
 import os
 import re
 import stat
+import sys
 import tomllib
 
 from .ui.tables import console
@@ -10,7 +11,7 @@ CLAUDE_SETTINGS = os.path.expanduser("~/.claude/settings.json")
 HOOK_SCRIPT_PATH = os.path.expanduser("~/.claude/tt-statusline.py")
 CODEX_CONFIG = os.path.expanduser("~/.codex/config.toml")
 CODEX_BACKUP = os.path.expanduser("~/.codex/tt-backup.json")
-HOOK_VERSION = "1.4"
+HOOK_VERSION = "1.5"
 _BACKUP_KEY = "tokenTracker"
 _PREV_SL_KEY = "previousStatusLine"
 _SL_REGEX = re.compile(r'status_line\s*=\s*\[.*?\]', re.DOTALL)
@@ -25,7 +26,7 @@ CODEX_STATUS_LINE = [
 
 HOOK_SCRIPT = r'''#!/usr/bin/env python3
 """Claude Code statusLine — 状态栏显示 + 数据持久化到 tt-status.json"""
-__version__ = "1.4"
+__version__ = "1.5"
 import json, os, re, subprocess, sys, tempfile
 from datetime import datetime, timezone
 
@@ -37,6 +38,9 @@ C = {
     "peach": "\033[38;5;216m", "dim": "\033[2m", "reset": "\033[0m",
 }
 
+if sys.platform == "win32":
+    sys.stdout.reconfigure(encoding="utf-8", errors="replace")
+
 
 def vlen(s):
     return len(ANSI_RE.sub("", s))
@@ -47,13 +51,15 @@ def get_width():
         return max(1, os.get_terminal_size(2).columns - 4)
     except Exception:
         pass
-    import fcntl, struct, termios
-    try:
-        with open('/dev/tty', 'r') as tty:
-            res = fcntl.ioctl(tty, termios.TIOCGWINSZ, b'\x00' * 8)
-            return max(1, struct.unpack('hh', res[:4])[1] - 4)
-    except Exception:
-        return 116
+    if os.name != "nt":
+        try:
+            import fcntl, struct, termios
+            with open('/dev/tty', 'r') as tty:
+                res = fcntl.ioctl(tty, termios.TIOCGWINSZ, b'\x00' * 8)
+                return max(1, struct.unpack('hh', res[:4])[1] - 4)
+        except Exception:
+            pass
+    return 116
 
 
 def color_by_pct(pct):
@@ -237,6 +243,7 @@ def render(data, now):
     output = [" | ".join(line) for line in (line1, line2, line3) if line]
     if output:
         print("\n".join(output))
+        sys.stdout.flush()
 
 
 def main():
@@ -310,13 +317,18 @@ def _installed_hook_version() -> str | None:
 
 
 def needs_update() -> bool:
+    if not os.path.isdir(os.path.dirname(HOOK_SCRIPT_PATH)):
+        return False
     return _installed_hook_version() != HOOK_VERSION
 
 
 def update_hook() -> None:
+    if not os.path.isdir(os.path.dirname(HOOK_SCRIPT_PATH)):
+        return
     with open(HOOK_SCRIPT_PATH, "w", encoding="utf-8") as f:
         f.write(HOOK_SCRIPT)
-    os.chmod(HOOK_SCRIPT_PATH, os.stat(HOOK_SCRIPT_PATH).st_mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
+    if os.name != "nt":
+        os.chmod(HOOK_SCRIPT_PATH, os.stat(HOOK_SCRIPT_PATH).st_mode | stat.S_IXUSR | stat.S_IXGRP | stat.S_IXOTH)
 
 
 # --- setup ---
@@ -358,7 +370,8 @@ def _setup_claude() -> None:
         console.print(f"[yellow]检测到已有 statusLine，备份后替换[/yellow]")
         settings.setdefault(_BACKUP_KEY, {})[_PREV_SL_KEY] = existing
 
-    settings["statusLine"] = {"type": "command", "command": f"python3 {HOOK_SCRIPT_PATH}"}
+    python = sys.executable or "python3"
+    settings["statusLine"] = {"type": "command", "command": f"{python} {HOOK_SCRIPT_PATH}"}
 
     with open(CLAUDE_SETTINGS, "w", encoding="utf-8") as f:
         json.dump(settings, f, indent=2, ensure_ascii=False)

{token_tracker-0.3.2 → token_tracker-0.3.3}/token_tracker.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.3.2
+Version: 0.3.3
 Summary: Track token usage across local AI agents (Claude Code, Codex)
 Requires-Python: >=3.11
 Requires-Dist: rich>=13.7