lago-python-client 1.15.1__tar.gz → 1.16.0__tar.gz

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lago-python-client
-Version: 1.15.1
+Version: 1.16.0
 Summary: Lago Python API Client
 Home-page: https://github.com/getlago/lago-python-client
 Author: Lovro Colic

lago_python_client-1.16.0/lago_python_client/billable_metrics/clients.py

@@ -0,0 +1,46 @@
+from typing import ClassVar, Type
+
+from ..base_client import BaseClient
+from ..mixins import (
+    CreateCommandMixin,
+    DestroyCommandMixin,
+    FindAllCommandMixin,
+    FindCommandMixin,
+    UpdateCommandMixin,
+)
+from ..models.billable_metric import (
+    BillableMetricEvaluateExpression,
+    BillableMetricEvaluateExpressionResponse,
+    BillableMetricResponse,
+)
+from ..services.request import make_headers, make_url, send_post_request
+from ..services.response import Response, prepare_object_response, get_response_data
+
+
+class BillableMetricClient(
+    CreateCommandMixin[BillableMetricResponse],
+    DestroyCommandMixin[BillableMetricResponse],
+    FindAllCommandMixin[BillableMetricResponse],
+    FindCommandMixin[BillableMetricResponse],
+    UpdateCommandMixin[BillableMetricResponse],
+    BaseClient,
+):
+    API_RESOURCE: ClassVar[str] = "billable_metrics"
+    RESPONSE_MODEL: ClassVar[Type[BillableMetricResponse]] = BillableMetricResponse
+    ROOT_NAME: ClassVar[str] = "billable_metric"
+
+    def evaluate_expression(
+        self, input_object: BillableMetricEvaluateExpression
+    ) -> BillableMetricEvaluateExpressionResponse:
+        api_response: Response = send_post_request(
+            url=make_url(
+                origin=self.base_url,
+                path_parts=(self.API_RESOURCE, "evaluate_expression"),
+            ),
+            headers=make_headers(api_key=self.api_key),
+        )
+
+        return prepare_object_response(
+            response_model=BillableMetricEvaluateExpressionResponse,
+            data=get_response_data(response=api_response, key="expression_result"),
+        )

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/lago_python_client/models/__init__.py

@@ -3,6 +3,9 @@ from .billable_metric import (
     BillableMetric as BillableMetric,
     BillableMetricFilter as BillableMetricFilter,
     BillableMetricFilters as BillableMetricFilters,
+    BillableMetricEvaluateExpressionEvent as BillableMetricEvaluateExpressionEvent,
+    BillableMetricEvaluateExpression as BillableMetricEvaluateExpression,
+    BillableMetricEvaluateExpressionResponse as BillableMetricEvaluateExpressionResponse,
 )
 from .charge import (
     Charge as Charge,

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/lago_python_client/models/billable_metric.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from lago_python_client.base_model import BaseModel
 
@@ -19,8 +19,11 @@ class BillableMetric(BaseModel):
     code: Optional[str]
     description: Optional[str]
     recurring: Optional[bool]
+    rounding_function: Optional[str]
+    rounding_precision: Optional[int]
     aggregation_type: Optional[str]
     weighted_interval: Optional[str]
+    expression: Optional[str]
     field_name: Optional[str]
     filters: Optional[BillableMetricFilters]
 
@@ -31,11 +34,29 @@ class BillableMetricResponse(BaseResponseModel):
     code: str
     description: Optional[str]
     recurring: Optional[bool]
+    rounding_function: Optional[str]
+    rounding_precision: Optional[int]
     aggregation_type: Optional[str]
     weighted_interval: Optional[str]
+    expression: Optional[str]
     field_name: Optional[str]
     created_at: str
     filters: BillableMetricFilters
     active_subscriptions_count: int
     draft_invoices_count: int
     plans_count: int
+
+
+class BillableMetricEvaluateExpressionEvent(BaseModel):
+    code: Optional[str]
+    timestamp: Optional[Union[str, int]]
+    properties: Optional[Dict[str, Any]]
+
+
+class BillableMetricEvaluateExpression(BaseModel):
+    expression: str
+    event: BillableMetricEvaluateExpressionEvent
+
+
+class BillableMetricEvaluateExpressionResponse(BaseResponseModel):
+    value: Union[str, int, float]

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/lago_python_client/models/plan.py

@@ -32,6 +32,7 @@ class Plan(BaseModel):
     minimum_commitment: Optional[MinimumCommitment]
     usage_thresholds: Optional[UsageThresholds]
     tax_codes: Optional[List[str]]
+    cascade_updates: Optional[bool]
 
 
 class PlanResponse(BaseResponseModel):

lago_python_client-1.16.0/lago_python_client/version.py

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

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/lago_python_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lago-python-client
-Version: 1.15.1
+Version: 1.16.0
 Summary: Lago Python API Client
 Home-page: https://github.com/getlago/lago-python-client
 Author: Lovro Colic

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/tests/test_billable_metric_client.py

@@ -9,6 +9,8 @@ from lago_python_client.models import (
     BillableMetric,
     BillableMetricFilter,
     BillableMetricFilters,
+    BillableMetricEvaluateExpression,
+    BillableMetricEvaluateExpressionEvent,
 )
 
 
@@ -25,6 +27,17 @@ def billable_metric_object():
     )
 
 
