lago-python-client 1.32.0__tar.gz → 1.33.0__tar.gz

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lago-python-client
-Version: 1.32.0
+Version: 1.33.0
 Summary: Lago Python API Client
 Home-page: https://github.com/getlago/lago-python-client
 Author: Lovro Colic
@@ -24,7 +24,7 @@ Requires-Dist: classes~=0.4.1
 Requires-Dist: httpx<1.0.0,>=0.24.0
 Requires-Dist: orjson~=3.8
 Requires-Dist: pydantic<3,>=1.10
-Requires-Dist: typeguard~=3.0.2
+Requires-Dist: typeguard~=4.0
 Provides-Extra: test
 Requires-Dist: mypy==0.971; extra == "test"
 Requires-Dist: pytest; extra == "test"

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/client.py

@@ -22,6 +22,7 @@ from .invoice_collections.clients import InvoiceCollectionClient
 from .plans.clients import PlanClient
 from .subscriptions.clients import SubscriptionClient
 from .taxes.clients import TaxClient
+from .usages.clients import UsageClient
 from .wallets.clients import WalletClient, WalletTransactionClient
 from .webhooks.clients import WebhookClient
 from .webhook_endpoints.clients import WebhookEndpointClient
@@ -151,6 +152,10 @@ class Client:
     def taxes(self) -> TaxClient:
         return TaxClient(self.base_api_url, self.api_key)
 
+    @callable_cached_property
+    def usages(self) -> UsageClient:
+        return UsageClient(self.base_api_url, self.api_key)
+
     @callable_cached_property
     def wallets(self) -> WalletClient:
         return WalletClient(self.base_api_url, self.api_key)

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/customers/clients.py

@@ -10,6 +10,7 @@ from ..mixins import (
 )
 from ..models.customer import CustomerResponse
 from ..models.customer_usage import CustomerUsageResponse
