codex-cli-usage 0.1.0__py3-none-any.whl → 0.1.3__py3-none-any.whl

codex_cli_usage/__init__.py

@@ -23,9 +23,49 @@ import urllib.request
 from datetime import datetime, timezone
 from pathlib import Path
 
+_TTY = sys.stdout.isatty()
+
+
+def _find_codex_path(filename: str) -> Path:
+    """Return the first existing path for a .codex file.
+
+    Checks the native path first (~/.codex/<filename>).  On Windows, if the
+    native path doesn't exist, also checks WSL distros via //wsl$/.
+    """
+    native = Path.home() / ".codex" / filename
+    if native.exists():
+        return native
+
+    if sys.platform == "win32":
+        import subprocess
+        try:
+            result = subprocess.run(
+                ["wsl", "-l", "-q"],
+                capture_output=True, timeout=5,
+            )
+            decoded = result.stdout.decode("utf-16-le", errors="ignore")
+            distros = [d.strip() for d in decoded.splitlines() if d.strip()]
+        except Exception:
+            distros = []
+
+        for distro in distros:
+            wsl_home = Path(f"//wsl$/{distro}/home")
+            try:
+                users = [u for u in wsl_home.iterdir() if u.is_dir()]
+            except OSError:
+                continue
+            for user_dir in users:
+                candidate = user_dir / ".codex" / filename
+                if candidate.exists():
+                    return candidate
+
+    # Fall back to native path even if it doesn't exist yet
+    return native
+
+
 CODEX_DIR = Path.home() / ".codex"
-AUTH_FILE = CODEX_DIR / "auth.json"
-USAGE_FILE = CODEX_DIR / "usage-limits.json"
+AUTH_FILE = _find_codex_path("auth.json")
+USAGE_FILE = _find_codex_path("usage-limits.json")
 DAEMON_INTERVAL = 300  # 5 minutes
 CLIENT_ID = "app_EMoamEEZ73f0CkXaXp7hrann"
 USAGE_URL = "https://chatgpt.com/backend-api/codex/usage"
@@ -205,11 +245,11 @@ def cmd_status(raw_json=False):
         print(json.dumps(data, indent=2))
         return
 
-    R = "\033[0;31m"
-    Y = "\033[0;33m"
-    G = "\033[0;32m"
-    D = "\033[0;90m"
-    RST = "\033[0m"
+    R = "\033[0;31m" if _TTY else ""
+    Y = "\033[0;33m" if _TTY else ""
+    G = "\033[0;32m" if _TTY else ""
+    D = "\033[0;90m" if _TTY else ""
+    RST = "\033[0m" if _TTY else ""
 
     def color_pct(pct):
         p = int(pct)
@@ -260,7 +300,8 @@ def cmd_status(raw_json=False):
 def cmd_daemon(interval: int = DAEMON_INTERVAL):
     """Run in foreground, refresh every `interval` seconds."""
     signal.signal(signal.SIGINT, lambda *_: sys.exit(0))
-    signal.signal(signal.SIGTERM, lambda *_: sys.exit(0))
+    if hasattr(signal, "SIGTERM"):
+        signal.signal(signal.SIGTERM, lambda *_: sys.exit(0))
 
     print(f"codex-cli-usage daemon started (refreshing every {interval}s)")
     print(f"Writing to {USAGE_FILE}")
@@ -306,11 +347,11 @@ def _get_cached_usage(max_age: int = DAEMON_INTERVAL) -> dict:
 
 def cmd_statusline():
     """Statusline command. Reads cached usage and prints compact summary."""
-    R = "\033[0;31m"
-    Y = "\033[0;33m"
-    G = "\033[0;32m"
-    D = "\033[0;90m"
-    RST = "\033[0m"
+    R = "\033[0;31m" if _TTY else ""
+    Y = "\033[0;33m" if _TTY else ""
+    G = "\033[0;32m" if _TTY else ""
+    D = "\033[0;90m" if _TTY else ""
+    RST = "\033[0m" if _TTY else ""
 
     def color_pct(pct: int) -> str:
         c = R if pct >= 70 else Y if pct >= 50 else G

{codex_cli_usage-0.1.0.dist-info → codex_cli_usage-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: codex-cli-usage
-Version: 0.1.0
+Version: 0.1.3
 Summary: Codex CLI usage monitor — fetches rate limits from OpenAI's ChatGPT backend API
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown

codex_cli_usage-0.1.3.dist-info/RECORD

@@ -0,0 +1,5 @@
+codex_cli_usage/__init__.py,sha256=FTedVfawFhSWGp-GRudHqf2t2Jt1GdBxqWwPgN2abrI,14357
+codex_cli_usage-0.1.3.dist-info/METADATA,sha256=UH_p08w2DFf37FNyX_nOfrmJYWnzm8YnGH6KUpaj-pI,2380
+codex_cli_usage-0.1.3.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+codex_cli_usage-0.1.3.dist-info/entry_points.txt,sha256=P4fiVVq76DXzfc_gXLICI7J4-xwdBqeZoQjlUxq2d4g,57
+codex_cli_usage-0.1.3.dist-info/RECORD,,

codex_cli_usage-0.1.0.dist-info/RECORD

@@ -1,5 +0,0 @@
-codex_cli_usage/__init__.py,sha256=CQi40y1Tabm0MjG4An5d79i_TtwfOOsWsTE-sVZrOhI,12884
-codex_cli_usage-0.1.0.dist-info/METADATA,sha256=sdh4xKIxg0sOeh_wyolepl_lOOMWlfleTbSy3R_5n4M,2380
-codex_cli_usage-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-codex_cli_usage-0.1.0.dist-info/entry_points.txt,sha256=P4fiVVq76DXzfc_gXLICI7J4-xwdBqeZoQjlUxq2d4g,57
-codex_cli_usage-0.1.0.dist-info/RECORD,,