From a98f95f8644e051e4ce5e752d77f575220f07866 Mon Sep 17 00:00:00 2001 From: Joostlek Date: Fri, 21 Jul 2023 16:55:34 +0200 Subject: [PATCH] Use pydantic shims --- src/python_opensky/models.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/python_opensky/models.py b/src/python_opensky/models.py index 8ca6c75..b572c67 100644 --- a/src/python_opensky/models.py +++ b/src/python_opensky/models.py @@ -3,20 +3,23 @@ from dataclasses import dataclass -from pydantic import BaseModel, Field +try: + from pydantic.v1 import BaseModel, Field +except ImportError: + from pydantic import BaseModel, Field from .const import AircraftCategory, PositionSource from .exceptions import OpenSkyCoordinateError -class StatesResponse(BaseModel): +class StatesResponse(BaseModel): # type: ignore[misc] """Represents the states response.""" states: list[StateVector] = Field(...) time: int = Field(...) -class StateVector(BaseModel): +class StateVector(BaseModel): # type: ignore[misc] """Represents the state of a vehicle at a particular time. Attributes