+from ..models.customer_projected_usage import CustomerProjectedUsageResponse
 from ..services.request import make_headers, make_url, send_get_request, send_post_request, QueryPairs
 from ..services.response import (
     get_response_data,
@@ -56,6 +57,27 @@ class CustomerClient(
             data=get_response_data(response=api_response, key="customer_usage"),
         )
 
+    def projected_usage(
+        self, resource_id: str, external_subscription_id: str, apply_taxes: Optional[str] = None
+    ) -> CustomerProjectedUsageResponse:
+        query_params = {"external_subscription_id": external_subscription_id}
+        if apply_taxes is not None:
+            query_params["apply_taxes"] = apply_taxes
+
+        api_response: Response = send_get_request(
+            url=make_url(
+                origin=self.base_url,
+                path_parts=(self.API_RESOURCE, resource_id, "projected_usage"),
+                query_pairs=query_params,
+            ),
+            headers=make_headers(api_key=self.api_key),
+        )
+
+        return prepare_object_response(
+            response_model=CustomerProjectedUsageResponse,
+            data=get_response_data(response=api_response, key="customer_projected_usage"),
+        )
+
     def past_usage(
         self,
         resource_id: str,

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/models/__init__.py

@@ -67,6 +67,10 @@ from .invoice import (
     InvoiceFee as InvoiceFee,
 )
 from .invoice_item import InvoiceItemResponse as InvoiceItemResponse
+from .invoice_custom_section import (
+    InvoiceCustomSectionResponse as InvoiceCustomSectionResponse,
+    InvoiceCustomSectionsResponseList as InvoiceCustomSectionsResponseList,
+)
 from .minimum_commitment import (
     MinimumCommitment as MinimumCommitment,
     MinimumCommitmentResponse as MinimumCommitmentResponse,

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/models/billing_entity.py

@@ -1,7 +1,9 @@
 from typing import Optional, List
+from datetime import datetime
+
 from ..base_model import BaseModel, BaseResponseModel
 from .tax import TaxesResponse
-from datetime import datetime
+from .invoice_custom_section import InvoiceCustomSectionsResponseList
 
 
 class BillingEntityBillingConfiguration(BaseModel):
@@ -60,6 +62,7 @@ class BillingEntityUpdate(BaseModel):
     tax_codes: Optional[List[str]]
     email_settings: Optional[List[str]]
     billing_configuration: Optional[BillingEntityBillingConfiguration]
+    invoice_custom_section_codes: Optional[List[str]]
 
 
 class BillingEntityResponse(BaseResponseModel):
@@ -92,5 +95,6 @@ class BillingEntityResponse(BaseResponseModel):
     invoice_grace_period: Optional[int]
     email_settings: Optional[List[str]]
     taxes: Optional[TaxesResponse]
+    selected_invoice_custom_sections: Optional[InvoiceCustomSectionsResponseList]
     created_at: datetime
     updated_at: datetime

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/models/customer.py

@@ -5,6 +5,7 @@ from lago_python_client.base_model import BaseModel
 from ..base_model import BaseResponseModel
 
 from .tax import TaxesResponse
+from .invoice_custom_section import InvoiceCustomSectionsResponseList
 
 
 class CustomerBillingConfiguration(BaseModel):
@@ -75,21 +76,6 @@ class MetadataResponseList(BaseModel):
     __root__: List[MetadataResponse]
 
 
-class InvoiceCustomSection(BaseModel):
-    lago_id: Optional[str]
-    code: Optional[str]
-    name: Optional[str]
-    description: Optional[str]
-    details: Optional[str]
-    display_name: Optional[str]
-    applied_to_organization: Optional[bool]
-    created_at: Optional[str]
-
-
-class InvoiceCustomSectionsResponseList(BaseModel):
-    __root__: List[InvoiceCustomSection]
-
-
 class Customer(BaseModel):
     external_id: str
     address_line1: Optional[str]

lago_python_client-1.33.0/lago_python_client/models/customer_projected_usage.py

@@ -0,0 +1,73 @@
+from typing import Dict, List, Optional
+
+from lago_python_client.base_model import BaseModel
+
+from ..base_model import BaseResponseModel
+
+
+class Metric(BaseModel):
+    lago_id: str
+    name: str
+    code: str
+    aggregation_type: str
+
+
+class PricingUnitDetails(BaseModel):
+    amount_cents: int
+    short_name: str
+    conversion_rate: float
+
+
+class ProjectedChargeFilterUsage(BaseModel):
+    invoice_display_name: Optional[str]
+    values: Dict[str, List[str]]
+    units: str
+    projected_units: str
+    amount_cents: int
+    projected_amount_cents: int
+    events_count: int
+    pricing_unit_details: Optional[PricingUnitDetails]
+
+
+class ChargeObject(BaseModel):
+    lago_id: str
+    charge_model: str
+    invoice_display_name: Optional[str]
+
+
+class ProjectedGroupedUsage(BaseModel):
+    amount_cents: int
+    projected_amount_cents: int
+    events_count: int
+    units: str
+    projected_units: str
+    grouped_by: Dict[str, str]
+    filters: List[ProjectedChargeFilterUsage]
+    pricing_unit_details: Optional[PricingUnitDetails]
+
+
+class ProjectedChargeUsage(BaseModel):
+    units: str
+    projected_units: str
+    events_count: int
+    amount_cents: int
+    projected_amount_cents: int
+    amount_currency: str
+    charge: ChargeObject
+    billable_metric: Metric
+    filters: List[ProjectedChargeFilterUsage]
+    grouped_usage: Optional[List[ProjectedGroupedUsage]]
+    pricing_unit_details: Optional[PricingUnitDetails]
+
+
+class CustomerProjectedUsageResponse(BaseResponseModel):
+    from_datetime: str
+    to_datetime: str
+    issuing_date: str
+    invoice_id: Optional[str]
+    currency: str
+    amount_cents: int
+    projected_amount_cents: int
+    total_amount_cents: int
+    taxes_amount_cents: int
+    charges_usage: List[ProjectedChargeUsage]

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/models/customer_usage.py

@@ -36,14 +36,14 @@ class ChargeObject(BaseModel):
 class GroupedUsage(BaseModel):
     amount_cents: int
     events_count: int
-    units: float
+    units: str
     grouped_by: Dict[str, str]
     filters: List[ChargeFilterUsage]
     pricing_unit_details: Optional[PricingUnitDetails]
 
 
 class ChargeUsage(BaseModel):
-    units: float
+    units: str
     events_count: int
     amount_cents: int
     amount_currency: str

lago_python_client-1.33.0/lago_python_client/models/invoice_custom_section.py

@@ -0,0 +1,17 @@
+from typing import List, Optional
+
+from ..base_model import BaseResponseModel
+
+
+class InvoiceCustomSectionResponse(BaseResponseModel):
+    lago_id: Optional[str]
+    code: Optional[str]
+    name: Optional[str]
+    description: Optional[str]
+    details: Optional[str]
+    display_name: Optional[str]
+    applied_to_organization: Optional[bool]  # Deprecated
+
+
+class InvoiceCustomSectionsResponseList(BaseResponseModel):
+    __root__: List[InvoiceCustomSectionResponse]

lago_python_client-1.33.0/lago_python_client/models/usage.py

@@ -0,0 +1,18 @@
+from typing import List
+
+from ..base_model import BaseResponseModel
+
+
+class UsageResponse(BaseResponseModel):
+    organization_id: str
+    start_of_period_dt: str
+    end_of_period_dt: str
+    amount_currency: str
+    amount_cents: int
+    billable_metric_code: str
+    units: float
+    is_billable_metric_deleted: bool
+
+
+class UsagesResponse(BaseResponseModel):
+    __root__: List[UsageResponse]

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/models/wallet.py

@@ -45,6 +45,7 @@ class RecurringTransactionRuleResponseList(BaseModel):
 
 class AppliesTo(BaseModel):
     fee_types: Optional[List[str]]
+    billable_metric_codes: Optional[List[str]]
 
 
 class Wallet(BaseModel):

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/payment_requests/clients.py

@@ -1,13 +1,14 @@
 from typing import ClassVar, Type
 
 from ..base_client import BaseClient
-from ..mixins import CreateCommandMixin, FindAllCommandMixin
+from ..mixins import CreateCommandMixin, FindAllCommandMixin, FindCommandMixin
 from ..models.payment_request import PaymentRequestResponse
 
 
 class PaymentRequestClient(
     CreateCommandMixin[PaymentRequestResponse],
     FindAllCommandMixin[PaymentRequestResponse],
+    FindCommandMixin[PaymentRequestResponse],
     BaseClient,
 ):
     API_RESOURCE: ClassVar[str] = "payment_requests"

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/services/response.py

@@ -13,7 +13,7 @@ except ImportError:  # Python 3.7
 
 from httpx import Response as Response  # not a typo! implicit reexport
 from lago_python_client.base_model import BaseModel
-import typeguard
+from typeguard import check_type, CollectionCheckStrategy, TypeCheckError
 
 from ..exceptions import LagoApiError
 from ..services.json import DeserializedData, from_json
@@ -25,8 +25,6 @@ else:
 
 _M = TypeVar("_M", bound=BaseModel)
 
-typeguard.config.collection_check_strategy = typeguard.CollectionCheckStrategy.ALL_ITEMS
-
 RESPONSE_SUCCESS_CODES: Final[Set[int]] = {
     HTTPStatus.OK,  # 200
     HTTPStatus.CREATED,  # 201
@@ -74,8 +72,10 @@ def get_response_data(*, response: Response, key: str = "") -> Optional[_Mapping
 
     # Ensure deserialized_data has correct type: sequence or mapping or raise LagoApiError
     try:
-        mapping_or_sequence_data = typeguard.check_type(deserialized_data, _MappingOrSequence)
-    except typeguard.TypeCheckError as exc:
+        mapping_or_sequence_data = check_type(
+            deserialized_data, _MappingOrSequence, collection_check_strategy=CollectionCheckStrategy.ALL_ITEMS
+        )
+    except TypeCheckError as exc:
         raise LagoApiError(
             status_code=HTTPStatus.INTERNAL_SERVER_ERROR,  # 500
             url=None,
@@ -113,8 +113,10 @@ def prepare_index_response(
     # Ensure deserialized_data has correct type: mapping with mapping or sequence inside or raise LagoApiError
 
     try:
-        response_data: Mapping[str, _MappingOrSequence] = typeguard.check_type(data, Mapping[str, _MappingOrSequence])
-    except typeguard.TypeCheckError as exc:
+        response_data: Mapping[str, _MappingOrSequence] = check_type(
+            data, Mapping[str, _MappingOrSequence], collection_check_strategy=CollectionCheckStrategy.ALL_ITEMS
+        )
+    except TypeCheckError as exc:
         raise LagoApiError(
             status_code=HTTPStatus.INTERNAL_SERVER_ERROR,  # 500
             url=None,
@@ -140,8 +142,10 @@ def prepare_object_list_response(
     # The only usage - ``WalletTransactionClient.create``
     # Ensure deserialized_data has correct type: sequence with mapping or sequence inside or raise LagoApiError
     try:
-        response_data: Sequence[_MappingOrSequence] = typeguard.check_type(data, Sequence[_MappingOrSequence])
-    except typeguard.TypeCheckError as exc:
+        response_data: Sequence[_MappingOrSequence] = check_type(
+            data, Sequence[_MappingOrSequence], collection_check_strategy=CollectionCheckStrategy.ALL_ITEMS
+        )
+    except TypeCheckError as exc:
         raise LagoApiError(
             status_code=HTTPStatus.INTERNAL_SERVER_ERROR,  # 500
             url=None,

lago_python_client-1.33.0/lago_python_client/usages/clients.py

@@ -0,0 +1,39 @@
+import sys
+from typing import Any, ClassVar, Type
+
+from ..base_client import BaseClient
+from ..mixins import FindAllCommandMixin
+from ..models.usage import UsageResponse
+from ..services.request import make_headers, make_url, send_get_request, QueryPairs
+from ..services.response import get_response_data, prepare_index_response, Response
+
+if sys.version_info >= (3, 9):
+    from collections.abc import Mapping
+else:
+    from typing import Mapping
+
+
+class UsageClient(
+    FindAllCommandMixin[UsageResponse],
+    BaseClient,
+):
+    API_RESOURCE: ClassVar[str] = "usages"
+    RESPONSE_MODEL: ClassVar[Type[UsageResponse]] = UsageResponse
+    ROOT_NAME: ClassVar[str] = "usage"
+
+    def find_all(self, options: QueryPairs = {}) -> Mapping[str, Any]:
+        api_response: Response = send_get_request(
+            url=make_url(
+                origin=self.base_url,
+                path_parts=("analytics", "usage"),
+                query_pairs=options,
+            ),
+            headers=make_headers(api_key=self.api_key),
+        )
+
+        # Process response data
+        return prepare_index_response(
+            api_resource=self.API_RESOURCE,
+            response_model=self.RESPONSE_MODEL,
+            data=get_response_data(response=api_response),
+        )

lago_python_client-1.33.0/lago_python_client/version.py

@@ -0,0 +1 @@
+LAGO_VERSION = "1.33.0"

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client/webhooks/clients.py

@@ -9,7 +9,7 @@ except ImportError:
     from typing_extensions import TypedDict
 
 from lago_python_client.base_model import BaseModel
-import typeguard
+from typeguard import check_type, CollectionCheckStrategy, TypeCheckError
 
 from ..base_client import BaseClient
 from ..exceptions import LagoApiError
@@ -21,8 +21,6 @@ if sys.version_info >= (3, 9):
 else:
     from typing import Mapping, Sequence
 
-typeguard.config.collection_check_strategy = typeguard.CollectionCheckStrategy.ALL_ITEMS
-
 
 class _ResponseWithPublicKeyInside(TypedDict):
     public_key: str
@@ -46,10 +44,10 @@ class WebhookClient(BaseClient):
         )
 
         try:
-            checked_response_data: _ResponseWithPublicKeyInside = typeguard.check_type(
-                response_data, _ResponseWithPublicKeyInside
+            checked_response_data: _ResponseWithPublicKeyInside = check_type(
+                response_data, _ResponseWithPublicKeyInside, collection_check_strategy=CollectionCheckStrategy.ALL_ITEMS
             )  # type: ignore
-        except typeguard.TypeCheckError:
+        except TypeCheckError:
             raise LagoApiError(
                 status_code=HTTPStatus.INTERNAL_SERVER_ERROR,  # 500
                 url=None,

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lago-python-client
-Version: 1.32.0
+Version: 1.33.0
 Summary: Lago Python API Client
 Home-page: https://github.com/getlago/lago-python-client
 Author: Lovro Colic
@@ -24,7 +24,7 @@ Requires-Dist: classes~=0.4.1
 Requires-Dist: httpx<1.0.0,>=0.24.0
 Requires-Dist: orjson~=3.8
 Requires-Dist: pydantic<3,>=1.10
-Requires-Dist: typeguard~=3.0.2
+Requires-Dist: typeguard~=4.0
 Provides-Extra: test
 Requires-Dist: mypy==0.971; extra == "test"
 Requires-Dist: pytest; extra == "test"

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client.egg-info/SOURCES.txt

@@ -52,6 +52,7 @@ setup.cfg
 ./lago_python_client/models/credit.py
 ./lago_python_client/models/credit_note.py
 ./lago_python_client/models/customer.py
+./lago_python_client/models/customer_projected_usage.py
 ./lago_python_client/models/customer_usage.py
 ./lago_python_client/models/error_detail.py
 ./lago_python_client/models/event.py
@@ -59,6 +60,7 @@ setup.cfg
 ./lago_python_client/models/gross_revenue.py
 ./lago_python_client/models/invoice.py
 ./lago_python_client/models/invoice_collection.py
+./lago_python_client/models/invoice_custom_section.py
 ./lago_python_client/models/invoice_item.py
 ./lago_python_client/models/invoiced_usage.py
 ./lago_python_client/models/lifetime_usage.py
@@ -72,6 +74,7 @@ setup.cfg
 ./lago_python_client/models/plan.py
 ./lago_python_client/models/subscription.py
 ./lago_python_client/models/tax.py
+./lago_python_client/models/usage.py
 ./lago_python_client/models/usage_threshold.py
 ./lago_python_client/models/wallet.py
 ./lago_python_client/models/wallet_transaction.py
@@ -98,6 +101,8 @@ setup.cfg
 ./lago_python_client/subscriptions/clients.py
 ./lago_python_client/taxes/__init__.py
 ./lago_python_client/taxes/clients.py
+./lago_python_client/usages/__init__.py
+./lago_python_client/usages/clients.py
 ./lago_python_client/wallets/__init__.py
 ./lago_python_client/wallets/clients.py
 ./lago_python_client/webhook_endpoints/__init__.py
@@ -134,6 +139,7 @@ setup.cfg
 ./tests/test_response_services.py
 ./tests/test_subscription_client.py
 ./tests/test_tax_client.py
+./tests/test_usage_client.py
 ./tests/test_wallet_client.py
 ./tests/test_wallet_transaction_client.py
 ./tests/test_webhook_client.py
@@ -193,6 +199,7 @@ lago_python_client/models/coupon.py
 lago_python_client/models/credit.py
 lago_python_client/models/credit_note.py
 lago_python_client/models/customer.py
+lago_python_client/models/customer_projected_usage.py
 lago_python_client/models/customer_usage.py
 lago_python_client/models/error_detail.py
 lago_python_client/models/event.py
@@ -200,6 +207,7 @@ lago_python_client/models/fee.py
 lago_python_client/models/gross_revenue.py
 lago_python_client/models/invoice.py
 lago_python_client/models/invoice_collection.py
+lago_python_client/models/invoice_custom_section.py
 lago_python_client/models/invoice_item.py
 lago_python_client/models/invoiced_usage.py
 lago_python_client/models/lifetime_usage.py
@@ -213,6 +221,7 @@ lago_python_client/models/payment_request.py
 lago_python_client/models/plan.py
 lago_python_client/models/subscription.py
 lago_python_client/models/tax.py
+lago_python_client/models/usage.py
 lago_python_client/models/usage_threshold.py
 lago_python_client/models/wallet.py
 lago_python_client/models/wallet_transaction.py
@@ -239,6 +248,8 @@ lago_python_client/subscriptions/__init__.py
 lago_python_client/subscriptions/clients.py
 lago_python_client/taxes/__init__.py
 lago_python_client/taxes/clients.py
+lago_python_client/usages/__init__.py
+lago_python_client/usages/clients.py
 lago_python_client/wallets/__init__.py
 lago_python_client/wallets/clients.py
 lago_python_client/webhook_endpoints/__init__.py
@@ -275,6 +286,7 @@ tests/test_request_services.py
 tests/test_response_services.py
 tests/test_subscription_client.py
 tests/test_tax_client.py
+tests/test_usage_client.py
 tests/test_wallet_client.py
 tests/test_wallet_transaction_client.py
 tests/test_webhook_client.py

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/lago_python_client.egg-info/requires.txt

@@ -2,7 +2,7 @@ classes~=0.4.1
 httpx<1.0.0,>=0.24.0
 orjson~=3.8
 pydantic<3,>=1.10
-typeguard~=3.0.2
+typeguard~=4.0
 
 [lint]
 ruff

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/setup.cfg

@@ -31,7 +31,7 @@ install_requires =
 	httpx>=0.24.0,<1.0.0 # minor auto updates allowed
 	orjson~=3.8  # minor auto updates allowed
 	pydantic>=1.10,<3 # minor auto updates allowed
-	typeguard~=3.0.2  # bugfix auto updates allowed
+	typeguard~=4.0  # bugfix auto updates allowed
 
 [options.extras_require]
 test = 

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/tests/test_billing_entity_client.py

@@ -77,6 +77,7 @@ def update_billing_entity_input():
             invoice_grace_period=5,
             document_locale="fr",
         ),
+        invoice_custom_section_codes=["custom_section_1", "custom_section_2"],
     )
 
 

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/tests/test_customer_client.py

@@ -95,6 +95,7 @@ def test_valid_create_customers_request(httpx_mock: HTTPXMock):
     assert response.shipping_address.state == "XZ"
     assert response.integration_customers.__root__[0].external_customer_id == "test-12345"
     assert response.integration_customers.__root__[0].type == "netsuite"
+    assert response.applicable_invoice_custom_sections.__root__[0].code == "custom_section_code"
     assert response.metadata.__root__[0].lago_id == "12345"
     assert response.metadata.__root__[0].key == "key"
     assert response.metadata.__root__[0].value == "value"
@@ -126,11 +127,34 @@ def test_valid_current_usage(httpx_mock: HTTPXMock):
 
     assert response.from_datetime == "2022-07-01T00:00:00Z"
     assert len(response.charges_usage) == 1
-    assert response.charges_usage[0].units == 1.0
+    assert response.charges_usage[0].units == "1.0"
     assert len(response.charges_usage[0].filters) == 1
     assert response.charges_usage[0].filters[0].values["country"] == ["france"]
 
 
+def test_valid_projected_usage(httpx_mock: HTTPXMock):
+    client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
+
+    httpx_mock.add_response(
+        method="GET",
+        url="https://api.getlago.com/api/v1/customers/external_customer_id/projected_usage?external_subscription_id=123",
+        content=mock_response("customer_projected_usage"),
+    )
+    response = client.customers.projected_usage("external_customer_id", "123")
+
+    assert response.from_datetime == "2022-07-01T00:00:00Z"
+    assert len(response.charges_usage) == 1
+    assert response.charges_usage[0].units == "1.0"
+    assert response.charges_usage[0].projected_units == "2.0"
+    assert response.charges_usage[0].amount_cents == 123
+    assert response.charges_usage[0].projected_amount_cents == 256
+    assert response.charges_usage[0].amount_currency == "EUR"
+    assert response.charges_usage[0].charge.lago_id == "5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba"
+    assert response.charges_usage[0].charge.charge_model == "graduated"
+    assert response.charges_usage[0].charge.invoice_display_name == "add_on_invoice_display_name"
+    assert response.charges_usage[0].billable_metric.lago_id == "99a6094e-199b-4101-896a-54e927ce7bd7"
+
+
 def test_valid_current_usage_without_taxes(httpx_mock: HTTPXMock):
     client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
 
@@ -143,7 +167,7 @@ def test_valid_current_usage_without_taxes(httpx_mock: HTTPXMock):
 
     assert response.from_datetime == "2022-07-01T00:00:00Z"
     assert len(response.charges_usage) == 1
-    assert response.charges_usage[0].units == 1.0
+    assert response.charges_usage[0].units == "1.0"
     assert len(response.charges_usage[0].filters) == 1
     assert response.charges_usage[0].filters[0].values["country"] == ["france"]
 
@@ -175,7 +199,7 @@ def test_valid_past_usage(httpx_mock: HTTPXMock):
     assert len(response["usage_periods"]) == 1
     assert response["usage_periods"][0].from_datetime == "2022-07-01T00:00:00Z"
     assert len(response["usage_periods"][0].charges_usage) == 1
-    assert response["usage_periods"][0].charges_usage[0].units == 1.0
+    assert response["usage_periods"][0].charges_usage[0].units == "1.0"
     assert len(response["usage_periods"][0].charges_usage[0].filters) == 1
 
 

{lago_python_client-1.32.0 → lago_python_client-1.33.0}/tests/test_payment_request_client.py

@@ -32,6 +32,37 @@ def mock_collection_response():
         return payment_requests_response.read()
 
 
+def test_valid_find_payment_request_request(httpx_mock: HTTPXMock):
+    client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
+    payment_request_id = "89b6b61e-4dbc-4307-ac96-4abcfa9e3e2d"
+
+    httpx_mock.add_response(
+        method="GET",
+        url="https://api.getlago.com/api/v1/payment_requests/" + payment_request_id,
+        content=mock_response(),
+    )
+    response = client.payment_requests.find(payment_request_id)
+
+    assert response.lago_id == "89b6b61e-4dbc-4307-ac96-4abcfa9e3e2d"
+    assert response.amount_cents == 19955
+    assert response.amount_currency == "EUR"
+
+
+def test_invalid_find_payment_request_request(httpx_mock: HTTPXMock):
+    client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
+    payment_request_id = "invalid"
+
+    httpx_mock.add_response(
+        method="GET",
+        url="https://api.getlago.com/api/v1/payment_requests/" + payment_request_id,
+        status_code=404,
+        content=b"",
+    )
+
+    with pytest.raises(LagoApiError):
+        client.payment_requests.find(payment_request_id)
+
+
 def test_valid_find_all_payment_requests_request(httpx_mock: HTTPXMock):
     client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")