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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: token-tracker
-Version: 0.3.2
+Version: 0.3.4
 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.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "token-tracker"
-version = "0.3.2"
+version = "0.3.4"
 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.4}/src/cli.py

@@ -92,8 +92,6 @@ def _fit_screen(text: str, height: int, scroll_offset: int) -> tuple[str, int]:
 
 
 def _show_interactive_dashboard(agents):
-    import tty
-    import termios
     import shutil
     from io import StringIO
     from rich.console import Console as RichConsole
@@ -135,7 +133,7 @@ def _show_interactive_dashboard(agents):
             sys.stdout.write("\033[2J\033[3J\033[H" + screen)
             sys.stdout.flush()
 
-            key = _read_key(tty, termios)
+            key = _read_key()
             if key == "left":
                 current = (current - 1) % len(agents)
                 scroll_offset = 0
@@ -158,9 +156,11 @@ def _show_interactive_dashboard(agents):
         _tables.console = orig
 
 
-def _read_key(tty, termios):
+def _read_key_unix():
     import os as _os
     import select
+    import tty
+    import termios
     fd = sys.stdin.fileno()
     old = termios.tcgetattr(fd)
     try:
@@ -204,6 +204,44 @@ def _read_key(tty, termios):
         termios.tcsetattr(fd, termios.TCSADRAIN, old)
 
 
+def _read_key_win():
+    import msvcrt
+    ch = msvcrt.getch()
+    if ch in (b"\xe0", b"\x00"):
+        ch2 = msvcrt.getch()
+        if ch2 == b"K":
+            return "left"
+        if ch2 == b"M":
+            return "right"
+        if ch2 == b"H":
+            return "up"
+        if ch2 == b"P":
+            return "down"
+        if ch2 == b"I":
+            return "page_up"
+        if ch2 == b"Q":
+            return "page_down"
+        return "other"
+    if ch == b"h":
+        return "left"
+    if ch == b"l":
+        return "right"
+    if ch == b"k":
+        return "up"
+    if ch == b"j":
+        return "down"
+    if ch == b"b":
+        return "page_up"
+    if ch == b"f":
+        return "page_down"
+    if ch in (b"q", b"Q", b"\x03", b"\x1b"):
+        return "quit"
+    return "other"
+
+
+_read_key = _read_key_win if sys.platform == "win32" else _read_key_unix
+
+
 def _get_version() -> str:
     from importlib.metadata import version
     return version("token-tracker")

{token_tracker-0.3.2 → token_tracker-0.3.4}/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.4}/token_tracker.egg-info/PKG-INFO

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