feat: add support for pydantic v2 (#227)

feat: add support for pydantic v2 backwards compatible import

misc: update install requirements allow pydantic v2.*

feat: update BaseModel imports

lago_python_client/base_client.py

@@ -1,7 +1,7 @@
 from abc import ABC, abstractmethod
 from typing import Type
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 
 class BaseClient(ABC):

lago_python_client/base_model.py

@@ -1,4 +1,7 @@
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel  # type: ignore
+except ImportError:
+    from pydantic import BaseModel
 
 
 class BaseResponseModel(BaseModel):

lago_python_client/events/clients.py

@@ -1,7 +1,7 @@
 import sys
 from typing import Any, ClassVar, Optional, Type
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_client import BaseClient
 from ..fees.clients import FeeClient

lago_python_client/mixins.py

@@ -8,7 +8,7 @@
 except ImportError:  # Python 3.7
     from typing_extensions import Protocol  # type: ignore
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .services.json import to_json
 from .services.request import make_headers, make_url, send_delete_request, send_get_request, send_post_request, send_put_request

lago_python_client/models/add_on.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/applied_coupon.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .credit import CreditsResponse
 from ..base_model import BaseResponseModel

lago_python_client/models/billable_metric.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/charge.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .tax import TaxesResponse
 from ..base_model import BaseResponseModel
@@ -63,4 +63,4 @@ class ChargeOverrides(BaseModel):
 
 
 class ChargesOverrides(BaseModel):
-    __root__: List[ChargeOverrides]
+    __root__: List[ChargeOverrides]

lago_python_client/models/coupon.py

@@ -1,6 +1,6 @@
 from typing import Any, List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/credit.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .invoice_item import InvoiceItemResponse
 from ..base_model import BaseResponseModel

lago_python_client/models/credit_note.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .fee import FeeResponse
 from ..base_model import BaseResponseModel

lago_python_client/models/customer.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/customer_usage.py

@@ -1,6 +1,6 @@
 from typing import Dict, List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/event.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 
@@ -30,4 +30,4 @@ class EventResponse(BaseResponseModel):
     created_at: str
 
 class BatchEventResponse(BaseResponseModel):
-    events: List[EventResponse]
+    events: List[EventResponse]

lago_python_client/models/group.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/invoice.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .credit import CreditsResponse
 from .customer import CustomerResponse

lago_python_client/models/organization.py

@@ -1,6 +1,6 @@
 from typing import Optional, List
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/plan.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from .charge import Charges, ChargesResponse, ChargesOverrides
 from .tax import Taxes, TaxesResponse
@@ -48,4 +48,4 @@ class PlanOverrides(BaseModel):
     amount_currency: Optional[str]
     trial_period: Optional[float]
     charges: Optional[ChargesOverrides]
-    tax_codes: Optional[List[str]]
+    tax_codes: Optional[List[str]]

lago_python_client/models/subscription.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 from .plan import PlanOverrides
 from ..base_model import BaseResponseModel
 

lago_python_client/models/tax.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/wallet.py

@@ -1,6 +1,6 @@
 from typing import List, Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/wallet_transaction.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/models/webhook_endpoint.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_model import BaseResponseModel
 

lago_python_client/services/response.py

@@ -11,7 +11,7 @@
     from typing_extensions import Final  # type: ignore
 
 from httpx import Response as Response  # not a typo! implicit reexport
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 import typeguard
 
 from ..exceptions import LagoApiError

lago_python_client/wallets/clients.py

@@ -1,7 +1,7 @@
 import sys
 from typing import Any, ClassVar, Type, Union
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_client import BaseClient
 from ..mixins import CreateCommandMixin, DestroyCommandMixin, FindAllCommandMixin, FindCommandMixin, UpdateCommandMixin

lago_python_client/webhook_endpoints/clients.py

@@ -1,7 +1,7 @@
 import sys
 from typing import Any, ClassVar, Type, Union
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 
 from ..base_client import BaseClient
 from ..mixins import CreateCommandMixin, DestroyCommandMixin, FindAllCommandMixin, FindCommandMixin, UpdateCommandMixin

lago_python_client/webhooks/clients.py

@@ -7,7 +7,7 @@
 except ImportError:
     from typing_extensions import TypedDict
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 import typeguard
 
 from ..base_client import BaseClient

setup.cfg

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

tests/test_response_services.py

@@ -1,7 +1,7 @@
 """Test response services."""
 from copy import deepcopy
 
-from pydantic import BaseModel
+from lago_python_client.base_model import BaseModel
 import pytest
 from httpx import Request, Response