tango-python 1.0.0__py3-none-any.whl → 1.1.1__py3-none-any.whl

tango/__init__.py

@@ -9,6 +9,7 @@ from .exceptions import (
     TangoValidationError,
 )
 from .models import (
+    BudgetAccount,
     GsaElibraryContract,
     ITDashboardInvestment,
     PaginatedResponse,
@@ -43,7 +44,7 @@ from .webhooks import (
 )
 from .webhooks.receiver import Delivery, WebhookReceiver
 
-__version__ = "1.0.0"
+__version__ = "1.1.1"
 __all__ = [
     "TangoClient",
     "TangoAPIError",
@@ -54,6 +55,7 @@ __all__ = [
     "RateLimitInfo",
     "ResolveCandidate",
     "ResolveResult",
+    "BudgetAccount",
     "GsaElibraryContract",
     "ITDashboardInvestment",
     "PaginatedResponse",

tango/client.py

@@ -4,7 +4,7 @@ import os
 import warnings
 from datetime import date, datetime
 from decimal import Decimal
-from typing import Any
+from typing import Any, Literal, cast
 from urllib.parse import urljoin
 
 import httpx
@@ -21,6 +21,7 @@ from tango.models import (
     OTA,
     OTIDV,
     Agency,
+    BudgetAccount,
     BusinessType,
     Contract,
     Entity,
@@ -220,9 +221,7 @@ class TangoClient:
         picking one — that ambiguity would hide caller bugs.
         """
         if json_data is not None and json is not None:
-            raise TangoValidationError(
-                "_post: pass `json_data` or `json`, not both."
-            )
+            raise TangoValidationError("_post: pass `json_data` or `json`, not both.")
         body = json_data if json_data is not None else json
         if body is None:
             body = {}
@@ -243,9 +242,7 @@ class TangoClient:
         picking one — that ambiguity would hide caller bugs.
         """
         if json_data is not None and json is not None:
-            raise TangoValidationError(
-                "_patch: pass `json_data` or `json`, not both."
-            )
+            raise TangoValidationError("_patch: pass `json_data` or `json`, not both.")
         body = json_data if json_data is not None else json
         if body is None:
             body = {}
@@ -673,8 +670,9 @@ class TangoClient:
         params: dict[str, Any] = {"limit": min(limit, 100)}
         if cursor:
             params["cursor"] = cursor
-        else:
-            params["page"] = 1
+        # /api/contracts/ is cursor-only (KeysetPagination). When no cursor is
+        # supplied, send neither page nor cursor — the API returns the first
+        # page by default. (Previously sent page=1, which the endpoint ignores.)
 
         # Handle legacy filters parameter (backward compatibility)
         filter_dict: dict[str, Any] = {}
@@ -772,6 +770,89 @@ class TangoClient:
             cursor=data.get("cursor"),
         )
 
+    def get_contract(
+        self,
+        key: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single contract by key (`/api/contracts/{key}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.CONTRACTS_MINIMAL
+        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/contracts/{key}/", params)
+        return self._parse_response_with_shape(
+            data, shape, Contract, flat, flat_lists, joiner=joiner
+        )
+
+    def get_contract_subawards(
+        self,
+        key: str,
+        page: int = 1,
+        limit: int = 25,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        ordering: str | None = None,
+    ) -> PaginatedResponse:
+        """List subawards under a contract (`/api/contracts/{key}/subawards/`)."""
+        params: dict[str, Any] = {"page": page, "limit": min(limit, 100)}
+        if shape is None:
+            shape = ShapeConfig.SUBAWARDS_MINIMAL
+        if shape:
+            params["shape"] = shape
+            if flat:
+                params["flat"] = "true"
+            if flat_lists:
+                params["flat_lists"] = "true"
+        if ordering:
+            params["ordering"] = ordering
+        data = self._get(f"/api/contracts/{key}/subawards/", params)
+        raw_results = data.get("results") or []
+        results = [
+            self._parse_response_with_shape(obj, shape, Subaward, flat, flat_lists)
+            for obj in raw_results
+        ]
+        return PaginatedResponse(
+            count=int(data.get("count") or len(results)),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=results,
+            cursor=data.get("cursor"),
+        )
+
+    def get_contract_transactions(
+        self,
+        key: str,
+        limit: int = 100,
+        cursor: str | None = None,
+        ordering: str | None = None,
+    ) -> PaginatedResponse:
+        """List transactions under a contract (`/api/contracts/{key}/transactions/`)."""
+        params: dict[str, Any] = {"limit": min(limit, 500)}
+        if cursor:
+            params["cursor"] = cursor
+        if ordering:
+            params["ordering"] = ordering
+        data = self._get(f"/api/contracts/{key}/transactions/", params)
+        return PaginatedResponse(
+            count=int(data.get("count") or len(data.get("results") or [])),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=data.get("results") or [],
+            cursor=data.get("cursor"),
+        )
+
     # ============================================================================
     # IDVs (Awards)
     # ============================================================================
@@ -1246,6 +1327,59 @@ class TangoClient:
         data = self._get(f"/api/otidvs/{key}/", params)
         return self._parse_response_with_shape(data, shape, OTIDV, flat, flat_lists, joiner=joiner)
 
+    def list_otidv_awards(
+        self,
+        key: str,
+        limit: int = 25,
+        cursor: str | None = None,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+        ordering: str | None = None,
+    ) -> PaginatedResponse:
+        """List child awards under an OTIDV (`/api/otidvs/{key}/awards/`).
+
+        Args:
+            key: The OTIDV key.
+            limit: Results per page (max 100).
+            cursor: Cursor token for keyset pagination.
+            shape: Response shape string (defaults to minimal contract shape).
+            flat: If True, flatten nested objects using dot notation.
+            flat_lists: If True, flatten arrays using indexed keys.
+            joiner: Separator used when flattening.
+            ordering: Server-side sort (prefix with '-' for descending).
+        """
+        params: dict[str, Any] = {"limit": min(limit, 100)}
+        if cursor:
+            params["cursor"] = cursor
+        if shape is None:
+            shape = ShapeConfig.CONTRACTS_MINIMAL
+        if shape:
+            params["shape"] = shape
+            if flat:
+                params["flat"] = "true"
+                if joiner:
+                    params["joiner"] = joiner
+            if flat_lists:
+                params["flat_lists"] = "true"
+        if ordering:
+            params["ordering"] = ordering
+        data = self._get(f"/api/otidvs/{key}/awards/", params)
+        raw_results = data.get("results") or []
+        results = [
+            self._parse_response_with_shape(obj, shape, Contract, flat, flat_lists, joiner=joiner)
+            for obj in raw_results
+        ]
+        return PaginatedResponse(
+            count=int(data.get("count") or len(results)),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=results,
+            cursor=data.get("cursor"),
+            page_metadata=data.get("page_metadata"),
+        )
+
     def list_subawards(
         self,
         page: int = 1,
@@ -1300,6 +1434,31 @@ class TangoClient:
             results=results,
         )
 
+    def get_subaward(
+        self,
+        key: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single subaward by key (`/api/subawards/{key}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.SUBAWARDS_MINIMAL
+        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/subawards/{key}/", params)
+        return self._parse_response_with_shape(
+            data, shape, Subaward, flat, flat_lists, joiner=joiner
+        )
+
     # ============================================================================
     # GSA eLibrary Contracts
     # ============================================================================
@@ -1922,6 +2081,12 @@ class TangoClient:
         data = self._get(f"/api/entities/{key}/", params)
         return self._parse_response_with_shape(data, shape, Entity, flat, flat_lists)
 
+    def get_entity_budget_flows(self, uei: str) -> dict[str, Any]:
+        """Get budget flows for an entity (`/api/entities/{uei}/budget-flows/`)."""
+        if not uei:
+            raise TangoValidationError("UEI is required")
+        return self._get(f"/api/entities/{uei}/budget-flows/")
+
     # Forecast endpoints
     def list_forecasts(
         self,
@@ -2014,6 +2179,31 @@ class TangoClient:
             results=results,
         )
 
+    def get_forecast(
+        self,
+        id: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single forecast by id (`/api/forecasts/{id}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.FORECASTS_MINIMAL
+        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/forecasts/{id}/", params)
+        return self._parse_response_with_shape(
+            data, shape, Forecast, flat, flat_lists, joiner=joiner
+        )
+
     # Opportunity endpoints
     def list_opportunities(
         self,
@@ -2112,6 +2302,31 @@ class TangoClient:
             results=results,
         )
 
+    def get_opportunity(
+        self,
+        opportunity_id: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single opportunity by id (`/api/opportunities/{opportunity_id}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.OPPORTUNITIES_MINIMAL
+        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/opportunities/{opportunity_id}/", params)
+        return self._parse_response_with_shape(
+            data, shape, Opportunity, flat, flat_lists, joiner=joiner
+        )
+
     # Notice endpoints
     def list_notices(
         self,
@@ -2201,6 +2416,29 @@ class TangoClient:
             results=results,
         )
 
+    def get_notice(
+        self,
+        notice_id: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single notice by id (`/api/notices/{notice_id}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.NOTICES_MINIMAL
+        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/notices/{notice_id}/", params)
+        return self._parse_response_with_shape(data, shape, Notice, flat, flat_lists, joiner=joiner)
+
     # Protest endpoints
     # See https://tango.makegov.com/docs/api-reference/protests.md
     # Note: Protests API does not support ordering (returns 400 if provided).
@@ -2328,6 +2566,170 @@ class TangoClient:
         data = self._get(f"/api/protests/{case_id}/", params)
         return self._parse_response_with_shape(data, shape, Protest, flat, flat_lists)
 
+    # ============================================================================
+    # Budget (federal account x fiscal year rollups)
+    # ============================================================================
+
+    def list_budget_accounts(
+        self,
+        page: int = 1,
+        limit: int = 25,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        federal_account_symbol: str | None = None,
+        fiscal_year: int | None = None,
+        fiscal_year_gte: int | None = None,
+        fiscal_year_lte: int | None = None,
+        agency_code: str | None = None,
+        bureau_name: str | None = None,
+        account_title: str | None = None,
+        bea_category: str | None = None,
+        on_off_budget: str | None = None,
+        subfunction_code: str | None = None,
+        search: str | None = None,
+        ordering: str | None = None,
+    ) -> PaginatedResponse:
+        """List budget accounts (`/api/budget/accounts/`).
+
+        One row per ``(federal_account_symbol, fiscal_year)`` covering the full
+        budget lifecycle, pre-computed ratios + trends, the
+        contract/assistance/unlinked breakdown, and request-vs-actual spend.
+
+        Args:
+            page: Page number.
+            limit: Results per page (max 100).
+            shape: Response shape string (defaults to minimal shape).
+            flat: If True, flatten nested objects using dot notation.
+            flat_lists: If True, flatten arrays using indexed keys.
+            federal_account_symbol: Exact federal account symbol.
+            fiscal_year: Fiscal year (exact).
+            fiscal_year_gte: Fiscal year >=.
+            fiscal_year_lte: Fiscal year <=.
+            agency_code: Agency code (exact).
+            bureau_name: Bureau name (exact).
+            account_title: Account title (icontains).
+            bea_category: BEA category (exact).
+            on_off_budget: On/off budget flag (exact).
+            subfunction_code: Subfunction code (exact).
+            search: Full-text search over account_title/agency_name/bureau_name.
+            ordering: Sort field (prefix with '-' for descending).
+        """
+        params: dict[str, Any] = {"page": page, "limit": min(limit, 100)}
+        if shape is None:
+            shape = ShapeConfig.BUDGET_ACCOUNTS_MINIMAL
+        if shape:
+            params["shape"] = shape
+            if flat:
+                params["flat"] = "true"
+            if flat_lists:
+                params["flat_lists"] = "true"
+        for key, val in (
+            ("federal_account_symbol", federal_account_symbol),
+            ("fiscal_year", fiscal_year),
+            ("fiscal_year__gte", fiscal_year_gte),
+            ("fiscal_year__lte", fiscal_year_lte),
+            ("agency_code", agency_code),
+            ("bureau_name", bureau_name),
+            ("account_title__icontains", account_title),
+            ("bea_category", bea_category),
+            ("on_off_budget", on_off_budget),
+            ("subfunction_code", subfunction_code),
+            ("search", search),
+            ("ordering", ordering),
+        ):
+            if val is not None:
+                params[key] = val
+        data = self._get("/api/budget/accounts/", params)
+        results = [
+            self._parse_response_with_shape(obj, shape, BudgetAccount, flat, flat_lists)
+            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_budget_account(
+        self,
+        id: str | int,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single budget account by id (`/api/budget/accounts/{id}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.BUDGET_ACCOUNTS_MINIMAL
+        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/budget/accounts/{id}/", params)
+        return self._parse_response_with_shape(
+            data, shape, BudgetAccount, flat, flat_lists, joiner=joiner
+        )
+
+    def get_budget_account_quarters(
+        self,
+        id: str | int,
+        tas: str | None = None,
+        limit: int = 25,
+    ) -> PaginatedResponse:
+        """Get quarterly TAS-grain flow for a budget account.
+
+        (`/api/budget/accounts/{id}/quarters/`). FY21+ only.
+
+        Args:
+            id: Budget account id.
+            tas: Narrow to a single Treasury Account Symbol.
+            limit: Results per page (max 100).
+        """
+        params: dict[str, Any] = {"limit": min(limit, 100)}
+        if tas:
+            params["tas"] = tas
+        data = self._get(f"/api/budget/accounts/{id}/quarters/", params)
+        return PaginatedResponse(
+            count=int(data.get("count") or len(data.get("results") or [])),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=data.get("results") or [],
+        )
+
+    def get_budget_account_recipients(
+        self,
+        id: str | int,
+        funding_organization_id: str | None = None,
+        limit: int = 25,
+    ) -> PaginatedResponse:
+        """Get funding-office x recipient contract-flow detail for a budget account.
+
+        (`/api/budget/accounts/{id}/recipients/`).
+
+        Args:
+            id: Budget account id.
+            funding_organization_id: Narrow to a single funding office
+                (Organization UUID).
+            limit: Results per page (max 100).
+        """
+        params: dict[str, Any] = {"limit": min(limit, 100)}
+        if funding_organization_id:
+            params["funding_organization_id"] = funding_organization_id
+        data = self._get(f"/api/budget/accounts/{id}/recipients/", params)
+        return PaginatedResponse(
+            count=int(data.get("count") or len(data.get("results") or [])),
+            next=data.get("next"),
+            previous=data.get("previous"),
+            results=data.get("results") or [],
+        )
+
     # Grant endpoints
     def list_grants(
         self,
@@ -2341,6 +2743,7 @@ class TangoClient:
         cfda_number: str | None = None,
         funding_categories: str | None = None,
         funding_instruments: str | None = None,
+        grant_id: str | None = None,
         opportunity_number: str | None = None,
         ordering: str | None = None,
         posted_date_after: str | None = None,
@@ -2364,6 +2767,8 @@ class TangoClient:
             cfda_number: CFDA number filter
             funding_categories: Funding categories filter
             funding_instruments: Funding instruments filter
+            grant_id: Filter by grant ID (matches the detail-endpoint
+                identifier). Supports multi-value OR via '|' (e.g. "123|456").
             opportunity_number: Opportunity number filter
             ordering: Sort field (prefix with '-' for descending)
             posted_date_after: Posted date after
@@ -2390,6 +2795,7 @@ class TangoClient:
             ("cfda_number", cfda_number),
             ("funding_categories", funding_categories),
             ("funding_instruments", funding_instruments),
+            ("grant_id", grant_id),
             ("opportunity_number", opportunity_number),
             ("ordering", ordering),
             ("posted_date_after", posted_date_after),
@@ -2417,6 +2823,29 @@ class TangoClient:
             results=results,
         )
 
+    def get_grant(
+        self,
+        grant_id: str,
+        shape: str | None = None,
+        flat: bool = False,
+        flat_lists: bool = False,
+        joiner: str = ".",
+    ) -> Any:
+        """Get a single grant by its grant id (`/api/grants/{grant_id}/`)."""
+        params: dict[str, Any] = {}
+        if shape is None:
+            shape = ShapeConfig.GRANTS_MINIMAL
+        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/grants/{grant_id}/", params)
+        return self._parse_response_with_shape(data, shape, Grant, flat, flat_lists, joiner=joiner)
+
     # ============================================================================
     # Webhooks (v2)
     # ============================================================================
@@ -2630,13 +3059,13 @@ class TangoClient:
         return WebhookAlert(
             alert_id=str(data.get("alert_id") or data.get("id") or ""),
             name=str(data.get("name") or data.get("subscription_name") or ""),
-            query_type=(str(data["query_type"]) if data.get("query_type") is not None else None),
-            filters=data.get("filters") or data.get("filter_definition"),
+            query_type=str(data.get("query_type") or ""),
+            filters=data.get("filters") or data.get("filter_definition") or {},
             frequency=str(data.get("frequency", "realtime")),
             cron_expression=(
                 str(data["cron_expression"]) if data.get("cron_expression") is not None else None
             ),
-            status=str(data.get("status", "active")),
+            status=cast("Literal['active', 'paused']", data.get("status", "active")),
             created_at=str(data.get("created_at", "")),
             last_checked_at=(
                 str(data["last_checked_at"]) if data.get("last_checked_at") is not None else None

tango/models.py

@@ -281,21 +281,66 @@ class AwardTransaction:
     obligated: Decimal | None = None
 
 
+@dataclass
+class OrganizationOfficePayload:
+    """Schema definition for OrganizationOfficePayload (not used for instances).
+
+    Returned for a contract's ``awarding_office`` / ``funding_office``.
+    """
+
+    organization_id: str | None = None
+    office_code: str | None = None
+    office_name: str | None = None
+    agency_code: str | None = None
+    agency_name: str | None = None
+    department_code: int | None = None
+    department_name: str | None = None
+
+
 @dataclass
 class Contract:
-    """Schema definition for Contract (not used for instances)"""
+    """Schema definition for Contract (not used for instances).
 
-    id: str
-    award_id: str
-    recipient_name: str
-    description: str
-    award_amount: Decimal | None = None
+    Mirrors the API ``ContractList`` schema (``ContractListSerializer``).
+    ``/api/contracts/`` is shape-on-demand: which fields appear in a response
+    depends on the ``?shape=`` query param, so every field is optional.
+    """
+
+    key: str | None = None
+    piid: str | None = None
     award_date: date | None = None
     fiscal_year: int | None = None
+    obligated: Decimal | None = None
+    base_and_exercised_options_value: Decimal | None = None
+    total_contract_value: Decimal | None = None
+    naics_code: int | None = None
+    psc_code: str | None = None
+    set_aside: str | None = None
+    solicitation_identifier: str | None = None
+    description: str | None = None
+    awarding_office: OrganizationOfficePayload | None = None
+    funding_office: OrganizationOfficePayload | None = None
     recipient: RecipientProfile | None = None
+    parent_award: ParentAward | None = None
+    legislative_mandates: LegislativeMandates | None = None
+    place_of_performance: PlaceOfPerformance | None = None
+    subawards_summary: SubawardsSummary | None = None
+
+    # --- Deprecated fields (never returned by the API; removed in 2.0.0) ---
+    # Declared with None defaults so existing consumers reading these do not
+    # raise AttributeError; they always resolve to None.
+    id: str | None = None
+    """.. deprecated:: Never returned by the API. Removed in 2.0.0."""
+    award_id: str | None = None
+    """.. deprecated:: Never returned by the API. Removed in 2.0.0."""
+    recipient_name: str | None = None
+    """.. deprecated:: Use ``recipient.display_name``. Removed in 2.0.0."""
+    award_amount: Decimal | None = None
+    """.. deprecated:: Use ``obligated`` / ``total_contract_value``. Removed in 2.0.0."""
     awarding_agency: Agency | None = None
+    """.. deprecated:: Use ``awarding_office``. Removed in 2.0.0."""
     funding_agency: Agency | None = None
-    place_of_performance: Location | None = None
+    """.. deprecated:: Use ``funding_office``. Removed in 2.0.0."""
 
 
 @dataclass
@@ -626,9 +671,7 @@ class WebhookSamplePayloadAllResponse(TypedDict):
     note: str
 
 
-WebhookSamplePayloadResponse = (
-    WebhookSamplePayloadSingleResponse | WebhookSamplePayloadAllResponse
-)
+WebhookSamplePayloadResponse = WebhookSamplePayloadSingleResponse | WebhookSamplePayloadAllResponse
 
 
 @dataclass
@@ -713,6 +756,90 @@ class ValidateResult:
     errors: list[str] | None = None
 
 
+@dataclass
+class BudgetAccount:
+    """Schema definition for BudgetAccount (not used for instances).
+
+    Federal account x fiscal year budget rollup. ``/api/budget/accounts/`` is
+    shape-on-demand: which fields appear depends on the ``?shape=`` query param,
+    so every field is optional. Mirrors the API ``BudgetAccount`` schema.
+    """
+
+    id: int | None = None
+    federal_account_symbol: str | None = None
+    fiscal_year: int | None = None
+    agency_code: str | None = None
+    agency_name: str | None = None
+    bureau_name: str | None = None
+    account_title: str | None = None
+    bea_category: str | None = None
+    on_off_budget: str | None = None
+    subfunction_code: str | None = None
+    # Lifecycle
+    requested_ba: Decimal | None = None
+    enacted_ba: Decimal | None = None
+    apportioned: Decimal | None = None
+    obligated_total: Decimal | None = None
+    outlayed_total: Decimal | None = None
+    unobligated_balance: Decimal | None = None
+    # Contract / assistance / unlinked breakdown
+    contract_obligated: Decimal | None = None
+    contract_outlayed: Decimal | None = None
+    n_contracts: int | None = None
+    n_unique_contract_recipients: int | None = None
+    assistance_obligated: Decimal | None = None
+    assistance_outlayed: Decimal | None = None
+    n_grants: int | None = None
+    n_unique_grant_recipients: int | None = None
+    unlinked_obligated: Decimal | None = None
+    contract_share_of_obligated: Decimal | None = None
+    contract_share_of_obligated_capped: Decimal | None = None
+    contract_share_capped_flag: bool | None = None
+    assistance_share_of_obligated: Decimal | None = None
+    assistance_share_of_obligated_capped: Decimal | None = None
+    assistance_share_capped_flag: bool | None = None
+    # Forward-look
+    next_year_requested_ba: Decimal | None = None
+    ba_growth_next_year: Decimal | None = None
+    ba_growth_next_year_pct: Decimal | None = None
+    # Ratios
+    enacted_to_requested_pct: Decimal | None = None
+    enacted_to_requested_pct_capped: Decimal | None = None
+    enacted_to_requested_pct_capped_flag: bool | None = None
+    apportioned_to_enacted_pct: Decimal | None = None
+    apportioned_to_enacted_pct_capped: Decimal | None = None
+    apportioned_to_enacted_pct_capped_flag: bool | None = None
+    obligated_to_apportioned_pct: Decimal | None = None
+    obligated_to_apportioned_pct_capped: Decimal | None = None
+    obligated_to_apportioned_pct_capped_flag: bool | None = None
+    obligated_to_enacted_pct: Decimal | None = None
+    obligated_to_enacted_pct_capped: Decimal | None = None
+    obligated_to_enacted_pct_capped_flag: bool | None = None
+    outlayed_to_obligated_pct: Decimal | None = None
+    outlayed_to_obligated_pct_capped: Decimal | None = None
+    outlayed_to_obligated_pct_capped_flag: bool | None = None
+    unobligated_pct: Decimal | None = None
+    # Trends
+    enacted_ba_yoy_pct: Decimal | None = None
+    obligated_yoy_pct: Decimal | None = None
+    contract_obligated_yoy_pct: Decimal | None = None
+    enacted_ba_5yr_cagr: Decimal | None = None
+    contract_obligated_5yr_cagr: Decimal | None = None
+    # Request-vs-actual
+    requested_contractual_services: Decimal | None = None
+    requested_personnel_share: Decimal | None = None
+    actual_vs_requested_contract: Decimal | None = None
+    actual_vs_requested_contract_capped: Decimal | None = None
+    actual_vs_requested_contract_capped_flag: bool | None = None
+    # Provenance + narrative
+    appendix_pdf_url: str | None = None
+    account_narrative_excerpt: str | None = None
+    top_contract_recipients: list[Any] | None = None
+    top_grant_recipients: list[Any] | None = None
+    created: str | None = None
+    modified: str | None = None
+
+
 @dataclass
 class PaginatedResponse[T]:
     """Paginated API response
@@ -831,6 +958,18 @@ class ShapeConfig:
         "key,piid,award_date,obligated,total_contract_value,description,recipient(display_name,uei)"
     )
 
+    # Default for list_budget_accounts() / get_budget_account()
+    # Mirrors the API's BUDGET_ACCOUNT_DEFAULT_SHAPE.
+    BUDGET_ACCOUNTS_MINIMAL: Final = (
+        "id,federal_account_symbol,fiscal_year,agency_code,agency_name,bureau_name,"
+        "account_title,bea_category,on_off_budget,subfunction_code,"
+        "requested_ba,enacted_ba,apportioned,obligated_total,outlayed_total,"
+        "unobligated_balance,contract_obligated,contract_share_of_obligated_capped,"
+        "assistance_obligated,obligated_to_apportioned_pct_capped,"
+        "obligated_to_enacted_pct_capped,outlayed_to_obligated_pct_capped,"
+        "ba_growth_next_year_pct"
+    )
+
     # Default for list_organizations()
     ORGANIZATIONS_MINIMAL: Final = "key,fh_key,name,level,type,short_name"
 

tango/shapes/explicit_schemas.py

@@ -1315,12 +1315,8 @@ SUBAWARD_SCHEMA: dict[str, FieldSchema] = {
     "recipient_dba_name": FieldSchema(
         name="recipient_dba_name", type=str, is_optional=True, is_list=False
     ),
-    "recipient_duns": FieldSchema(
-        name="recipient_duns", type=str, is_optional=True, is_list=False
-    ),
-    "recipient_name": FieldSchema(
-        name="recipient_name", type=str, is_optional=True, is_list=False
-    ),
+    "recipient_duns": FieldSchema(name="recipient_duns", type=str, is_optional=True, is_list=False),
+    "recipient_name": FieldSchema(name="recipient_name", type=str, is_optional=True, is_list=False),
     "recipient_parent_duns": FieldSchema(
         name="recipient_parent_duns", type=str, is_optional=True, is_list=False
     ),
@@ -1330,9 +1326,7 @@ SUBAWARD_SCHEMA: dict[str, FieldSchema] = {
     "recipient_parent_uei": FieldSchema(
         name="recipient_parent_uei", type=str, is_optional=True, is_list=False
     ),
-    "recipient_uei": FieldSchema(
-        name="recipient_uei", type=str, is_optional=True, is_list=False
-    ),
+    "recipient_uei": FieldSchema(name="recipient_uei", type=str, is_optional=True, is_list=False),
     # Expandable nested objects
     "awarding_office": FieldSchema(
         name="awarding_office",
@@ -1493,6 +1487,10 @@ EXPLICIT_SCHEMAS: dict[str, dict[str, FieldSchema]] = {
     "Location": LOCATION_SCHEMA,
     "PlaceOfPerformance": PLACE_OF_PERFORMANCE_SCHEMA,
     "Competition": COMPETITION_SCHEMA,
+    # Alias: CONTRACT_SCHEMA/IDV_SCHEMA reference the competition leaf as
+    # "ContractOrIDVCompetition" (the models.py dataclass name); it is the same
+    # field set as Competition. Register both so nested shape selection resolves.
+    "ContractOrIDVCompetition": COMPETITION_SCHEMA,
     "ParentAward": PARENT_AWARD_SCHEMA,
     "LegislativeMandates": LEGISLATIVE_MANDATES_SCHEMA,
     "SubawardsSummary": SUBAWARDS_SUMMARY_SCHEMA,

tango/webhooks/cli.py

@@ -12,7 +12,7 @@ import json
 import sys
 import threading
 from pathlib import Path
-from typing import Any
+from typing import Any, cast
 
 try:
     import click
@@ -197,7 +197,7 @@ def simulate_cmd(
         from tango import TangoClient
 
         client = TangoClient(api_key=api_key, base_url=base_url)
-        payload = client.get_webhook_sample_payload(event_type=event_type)
+        payload = cast("dict[str, Any]", client.get_webhook_sample_payload(event_type=event_type))
     else:
         payload = {"events": [{"event_type": "tango.cli.simulated"}]}
 

{tango_python-1.0.0.dist-info → tango_python-1.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tango-python
-Version: 1.0.0
+Version: 1.1.1
 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
@@ -48,15 +48,16 @@ Requires-Dist: pytest>=8.0; extra == 'dev'
 Requires-Dist: python-dotenv>=1.0.0; extra == 'dev'
 Requires-Dist: pyyaml>=6.0; extra == 'dev'
 Requires-Dist: ruff>=0.3.0; extra == 'dev'
-Provides-Extra: notebooks
-Requires-Dist: ipykernel>=6.25.0; extra == 'notebooks'
-Requires-Dist: jupyter>=1.0.0; extra == 'notebooks'
 Provides-Extra: webhooks
 Requires-Dist: click>=8.1; extra == 'webhooks'
 Description-Content-Type: text/markdown
 
 # Tango Python SDK
 
+[![PyPI](https://img.shields.io/pypi/v/tango-python.svg)](https://pypi.org/project/tango-python/)
+[![Python Versions](https://img.shields.io/pypi/pyversions/tango-python.svg)](https://pypi.org/project/tango-python/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
 A modern Python SDK for the [Tango API](https://tango.makegov.com) by MakeGov, featuring dynamic response shaping and comprehensive type hints.
 
 ## Features

{tango_python-1.0.0.dist-info → tango_python-1.1.1.dist-info}/RECORD

@@ -1,9 +1,9 @@
-tango/__init__.py,sha256=36kjHn7zU7Fs9TTHShpW6cv_erODBUbfsBfQSyoVdeM,1872
-tango/client.py,sha256=52q9b2Lu9g6paqzKq8Okh_9GEbLeXsw7H0mvwzu0p5s,127938
+tango/__init__.py,sha256=3H_QVdoCe1ziPTZePlkIoA7j1jcZ11gotzZ2NslWms0,1912
+tango/client.py,sha256=CAejxkwhyMwisVMGSdVqlKGkSNfNT7G60MtZb6BtEYk,143984
 tango/exceptions.py,sha256=aRvDm0dUCEtNDfRVYCX7SEDdd1WlIVVY6sN78Tzo-a0,3114
-tango/models.py,sha256=-UreQ2AWmvL3ikJNUp4H2GmlkdLysWAguBUk3VFS6Ro,27833
+tango/models.py,sha256=QqUPtO7HJJDUaJDAMUkzvqR7pj1YIr8BetbS4palxc0,34261
 tango/shapes/__init__.py,sha256=7ea1WU74jp4znhNw-gXruag6m6eyPZtbVgbDFmFUWro,1072
-tango/shapes/explicit_schemas.py,sha256=D8CF5EertojSh5o37DkFnA32PcqbtoBUbfezfa9Vfy4,71175
+tango/shapes/explicit_schemas.py,sha256=8HgxKpAi2U80JP_MnABnW1JpYJA05aoOZSJsJIcyIWQ,71421
 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
@@ -11,12 +11,12 @@ tango/shapes/parser.py,sha256=-2Ap5jgeAvKsKtA-MaXPGE6PRB93GPV8GK99Z0geW_s,27468
 tango/shapes/schema.py,sha256=VRPOB1sBdjFyimNchrZKIpTHn83CyX4RfU9077aQtIU,14136
 tango/shapes/types.py,sha256=27jrAE0VIdrKaLjR_FK71hfIIGX2Tg3ex7REEBV1TFE,1301
 tango/webhooks/__init__.py,sha256=3bbiiGoB3s5iqqmQceroN0-MCSm-ZOZQx3M6JAknIUo,774
-tango/webhooks/cli.py,sha256=9J5jdgxeC2zHM0iif7GMpg_Mh7bv0HPQnADEvlv9OaE,13828
+tango/webhooks/cli.py,sha256=f_vQbJ2AeSQjKnQo7MxHFyUB2SAKcgHYmGQULS0isqQ,13858
 tango/webhooks/receiver.py,sha256=5yhsVhlLyoxmOCGvmbynWAIlDB2OaCPVf1H4GA1SxmU,9279
 tango/webhooks/signing.py,sha256=92Ee-0B6PR7ZkvY3Np3gzl88-mtfKkh-I7lxqCe2lGw,2374
 tango/webhooks/simulate.py,sha256=g2Osa0FYU5mJuon07T2aUCtmkUoTEzsY261tlp76fF0,3165
-tango_python-1.0.0.dist-info/METADATA,sha256=tbMMjm-Trd9JgpynHZFaShwxAIlaBU0p4QQV8E_DEI8,20435
-tango_python-1.0.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-tango_python-1.0.0.dist-info/entry_points.txt,sha256=kGLUbglUjuaAqEFvOZ1QuSW0vWb6VeSpCIFKaOFkKoQ,50
-tango_python-1.0.0.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
-tango_python-1.0.0.dist-info/RECORD,,
+tango_python-1.1.1.dist-info/METADATA,sha256=lUFYfhe3_JPafL82HboUatKTDzWWVeemou58w5ofYFo,20599
+tango_python-1.1.1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+tango_python-1.1.1.dist-info/entry_points.txt,sha256=kGLUbglUjuaAqEFvOZ1QuSW0vWb6VeSpCIFKaOFkKoQ,50
+tango_python-1.1.1.dist-info/licenses/LICENSE,sha256=j2kYVHMwTkoGn3ZNScnrdIueG0k2XzB_LCPFoyBc2wk,1064
+tango_python-1.1.1.dist-info/RECORD,,