tango-python 0.4.2__py3-none-any.whl → 0.4.4__py3-none-any.whl

tango/__init__.py

@@ -11,6 +11,7 @@ from .exceptions import (
 from .models import (
     GsaElibraryContract,
     PaginatedResponse,
+    RateLimitInfo,
     SearchFilters,
     ShapeConfig,
     WebhookEndpoint,
@@ -27,7 +28,7 @@ from .shapes import (
     TypeGenerator,
 )
 
-__version__ = "0.4.1"
+__version__ = "0.4.3"
 __all__ = [
     "TangoClient",
     "TangoAPIError",
@@ -35,6 +36,7 @@ __all__ = [
     "TangoNotFoundError",
     "TangoValidationError",
     "TangoRateLimitError",
+    "RateLimitInfo",
     "GsaElibraryContract",
     "PaginatedResponse",
     "SearchFilters",

tango/client.py

@@ -32,6 +32,7 @@ from tango.models import (
     Organization,
     PaginatedResponse,
     Protest,
+    RateLimitInfo,
     SearchFilters,
     ShapeConfig,
     Subaward,
@@ -58,6 +59,8 @@ class TangoClient:
         self,
         api_key: str | None = None,
         base_url: str = "https://tango.makegov.com",
+        user_agent: str | None = None,
+        extra_headers: dict[str, str] | None = None,
     ):
         """
         Initialize the Tango API client
@@ -66,6 +69,8 @@ class TangoClient:
             api_key: API key for authentication. If not provided, will attempt to load from
                     TANGO_API_KEY environment variable.
             base_url: Base URL for the API
+            user_agent: Custom User-Agent header value.
+            extra_headers: Additional headers to include in every request.
         """
         # Load API key from environment if not provided
         self.api_key = api_key or os.getenv("TANGO_API_KEY")
@@ -75,8 +80,14 @@ class TangoClient:
         headers = {}
         if self.api_key:
             headers["X-API-KEY"] = self.api_key
+        if user_agent:
+            headers["User-Agent"] = user_agent
+        if extra_headers:
+            headers.update(extra_headers)
 
         self.client = httpx.Client(headers=headers, timeout=30.0)
+        self._last_rate_limit_info: RateLimitInfo | None = None
+        self._last_response_headers: httpx.Headers | None = None
 
         # Use hardcoded sensible defaults
         cache_size = 100
@@ -98,6 +109,39 @@ class TangoClient:
     # Core HTTP Request Utilities
     # ============================================================================
 
+    @property
+    def rate_limit_info(self) -> RateLimitInfo | None:
+        """Rate limit info from the most recent API response."""
+        return self._last_rate_limit_info
+
+    @property
+    def last_response_headers(self) -> httpx.Headers | None:
+        """Full HTTP headers from the most recent API response."""
+        return self._last_response_headers
+
+    @staticmethod
+    def _parse_rate_limit_headers(headers: httpx.Headers) -> RateLimitInfo:
+        """Extract rate limit info from response headers."""
+        def _int_or_none(val: str | None) -> int | None:
+            if val is None:
+                return None
+            try:
+                return int(val)
+            except (ValueError, TypeError):
+                return None
+
+        return RateLimitInfo(
+            limit=_int_or_none(headers.get("X-RateLimit-Limit")),
+            remaining=_int_or_none(headers.get("X-RateLimit-Remaining")),
+            reset=_int_or_none(headers.get("X-RateLimit-Reset")),
+            daily_limit=_int_or_none(headers.get("X-RateLimit-Daily-Limit")),
+            daily_remaining=_int_or_none(headers.get("X-RateLimit-Daily-Remaining")),
+            daily_reset=_int_or_none(headers.get("X-RateLimit-Daily-Reset")),
+            burst_limit=_int_or_none(headers.get("X-RateLimit-Burst-Limit")),
+            burst_remaining=_int_or_none(headers.get("X-RateLimit-Burst-Remaining")),
+            burst_reset=_int_or_none(headers.get("X-RateLimit-Burst-Reset")),
+        )
+
     def _request(
         self,
         method: str,
@@ -110,6 +154,8 @@ class TangoClient:
 
         try:
             response = self.client.request(method=method, url=url, params=params, json=json_data)
+            self._last_response_headers = response.headers
+            self._last_rate_limit_info = self._parse_rate_limit_headers(response.headers)
 
             if response.status_code == 401:
                 raise TangoAuthError(
@@ -136,7 +182,9 @@ class TangoClient:
                     error_data,
                 )
             elif response.status_code == 429:
-                raise TangoRateLimitError("Rate limit exceeded", response.status_code)
+                error_data = response.json() if response.content else {}
+                detail = error_data.get("detail", "Rate limit exceeded")
+                raise TangoRateLimitError(detail, response.status_code, error_data)
             elif not response.is_success:
                 raise TangoAPIError(
                     f"API request failed with status {response.status_code}", response.status_code

tango/exceptions.py

@@ -39,7 +39,34 @@ class TangoValidationError(TangoAPIError):
 class TangoRateLimitError(TangoAPIError):
     """Rate limit exceeded error"""
 
-    pass
+    @property
+    def wait_in_seconds(self) -> int | None:
+        """Seconds to wait before retrying, from API response."""
+        val = self.response_data.get("wait_in_seconds")
+        if val is not None:
+            try:
+                return int(val)
+            except (ValueError, TypeError):
+                return None
+        return None
+
+    @property
+    def detail(self) -> str | None:
+        """Human-readable detail from API response."""
+        return self.response_data.get("detail")
+
+    @property
+    def limit_type(self) -> str | None:
+        """Which limit was hit: 'burst' or 'daily', parsed from detail."""
+        d = self.detail
+        if not d:
+            return None
+        lower = d.lower()
+        if "burst" in lower or "minute" in lower:
+            return "burst"
+        if "daily" in lower or "day" in lower:
+            return "daily"
+        return None
 
 
 class ShapeError(TangoAPIError):

tango/models.py

@@ -23,6 +23,21 @@ T = TypeVar("T")
 # ============================================================================
 
 
+@dataclass
+class RateLimitInfo:
+    """Rate limit information from API response headers."""
+
+    limit: int | None = None
+    remaining: int | None = None
+    reset: int | None = None
+    daily_limit: int | None = None
+    daily_remaining: int | None = None
+    daily_reset: int | None = None
+    burst_limit: int | None = None
+    burst_remaining: int | None = None
+    burst_reset: int | None = None
+
+
 @dataclass
 class SearchFilters:
     """Search filter parameters for contract search

{tango_python-0.4.2.dist-info → tango_python-0.4.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tango-python
-Version: 0.4.2
+Version: 0.4.4
 Summary: Python SDK for the Tango API
 Project-URL: Homepage, https://github.com/makegov/tango-python
 Project-URL: Documentation, https://docs.makegov.com/tango-python

{tango_python-0.4.2.dist-info → tango_python-0.4.4.dist-info}/RECORD

@@ -1,7 +1,7 @@
-tango/__init__.py,sha256=7EdEUgCcK0lDNJWWDgdSWxx8xxryO-fHL-iml9PaYLw,1102
-tango/client.py,sha256=HvS-ss9QOeajLOkP2IsX9gTF_KvOb9_NOlLUphd8gSc,86111
-tango/exceptions.py,sha256=JmtbOY0ofBnX24pUErh2XFlTj9dim2ngyboserEGRFw,2226
-tango/models.py,sha256=4sLmWjyqsCU7-b8lr9ZRRG_ElBKz-5txibcOFcScEDY,20555
+tango/__init__.py,sha256=MkZ4VcgIB5WFfCaxK6XXrK25icagQ2o9ZNta64xsqgw,1142
+tango/client.py,sha256=Dm8v3xMO3uX-GGM1ioLySzjLavPmlxhkrqZw4q6wTWw,88396
+tango/exceptions.py,sha256=aRvDm0dUCEtNDfRVYCX7SEDdd1WlIVVY6sN78Tzo-a0,3114
+tango/models.py,sha256=EDKsZ4fsxkAbDhX5roOfiKYyPjZaTV-5QKrPaQCsb0Y,20959
 tango/shapes/__init__.py,sha256=7ea1WU74jp4znhNw-gXruag6m6eyPZtbVgbDFmFUWro,1072
 tango/shapes/explicit_schemas.py,sha256=H4pYs0LCTSV5msRCxftmgiM_-3sc4LsqpDPgj36DkPY,55202
 tango/shapes/factory.py,sha256=ytpMi5Uw72XZ8MimhuSsLDVXF3zO_Zt3_tAL6NF7LnU,34318
@@ -10,7 +10,7 @@ tango/shapes/models.py,sha256=h3pIhOqrrdlN953Y6r0oney5HFbKPOD-frRndRWimJ0,3018
 tango/shapes/parser.py,sha256=k6OsI2w3GH6-IBbc-XTLgL1mWH7bMf7A_dA6pr1xKfw,24619
 tango/shapes/schema.py,sha256=VRPOB1sBdjFyimNchrZKIpTHn83CyX4RfU9077aQtIU,14136
 tango/shapes/types.py,sha256=27jrAE0VIdrKaLjR_FK71hfIIGX2Tg3ex7REEBV1TFE,1301
-tango_python-0.4.2.dist-info/METADATA,sha256=h3QpwadNjsoHnMuXXpFj6BScFCvx61NGGOM43gG6GLQ,17595
-tango_python-0.4.2.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-tango_python-0.4.2.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
-tango_python-0.4.2.dist-info/RECORD,,
+tango_python-0.4.4.dist-info/METADATA,sha256=QGTVP5M9kTipieVrII_ocUU061ukbG61bgDfscHIpgg,17595
+tango_python-0.4.4.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+tango_python-0.4.4.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
+tango_python-0.4.4.dist-info/RECORD,,