usage-cli 0.34.3__tar.gz → 0.34.5__tar.gz

{usage_cli-0.34.3/usage_cli.egg-info → usage_cli-0.34.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usage-cli
-Version: 0.34.3
+Version: 0.34.5
 Summary: usage — 在 macOS menu bar 與 Windows 系統匣顯示 Claude Code、Codex、Antigravity 用量的小工具（也提供終端機 TUI）
 License-Expression: AGPL-3.0-only
 Requires-Python: >=3.13

{usage_cli-0.34.3 → usage_cli-0.34.5}/loaders/muse_quota_probe.py

@@ -10,12 +10,14 @@ import json
 import logging
 import math
 import os
+import re
 import shutil
 import subprocess
 import tempfile
 import threading
 import time
 from contextlib import suppress
+from datetime import datetime
 from pathlib import Path
 from typing import TypedDict, TypeGuard
 
@@ -27,6 +29,9 @@ CACHE_PATH = Path.home() / ".usage/muse_quota.json"
 PROBE_DIR = Path.home() / ".usage/muse-probe"
 MUSE_MODEL = "muse-spark-1.3"
 STALE_SECONDS = 30 * 60
+_EXHAUSTED = re.compile(
+    r"Subscription quota exhausted\. Your usage window resets at (\d{4}-\d\d-\d\dT[\d:.]+Z)"
+)
 _probe_lock = threading.Lock()
 _probe_running = False
 _retry_after = 0.0
@@ -76,7 +81,8 @@ def validate_muse_cache(payload: object) -> MuseQuotaResult | None:
         if not isinstance(window, dict):
             return None
         used, reset = window.get("used_percent"), window.get("resets_at")
-        if not _number(used) or not _number(reset) or not 0 <= used <= 100:
+        # An exhausted window reports past 100 (e.g. 104); the row clamps it for display.
+        if not _number(used) or not _number(reset) or used < 0:
             return None
         windows.append(MuseQuotaWindow(used_percent=float(used), resets_at=float(reset)))
     return MuseQuotaResult(
@@ -108,6 +114,25 @@ def parse_muse_allowance(stdout: str, checked_at: float) -> MuseQuotaResult | No
     return None
 
 
+def parse_exhausted_window(stdout: str, cached: MuseQuotaResult | None) -> MuseQuotaResult | None:
+    """Read the 429 Muse returns once the 5-hour window is used up; it carries no allowance.
+
+    Only the window's reset time is known, so the weekly reading and its age stay from ``cached``.
+    """
+    match = _EXHAUSTED.search(stdout)
+    if cached is None or match is None:
+        return None
+    try:
+        reset = datetime.fromisoformat(match[1]).timestamp()
+    except ValueError:
+        return None
+    return MuseQuotaResult(
+        checked_at=cached["checked_at"],
+        window=MuseQuotaWindow(used_percent=100.0, resets_at=reset),
+        weekly=cached["weekly"],
+    )
+
+
 def read_cache() -> MuseQuotaResult | None:
     """Ignore invalid shared snapshots without deleting or quarantining them."""
     try:
@@ -213,7 +238,10 @@ def _probe_worker(executable: str) -> None:
             process.kill()
             process.communicate()
             raise
-        result = parse_muse_allowance(stdout, time.time()) if process.returncode == 0 else None
+        if process.returncode == 0:
+            result = parse_muse_allowance(stdout, time.time())
+        else:
+            result = parse_exhausted_window(stdout, read_cache())
         if result is not None:
             _write_cache(result)
             succeeded = True

{usage_cli-0.34.3 → usage_cli-0.34.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "usage-cli"
-version = "0.34.3"
+version = "0.34.5"
 description = "usage — 在 macOS menu bar 與 Windows 系統匣顯示 Claude Code、Codex、Antigravity 用量的小工具（也提供終端機 TUI）"
 readme = "README.md"
 requires-python = ">=3.13"

{usage_cli-0.34.3 → usage_cli-0.34.5}/tests/test_muse_quota_probe.py

@@ -60,13 +60,51 @@ def test_read_missing_corrupt_and_good_cache() -> None:
     assert probe.read_cache() == quota()
 
 
-@pytest.mark.parametrize("value", [True, None, "25", -1, 101, float("nan"), float("inf"), 10**400])
+@pytest.mark.parametrize("value", [True, None, "25", -1, float("nan"), float("inf"), 10**400])
 def test_invalid_percent_rejected(value: object) -> None:
     raw = event()
     raw["payload"]["allowance"]["window"]["used_percent"] = value
     assert probe.parse_muse_allowance(json.dumps(raw), 1000) is None
 
 
+@pytest.mark.parametrize("value", [100, 104, 105, 250])
+def test_exhausted_window_past_100_is_kept(value: int) -> None:
+    raw = event()
+    raw["payload"]["allowance"]["window"]["used_percent"] = value
+    result = probe.parse_muse_allowance(json.dumps(raw), 1000)
+    assert result is not None and result["window"]["used_percent"] == value
+
+
+# Captured from Muse Code 1.4.11 on 2026-10-21 while the 5-hour window was used up.
+EXHAUSTED = json.dumps(
+    {
+        "payload_type": "run.terminal.failed",
+        "payload": {
+            "reason": "API error 429 [request_id=x]: Subscription quota exhausted. "
+            "Your usage window resets at 2026-10-21T08:28:56Z. (rate_limit_error)"
+        },
+    }
+)
+EXHAUSTED_QUOTA = {**quota(), "window": {"used_percent": 100.0, "resets_at": 1792571336.0}}
+
+
+def test_exhausted_429_keeps_weekly_and_snapshot_age() -> None:
+    assert probe.parse_exhausted_window(EXHAUSTED, quota()) == EXHAUSTED_QUOTA
+    assert probe.parse_exhausted_window(EXHAUSTED, None) is None
+    assert probe.parse_exhausted_window(json.dumps(event()), quota()) is None
+
+
+def test_exhausted_probe_updates_session_window(monkeypatch: pytest.MonkeyPatch) -> None:
+    probe.CACHE_PATH.write_text(json.dumps(quota()))
+    process = Mock(returncode=1)
+    process.poll.return_value = 1
+    process.communicate.return_value = (EXHAUSTED, "")
+    monkeypatch.setattr(subprocess, "Popen", Mock(return_value=process))
+    probe._probe_worker("/fake/muse")
+    assert probe.read_cache() == EXHAUSTED_QUOTA
+    assert probe._retry_after == 0.0
+
+
 def test_missing_weekly_is_rejected() -> None:
     raw = event()
     del raw["payload"]["allowance"]["weekly"]

{usage_cli-0.34.3 → usage_cli-0.34.5/usage_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: usage-cli
-Version: 0.34.3
+Version: 0.34.5
 Summary: usage — 在 macOS menu bar 與 Windows 系統匣顯示 Claude Code、Codex、Antigravity 用量的小工具（也提供終端機 TUI）
 License-Expression: AGPL-3.0-only
 Requires-Python: >=3.13