add billing_entities support (#306)

Author: Anna Velentsevich

lago_python_client/models/customer.py

@@ -120,6 +120,7 @@ class Customer(BaseModel):
     tax_codes: Optional[List[str]]
     skip_invoice_custom_sections: Optional[bool]
     invoice_custom_section_codes: Optional[List[str]]
+    billing_entity_code: Optional[str]
 
 
 class CustomerResponse(BaseResponseModel):
@@ -156,3 +157,4 @@ class CustomerResponse(BaseResponseModel):
     taxes: Optional[TaxesResponse]
     skip_invoice_custom_sections: Optional[bool]
     applicable_invoice_custom_sections: Optional[InvoiceCustomSectionsResponseList]
+    billing_entity_code: Optional[str]

lago_python_client/models/fee.py

@@ -59,6 +59,7 @@ class FeeResponse(BaseResponseModel):
     from_date: Optional[str]
     to_date: Optional[str]
     amount_details: Optional[Dict[str, Any]]
+    billing_entity_code: Optional[str]
 
     item: Optional[InvoiceItemResponse]
     applied_taxes: Optional[FeeAppliedTaxes]

lago_python_client/models/invoice.py

@@ -139,3 +139,4 @@ class InvoiceResponse(BaseResponseModel):
     applied_invoice_custom_sections: Optional[InvoiceAppliedInvoiceCustomSections]
     applied_usage_thresholds: Optional[InvoiceAppliedUsageThresholds]
     error_details: Optional[ErrorDetailsResponse]
+    billing_entity_code: Optional[str]

lago_python_client/version.py

@@ -1 +1 @@
-LAGO_VERSION = "1.22.0"
+LAGO_VERSION = "1.23.0"

tests/fixtures/customer.json

@@ -2,6 +2,7 @@
   "customer": {
     "lago_id": "99a6094e-199b-4101-896a-54e927ce7bd7",
     "external_id": "5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba",
+    "billing_entity_code": "test-company",
     "address_line1": "5230 Penfield Ave",
     "address_line2": null,
     "city": "Woodland Hills",
@@ -33,7 +34,10 @@
       "provider_customer_id": "cus_12345",
       "sync_with_provider": true,
       "document_locale": "fr",
-      "provider_payment_methods": ["card", "sepa_debit"]
+      "provider_payment_methods": [
+        "card",
+        "sepa_debit"
+      ]
     },
     "shipping_address": {
       "city": "Paris",

tests/fixtures/fee.json

@@ -1,6 +1,7 @@
 {
   "fee": {
     "lago_id": "5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba",
+    "billing_entity_code": "test-company",
     "lago_charge_id": "1a901a90-1a90-1a90-1a90-1a901a901a90",
     "lago_charge_filter_id": "1a2d9c8d-5875-4688-9854-5ccfd414bc5e",
     "lago_invoice_id": "1a2d9c8d-5875-4688-9854-5ccfd414bc5e",

tests/fixtures/invoice.json

@@ -1,6 +1,7 @@
 {
   "invoice": {
     "lago_id": "5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba",
+    "billing_entity_code": "test-company",
     "sequential_id": 15,
     "number": "LAG-1234-001-002",
     "issuing_date": "2022-06-02",
@@ -139,7 +140,9 @@
           "name": "User Seats",
           "invoice_display_name": "charge_invoice_display_name",
           "filters": {
-            "region": ["us-east-1"]
+            "region": [
+              "us-east-1"
+            ]
           }
         },
         "amount_cents": 100,

tests/test_customer_client.py

@@ -30,6 +30,7 @@ def create_customer():
 
     return Customer(
         external_id="5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba",
+        billing_entity_code="test-company",
         name="Gavin Belson",
         currency="EUR",
         tax_identification_number="EU123456789",
@@ -74,6 +75,7 @@ def test_valid_create_customers_request(httpx_mock: HTTPXMock):
     response = client.customers.create(create_customer())
 
     assert response.external_id == "5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba"
+    assert response.billing_entity_code == "test-company"
     assert response.name == "Gavin Belson"
     assert response.email == "[email protected]"
     assert response.currency == "EUR"