tango-python 0.4.3__py3-none-any.whl → 0.5.0__py3-none-any.whl

tango/__init__.py

@@ -10,6 +10,7 @@ from .exceptions import (
 )
 from .models import (
     GsaElibraryContract,
+    ITDashboardInvestment,
     PaginatedResponse,
     RateLimitInfo,
     SearchFilters,
@@ -28,7 +29,7 @@ from .shapes import (
     TypeGenerator,
 )
 
-__version__ = "0.4.3"
+__version__ = "0.5.0"
 __all__ = [
     "TangoClient",
     "TangoAPIError",
@@ -38,6 +39,7 @@ __all__ = [
     "TangoRateLimitError",
     "RateLimitInfo",
     "GsaElibraryContract",
+    "ITDashboardInvestment",
     "PaginatedResponse",
     "SearchFilters",
     "ShapeConfig",

tango/client.py

@@ -26,6 +26,7 @@ from tango.models import (
     Forecast,
     Grant,
     GsaElibraryContract,
+    ITDashboardInvestment,
     Location,
     Notice,
     Opportunity,
@@ -59,6 +60,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
@@ -67,6 +70,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")
@@ -76,9 +81,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
@@ -105,6 +115,11 @@ class TangoClient:
         """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."""
@@ -140,6 +155,7 @@ 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:
@@ -1321,6 +1337,112 @@ class TangoClient:
             data, shape, GsaElibraryContract, flat, flat_lists, joiner=joiner
         )
 
+    # ============================================================================
+    # IT Dashboard Investments
+    # ============================================================================
+
+    def list_itdashboard_investments(
+        self,
+        page: int = 1,
+        limit: int = 25,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+        search: str | None = None,
+        agency_code: int | None = None,
+        agency_name: str | None = None,
+        type_of_investment: str | None = None,
+        updated_time_after: str | date | datetime | None = None,
+        updated_time_before: str | date | datetime | None = None,
+        cio_rating: int | None = None,
+        cio_rating_max: int | None = None,
+        performance_risk: bool | None = None,
+    ) -> PaginatedResponse:
+        """List federal IT investments from the IT Dashboard (`/api/itdashboard/`).
+
+        Filters are tier-gated by the API:
+
+        - **Free**: ``search`` (full-text across UII, title, description, agency, bureau)
+        - **Pro**: ``agency_code``, ``type_of_investment``,
+          ``updated_time_after`` / ``updated_time_before``
+        - **Business+**: ``agency_name`` (text), ``cio_rating``,
+          ``cio_rating_max``, ``performance_risk``
+
+        Hitting a gated filter on a lower tier returns a 403 with upgrade info.
+
+        CIO ratings: 1=High Risk, 2=Moderately High, 3=Medium, 4=Moderately Low, 5=Low.
+        ``performance_risk=True`` returns investments with at least one NOT MET metric.
+        """
+        params: dict[str, Any] = {"page": page, "limit": min(limit, 100)}
+        if shape is None:
+            shape = ShapeConfig.ITDASHBOARD_INVESTMENTS_MINIMAL
+        if shape:
+            params["shape"] = shape
+            if flat:
+                params["flat"] = "true"
+                if joiner:
+                    params["joiner"] = joiner
+            if flat_lists:
+                params["flat_lists"] = "true"
+        for k, val in (
+            ("search", search),
+            ("agency_code", agency_code),
+            ("agency_name", agency_name),
+            ("type_of_investment", type_of_investment),
+            ("updated_time_after", updated_time_after),
+            ("updated_time_before", updated_time_before),
+            ("cio_rating", cio_rating),
+            ("cio_rating_max", cio_rating_max),
+            ("performance_risk", performance_risk),
+        ):
+            if val is None:
+                continue
+            if isinstance(val, bool):
+                params[k] = "true" if val else "false"
+            elif isinstance(val, (date, datetime)):
+                params[k] = val.isoformat()
+            else:
+                params[k] = val
+        data = self._get("/api/itdashboard/", params)
+        results = [
+            self._parse_response_with_shape(
+                obj, shape, ITDashboardInvestment, flat, flat_lists, joiner=joiner
+            )
+            for obj in data.get("results", [])
+        ]
+        return PaginatedResponse(
+            count=data.get("count", 0),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=results,
+        )
+
+    def get_itdashboard_investment(
+        self,
+        uii: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single IT Dashboard investment by UII (`/api/itdashboard/{uii}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.ITDASHBOARD_INVESTMENTS_COMPREHENSIVE
+        if shape:
+            params["shape"] = shape
+            if flat:
+                params["flat"] = "true"
+                if joiner:
+                    params["joiner"] = joiner
+            if flat_lists:
+                params["flat_lists"] = "true"
+        data = self._get(f"/api/itdashboard/{uii}/", params)
+        return self._parse_response_with_shape(
+            data, shape, ITDashboardInvestment, flat, flat_lists, joiner=joiner
+        )
+
     # ============================================================================
     # Vehicles (Awards)
     # ============================================================================

tango/models.py

@@ -367,6 +367,45 @@ class GsaElibraryContract:
     sins: list[str] | None = None
 
 
+@dataclass
+class ITDashboardInvestment:
+    """Schema definition for IT Dashboard Investment (not used for instances)
+
+    Federal IT investment from itdashboard.gov, exposed at /api/itdashboard/.
+    Identified by ``uii`` (Unique Investment Identifier).
+
+    Tier-gated shape expansions:
+        Free        base fields only
+        Pro+        ``funding`` and ``details`` expansions
+        Business+   nested sub-tables (``cio_evaluation``, ``contracts``,
+                    ``projects``, ``cost_pools_towers``, ``funding_sources``,
+                    ``performance_metrics``, ``performance_actual``,
+                    ``operational_analysis``) and ``business_case_html``
+    """
+
+    uii: str
+    agency_code: int | None = None
+    agency_name: str | None = None
+    bureau_code: int | None = None
+    bureau_name: str | None = None
+    investment_title: str | None = None
+    type_of_investment: str | None = None
+    part_of_it_portfolio: str | None = None
+    updated_time: datetime | None = None
+    url: str | None = None
+    business_case_html: str | None = None
+    funding: dict[str, Any] | None = None
+    details: dict[str, Any] | None = None
+    cio_evaluation: list[dict[str, Any]] | None = None
+    contracts: list[dict[str, Any]] | None = None
+    projects: list[dict[str, Any]] | None = None
+    cost_pools_towers: list[dict[str, Any]] | None = None
+    funding_sources: list[dict[str, Any]] | None = None
+    performance_metrics: list[dict[str, Any]] | None = None
+    performance_actual: list[dict[str, Any]] | None = None
+    operational_analysis: list[dict[str, Any]] | None = None
+
+
 @dataclass
 class Vehicle:
     """Schema definition for Vehicle (not used for instances)"""
@@ -687,3 +726,18 @@ class ShapeConfig:
     GSA_ELIBRARY_CONTRACTS_MINIMAL: Final = (
         "uuid,contract_number,schedule,recipient(display_name,uei),idv(key,award_date)"
     )
+
+    # Default for list_itdashboard_investments()
+    # Free-tier safe: matches the API's INVESTMENT_LIST_DEFAULT_SHAPE.
+    ITDASHBOARD_INVESTMENTS_MINIMAL: Final = (
+        "uii,agency_name,bureau_name,investment_title,"
+        "type_of_investment,part_of_it_portfolio,updated_time,url"
+    )
+
+    # Default for get_itdashboard_investment()
+    # Free-tier safe: matches the API's INVESTMENT_RETRIEVE_DEFAULT_SHAPE.
+    ITDASHBOARD_INVESTMENTS_COMPREHENSIVE: Final = (
+        "uii,agency_code,agency_name,bureau_code,bureau_name,"
+        "investment_title,type_of_investment,part_of_it_portfolio,"
+        "updated_time,url"
+    )

tango/shapes/explicit_schemas.py

@@ -1132,6 +1132,67 @@ GSA_ELIBRARY_CONTRACT_SCHEMA: dict[str, FieldSchema] = {
     ),
 }
 
+# IT Dashboard Investment
+ITDASHBOARD_INVESTMENT_SCHEMA: dict[str, FieldSchema] = {
+    "uii": FieldSchema(name="uii", type=str, is_optional=False, is_list=False),
+    "agency_code": FieldSchema(
+        name="agency_code", type=int, is_optional=True, is_list=False
+    ),
+    "agency_name": FieldSchema(
+        name="agency_name", type=str, is_optional=True, is_list=False
+    ),
+    "bureau_code": FieldSchema(
+        name="bureau_code", type=int, is_optional=True, is_list=False
+    ),
+    "bureau_name": FieldSchema(
+        name="bureau_name", type=str, is_optional=True, is_list=False
+    ),
+    "investment_title": FieldSchema(
+        name="investment_title", type=str, is_optional=True, is_list=False
+    ),
+    "type_of_investment": FieldSchema(
+        name="type_of_investment", type=str, is_optional=True, is_list=False
+    ),
+    "part_of_it_portfolio": FieldSchema(
+        name="part_of_it_portfolio", type=str, is_optional=True, is_list=False
+    ),
+    "updated_time": FieldSchema(
+        name="updated_time", type=datetime, is_optional=True, is_list=False
+    ),
+    "url": FieldSchema(name="url", type=str, is_optional=True, is_list=False),
+    "business_case_html": FieldSchema(
+        name="business_case_html", type=str, is_optional=True, is_list=False
+    ),
+    # Expansions: dict (funding/details) and list-of-dict (nested sub-tables).
+    # Modeled as opaque dict/list since their inner shapes are dynamic.
+    "funding": FieldSchema(name="funding", type=dict, is_optional=True, is_list=False),
+    "details": FieldSchema(name="details", type=dict, is_optional=True, is_list=False),
+    "cio_evaluation": FieldSchema(
+        name="cio_evaluation", type=list, is_optional=True, is_list=True
+    ),
+    "contracts": FieldSchema(
+        name="contracts", type=list, is_optional=True, is_list=True
+    ),
+    "projects": FieldSchema(
+        name="projects", type=list, is_optional=True, is_list=True
+    ),
+    "cost_pools_towers": FieldSchema(
+        name="cost_pools_towers", type=list, is_optional=True, is_list=True
+    ),
+    "funding_sources": FieldSchema(
+        name="funding_sources", type=list, is_optional=True, is_list=True
+    ),
+    "performance_metrics": FieldSchema(
+        name="performance_metrics", type=list, is_optional=True, is_list=True
+    ),
+    "performance_actual": FieldSchema(
+        name="performance_actual", type=list, is_optional=True, is_list=True
+    ),
+    "operational_analysis": FieldSchema(
+        name="operational_analysis", type=list, is_optional=True, is_list=True
+    ),
+}
+
 # ============================================================================
 # SCHEMA REGISTRY MAPPING
 # ============================================================================
@@ -1176,6 +1237,8 @@ EXPLICIT_SCHEMAS: dict[str, dict[str, FieldSchema]] = {
     # GSA eLibrary
     "GsaElibraryContract": GSA_ELIBRARY_CONTRACT_SCHEMA,
     "GsaElibraryIdvRef": GSA_ELIBRARY_IDV_REF_SCHEMA,
+    # IT Dashboard
+    "ITDashboardInvestment": ITDASHBOARD_INVESTMENT_SCHEMA,
 }
 
 

{tango_python-0.4.3.dist-info → tango_python-0.5.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tango-python
-Version: 0.4.3
+Version: 0.5.0
 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.3.dist-info → tango_python-0.5.0.dist-info}/RECORD

@@ -1,16 +1,16 @@
-tango/__init__.py,sha256=MkZ4VcgIB5WFfCaxK6XXrK25icagQ2o9ZNta64xsqgw,1142
-tango/client.py,sha256=kzUVPPOur45QobKBrRzPXGOOYZHXg-cBEWlSIB8vvAo,87725
+tango/__init__.py,sha256=jhfw7xJNlVM9N2MjC42_i-mGFMgvKlKOZuC-4-d8z_I,1198
+tango/client.py,sha256=Pr9lYMj5diZ2ecyrHL88y2B_WF41tYZ9Qm2S2xT3NPU,92545
 tango/exceptions.py,sha256=aRvDm0dUCEtNDfRVYCX7SEDdd1WlIVVY6sN78Tzo-a0,3114
-tango/models.py,sha256=EDKsZ4fsxkAbDhX5roOfiKYyPjZaTV-5QKrPaQCsb0Y,20959
+tango/models.py,sha256=fe8SDB1n8sG4DgVnCYnpkrWpY6Kz4CE475lh8MCCJfs,23207
 tango/shapes/__init__.py,sha256=7ea1WU74jp4znhNw-gXruag6m6eyPZtbVgbDFmFUWro,1072
-tango/shapes/explicit_schemas.py,sha256=H4pYs0LCTSV5msRCxftmgiM_-3sc4LsqpDPgj36DkPY,55202
+tango/shapes/explicit_schemas.py,sha256=_99ywtXRQoJ6KxesVTzWjOWMzxE1h9f2YGx19kW0FnA,57834
 tango/shapes/factory.py,sha256=ytpMi5Uw72XZ8MimhuSsLDVXF3zO_Zt3_tAL6NF7LnU,34318
 tango/shapes/generator.py,sha256=61V1T3lm8Ps_KSMJAezQJLQVFbNKt1jtoLyhiqNtFTs,23380
 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.3.dist-info/METADATA,sha256=OhV1f_Hifu1hUd2c0zfGc91rIIe07ddJ3Nw3MDnqFl8,17595
-tango_python-0.4.3.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-tango_python-0.4.3.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
-tango_python-0.4.3.dist-info/RECORD,,
+tango_python-0.5.0.dist-info/METADATA,sha256=Ht7PpwMjaP7R4yX-tWAvdvzsRIzlfFu5ZPwJybmH04s,17595
+tango_python-0.5.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+tango_python-0.5.0.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
+tango_python-0.5.0.dist-info/RECORD,,