codex-cli-usage 0.1.9__py3-none-any.whl → 0.1.10__py3-none-any.whl

codex_cli_usage/__init__.py

@@ -16,6 +16,8 @@ Usage:
 
 import argparse
 import json
+import math
+import os
 import queue
 import shutil
 import signal
@@ -66,8 +68,15 @@ def _find_codex_path(filename: str) -> Path:
     return native
 
 
+def _find_codex_auth_path() -> Path:
+    codex_home = os.environ.get("CODEX_HOME")
+    if codex_home:
+        return Path(codex_home) / "auth.json"
+    return _find_codex_path("auth.json")
+
+
 CODEX_DIR = Path.home() / ".codex"
-AUTH_FILE = _find_codex_path("auth.json")
+AUTH_FILE = _find_codex_auth_path()
 USAGE_FILE = _find_codex_path("usage-limits.json")
 DAEMON_INTERVAL = 300  # 5 minutes
 USAGE_URL = "https://chatgpt.com/backend-api/codex/usage"
@@ -323,7 +332,7 @@ def _fetch_usage_direct() -> dict:
     """
     auth = get_auth()
     if not auth:
-        raise RuntimeError("No auth at ~/.codex/auth.json — run `codex` first")
+        raise RuntimeError(f"No auth at {AUTH_FILE}. Run `codex` first")
 
     tokens = auth.get("tokens", {})
     access_token = tokens.get("access_token")
@@ -381,13 +390,36 @@ def classify_window(seconds: int | float | None, kind: str) -> tuple[str, str]:
 
 
 def _build_window(kind: str, window: dict) -> dict:
+    used_percent = window.get("used_percent")
     seconds = window.get("limit_window_seconds")
-    key, label = classify_window(seconds, kind)
     reset_at = window.get("reset_at")
+    for value in (used_percent, seconds, reset_at):
+        if value is not None and (
+            not isinstance(value, int | float)
+            or isinstance(value, bool)
+            or (isinstance(value, float) and not math.isfinite(value))
+        ):
+            raise UsageServiceError("Invalid rate limit response")
+    if seconds is not None and seconds <= 0:
+        raise UsageServiceError("Invalid rate limit response")
+
+    try:
+        key, label = classify_window(seconds, kind)
+    except OverflowError:
+        raise UsageServiceError("Invalid rate limit response") from None
+    try:
+        resets_at = (
+            datetime.fromtimestamp(reset_at, tz=timezone.utc).isoformat()
+            if reset_at is not None
+            else None
+        )
+    except (OSError, OverflowError, ValueError):
+        raise UsageServiceError("Invalid rate limit response") from None
+
     return {
         "kind": kind,
-        "pct": window.get("used_percent"),
-        "resets_at": datetime.fromtimestamp(reset_at, tz=timezone.utc).isoformat() if reset_at else None,
+        "pct": used_percent,
+        "resets_at": resets_at,
         "window_secs": seconds,
         "key": key,
         "label": label,
@@ -442,11 +474,13 @@ def build_usage_json(api_data: dict) -> dict:
     # Code review limits
     cr = api_data.get("code_review_rate_limit") or {}
     cr_primary = cr.get("primary_window")
-    if cr_primary and cr_primary.get("used_percent", 0) > 0:
-        result["code_review"] = {
-            "pct": cr_primary["used_percent"],
-            "resets_at": datetime.fromtimestamp(cr_primary["reset_at"], tz=timezone.utc).isoformat() if cr_primary.get("reset_at") else None,
-        }
+    if isinstance(cr_primary, dict):
+        window = _build_window("code_review", cr_primary)
+        if window["pct"] is not None and window["pct"] > 0:
+            result["code_review"] = {
+                "pct": window["pct"],
+                "resets_at": window["resets_at"],
+            }
 
     # Credits
     credits = api_data.get("credits") or {}

{codex_cli_usage-0.1.9.dist-info → codex_cli_usage-0.1.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: codex-cli-usage
-Version: 0.1.9
+Version: 0.1.10
 Summary: Codex CLI usage monitor — fetches rate limits through the Codex app-server
 Project-URL: Homepage, https://github.com/wakamex/codex-cli-usage
 Project-URL: Source, https://github.com/wakamex/codex-cli-usage
@@ -100,7 +100,10 @@ window actually classifies as 5-hour or weekly. Readers should prefer
 
 ### Authentication
 
-The OAuth tokens live at `~/.codex/auth.json`, written by the Codex CLI on login.
+The OAuth tokens live at `~/.codex/auth.json` by default and are written by the
+Codex CLI on login. When `CODEX_HOME` is set, the direct compatibility fallback
+reads `$CODEX_HOME/auth.json`. The preferred app-server path leaves credential
+resolution to Codex.
 
 Codex owns automatic token refresh and credential persistence. The compatibility
 fallback reads `auth.json` but never modifies credentials.

codex_cli_usage-0.1.10.dist-info/RECORD

@@ -0,0 +1,6 @@
+codex_cli_usage/__init__.py,sha256=I24UubGF-NXScacl-hRXOM1Is7lXmHtExIjHJ8vZP2k,23935
+codex_cli_usage/__main__.py,sha256=V_ll4j4TO5TIww8Uh9mQZwAGMMvb66-9JSXuTkYbr3A,59
+codex_cli_usage-0.1.10.dist-info/METADATA,sha256=1wY8qw8fG0MIigjAnT1F5ywYE27IoUQS-bratXXbbC8,3634
+codex_cli_usage-0.1.10.dist-info/WHEEL,sha256=zOwg4jB6zX2kU910N-cMawjivD6tO8NEWvE12je1bVk,87
+codex_cli_usage-0.1.10.dist-info/entry_points.txt,sha256=P4fiVVq76DXzfc_gXLICI7J4-xwdBqeZoQjlUxq2d4g,57
+codex_cli_usage-0.1.10.dist-info/RECORD,,

{codex_cli_usage-0.1.9.dist-info → codex_cli_usage-0.1.10.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: hatchling 1.31.0
+Generator: hatchling 1.32.0
 Root-Is-Purelib: true
 Tag: py3-none-any

codex_cli_usage-0.1.9.dist-info/RECORD

@@ -1,6 +0,0 @@
-codex_cli_usage/__init__.py,sha256=FPHWaTVqb2-rVXovkA_iE-iqABvZIpApKDFk3sNxGog,22939
-codex_cli_usage/__main__.py,sha256=V_ll4j4TO5TIww8Uh9mQZwAGMMvb66-9JSXuTkYbr3A,59
-codex_cli_usage-0.1.9.dist-info/METADATA,sha256=eO2PlghUxjqpG2IuP1khkCCQKH6EiKPDDrSoj7p8eZI,3455
-codex_cli_usage-0.1.9.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
-codex_cli_usage-0.1.9.dist-info/entry_points.txt,sha256=P4fiVVq76DXzfc_gXLICI7J4-xwdBqeZoQjlUxq2d4g,57
-codex_cli_usage-0.1.9.dist-info/RECORD,,