Move common utility function parse_datetime into utils module

Author: nlopes

duffel_api/models/offer.py

@@ -3,7 +3,7 @@
 from typing import Optional, Sequence
 
 from duffel_api.models import Aircraft, Airline, Airport, LoyaltyProgrammeAccount, Place
-from duffel_api.utils import get_and_transform
+from duffel_api.utils import get_and_transform, parse_datetime
 
 
 @dataclass
@@ -499,12 +499,3 @@ def from_json(cls, json: dict):
             total_currency=json["total_currency"],
             total_emissions_kg=json["total_emissions_kg"],
         )
-
-
-def parse_datetime(value: str) -> datetime:
-    # There are inconsistent formats used for this field depending on the
-    # endpoint
-    if len(value) == 20:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
-    else:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")

duffel_api/models/order_cancellation.py

@@ -2,7 +2,7 @@
 from datetime import datetime
 from typing import Optional
 
-from duffel_api.utils import get_and_transform
+from duffel_api.utils import get_and_transform, parse_datetime
 
 
 @dataclass
@@ -57,12 +57,3 @@ def from_json(cls, json: dict):
             confirmed_at=get_and_transform(json, "confirmed_at", parse_datetime),
             created_at=datetime.strptime(json["created_at"], "%Y-%m-%dT%H:%M:%S.%fZ"),
         )
-
-
-def parse_datetime(value: str) -> datetime:
-    # There are inconsistent formats used for this field depending on the
-    # endpoint
-    if len(value) == 20:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
-    else:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")

duffel_api/models/order_change.py

@@ -3,7 +3,7 @@
 from typing import Optional, Sequence
 
 from duffel_api.models import Aircraft, Airline, Place
-from duffel_api.utils import get_and_transform
+from duffel_api.utils import get_and_transform, parse_datetime
 
 
 @dataclass
@@ -164,12 +164,3 @@ def from_json(cls, json: dict):
             created_at=datetime.strptime(json["created_at"], "%Y-%m-%dT%H:%M:%S.%fZ"),
             slices=OrderChangeSlices.from_json(json["slices"]),
         )
-
-
-def parse_datetime(value: str) -> datetime:
-    # There are inconsistent formats used for this field depending on the
-    # endpoint
-    if len(value) == 20:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
-    else:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")

duffel_api/models/order_change_offer.py

@@ -3,7 +3,7 @@
 from typing import Optional, Sequence
 
 from duffel_api.models import Aircraft, Airline, Place, Airport
-from duffel_api.utils import get_and_transform
+from duffel_api.utils import get_and_transform, parse_datetime
 
 
 @dataclass
@@ -175,12 +175,3 @@ def from_json(cls, json: dict):
             expires_at=parse_datetime(json["expires_at"]),
             slices=OrderChangeOfferSlices.from_json(json["slices"]),
         )
-
-
-def parse_datetime(value: str) -> datetime:
-    # There are inconsistent formats used for this field depending on the
-    # endpoint
-    if len(value) == 20:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
-    else:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")

duffel_api/models/payment_intent.py

@@ -3,7 +3,7 @@
 from typing import Optional, Sequence
 
 from duffel_api.models import Refund
-from duffel_api.utils import get_and_transform
+from duffel_api.utils import get_and_transform, parse_datetime
 
 
 @dataclass
@@ -62,12 +62,3 @@ def from_json(cls, json: dict):
             created_at=datetime.strptime(json["created_at"], "%Y-%m-%dT%H:%M:%S.%fZ"),
             updated_at=datetime.strptime(json["updated_at"], "%Y-%m-%dT%H:%M:%S.%fZ"),
         )
-
-
-def parse_datetime(value: str) -> datetime:
-    # There are inconsistent formats used for this field depending on the
-    # endpoint
-    if len(value) == 20:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
-    else:
-        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")

duffel_api/utils.py

@@ -1,4 +1,5 @@
 """Assorted auxiliary functions"""
+from datetime import datetime
 from typing import Any
 
 
@@ -25,3 +26,13 @@ def version() -> str:
     import pkg_resources
 
     return pkg_resources.require("duffel_api")[0].version
+
+
+def parse_datetime(value: str) -> datetime:
+    """Parse a datetime string regardless of having milliseconds or not"""
+    # There are inconsistent formats used for the field, therefore we try to accomodate
+    # instead of making an API breaking change.
+    if len(value) == 20:
+        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%SZ")
+    else:
+        return datetime.strptime(value, "%Y-%m-%dT%H:%M:%S.%fZ")