epoch - use pydantic-extra-types

Author: commonism

aiopenapi3/model.py

@@ -58,13 +58,16 @@ def generate_type_format_to_class():
 
     type_format_to_class["string"]["byte"] = Base64Str
 
-    from aiopenapi3.models import epoch
+    type_format_to_class["integer"][None] = int
 
-    type_format_to_class["number"]["date-time"] = epoch.Number
+    try:
+        from pydantic_extra_types import epoch
 
-    if "integer" not in type_format_to_class:
-        type_format_to_class["integer"][None] = int
-    type_format_to_class["integer"]["date-time"] = epoch.Integer
+        type_format_to_class["number"]["date-time"] = epoch.Number
+        type_format_to_class["integer"]["date-time"] = epoch.Integer
+
+    except ImportError:
+        pass
 
 
 def class_from_schema(s, _type):

pyproject.toml

@@ -54,7 +54,9 @@ auth = [
 socks = [
     "httpx-socks",
 ]
-
+types =[
+    "pydantic-extra-types @ git+https://github.com/pydantic/pydantic-extra-types.git@3668b3af8ab378c56342c613672aa9415dab865b",
+]
 [project.scripts]
 aiopenapi3 = "aiopenapi3.cli:main"
 
@@ -118,6 +120,7 @@ tests = [
     "bootstrap-flask",
     "ijson",
     "python-multipart>=0.0.6",
+    "pydantic-extra-types @ git+https://github.com/pydantic/pydantic-extra-types.git@3668b3af8ab378c56342c613672aa9415dab865b"
 ]
 
 [tool.pdm]