+def billable_metric_evaluate_expression_object():
+    return BillableMetricEvaluateExpression(
+        expression="round(event.properties.value * event.properties.units)",
+        event=BillableMetricEvaluateExpressionEvent(
+            code="event_code",
+            timestamp=1651240791,
+            properties={"value": 10, "units": 2},
+        ),
+    )
+
+
 def filters():
     return BillableMetricFilters(__root__=[BillableMetricFilter(key="country", values=["france", "italy", "spain"])])
 
@@ -45,6 +58,14 @@ def mock_collection_response():
         return billable_metric_response.read()
 
 
+def mock_evaluate_expression_response():
+    this_dir = os.path.dirname(os.path.abspath(__file__))
+    data_path = os.path.join(this_dir, "fixtures/billable_metric_evaluate_expression.json")
+
+    with open(data_path, "rb") as evaluate_expression_response:
+        return evaluate_expression_response.read()
+
+
 def test_valid_create_billable_metric_request(httpx_mock: HTTPXMock):
     client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
 
@@ -204,3 +225,16 @@ def test_invalid_find_all_billable_metric_request(httpx_mock: HTTPXMock):
 
     with pytest.raises(LagoApiError):
         client.billable_metrics.find_all()
+
+
+def test_valid_evaluate_billable_metric_expression(httpx_mock: HTTPXMock):
+    client = Client(api_key="886fe239-927d-4072-ab72-6dd345e8dd0d")
+
+    httpx_mock.add_response(
+        method="POST",
+        url="https://api.getlago.com/api/v1/billable_metrics/evaluate_expression",
+        content=mock_evaluate_expression_response(),
+    )
+    response = client.billable_metrics.evaluate_expression(billable_metric_evaluate_expression_object())
+
+    assert response.value == "2.0"

{lago_python_client-1.15.1 → lago_python_client-1.16.0}/tests/test_plan_client.py

@@ -63,6 +63,7 @@ def plan_object():
         trial_period=None,
         bill_charges_monthly=None,
         tax_codes=None,
+        cascade_updates=None,
     )
 
 

lago_python_client-1.15.1/lago_python_client/billable_metrics/clients.py

@@ -1,24 +0,0 @@
-from typing import ClassVar, Type
-
-from ..base_client import BaseClient
-from ..mixins import (
-    CreateCommandMixin,
-    DestroyCommandMixin,
-    FindAllCommandMixin,
-    FindCommandMixin,
-    UpdateCommandMixin,
-)
-from ..models.billable_metric import BillableMetricResponse
-
-
-class BillableMetricClient(
-    CreateCommandMixin[BillableMetricResponse],
-    DestroyCommandMixin[BillableMetricResponse],
-    FindAllCommandMixin[BillableMetricResponse],
-    FindCommandMixin[BillableMetricResponse],
-    UpdateCommandMixin[BillableMetricResponse],
-    BaseClient,
-):
-    API_RESOURCE: ClassVar[str] = "billable_metrics"
-    RESPONSE_MODEL: ClassVar[Type[BillableMetricResponse]] = BillableMetricResponse
-    ROOT_NAME: ClassVar[str] = "billable_metric"

lago_python_client-1.15.1/lago_python_client/version.py

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