tracktolib 0.38.1__py3-none-any.whl → 0.39.1__py3-none-any.whl

tracktolib/api.py

@@ -13,6 +13,7 @@ from .utils import json_serial, to_camel_case
 try:
     from fastapi import params, APIRouter
     from fastapi.responses import JSONResponse
+    import pydantic
     from pydantic import BaseModel
     import starlette.status
 except ImportError:
@@ -53,7 +54,7 @@ class MethodMeta(TypedDict):
     dependencies: Dependencies
     path: str | None
     response_model: Type[BaseModel | None | Sequence[BaseModel]] | None
-
+    openapi_extra: dict[str, Any] | None
 
 @dataclass
 class Endpoint:
@@ -67,52 +68,62 @@ class Endpoint:
     def get(self, status_code: StatusCode = None,
             dependencies: Dependencies = None,
             path: str | None = None,
-            model: Type[B] | None = None):
+            model: Type[B] | None = None,
+            openapi_extra: dict[str, Any] | None = None):
         return _get_method_wrapper(cls=self, method='GET',
                                    status_code=status_code,
                                    dependencies=dependencies,
                                    path=path,
-                                   model=model)
+                                   model=model,
+                                   openapi_extra=openapi_extra)
 
     def post(self, *, status_code: StatusCode = None,
              dependencies: Dependencies = None,
              path: str | None = None,
-             model: Type[B] | None = None):
+             model: Type[B] | None = None,
+             openapi_extra: dict[str, Any] | None = None):
         return _get_method_wrapper(cls=self, method='POST',
                                    status_code=status_code,
                                    dependencies=dependencies,
                                    path=path,
-                                   model=model)
+                                   model=model,
+                                   openapi_extra=openapi_extra)
 
     def put(self, status_code: StatusCode = None,
             dependencies: Dependencies = None,
             path: str | None = None,
-            model: Type[B] | None = None):
+            model: Type[B] | None = None,
+            openapi_extra: dict[str, Any] | None = None):
         return _get_method_wrapper(cls=self, method='PUT',
                                    status_code=status_code,
                                    dependencies=dependencies,
                                    path=path,
-                                   model=model)
+                                   model=model,
+                                   openapi_extra=openapi_extra)
 
     def delete(self, status_code: StatusCode = None,
                dependencies: Dependencies = None,
                path: str | None = None,
-               model: Type[B] | None = None):
+               model: Type[B] | None = None,
+               openapi_extra: dict[str, Any] | None = None):
         return _get_method_wrapper(cls=self, method='DELETE',
                                    status_code=status_code,
                                    dependencies=dependencies,
                                    path=path,
-                                   model=model)
+                                   model=model,
+                                   openapi_extra=openapi_extra)
 
     def patch(self, status_code: StatusCode = None,
               dependencies: Dependencies = None,
               path: str | None = None,
-              model: Type[B] | None = None):
+              model: Type[B] | None = None,
+              openapi_extra: dict[str, Any] | None = None):
         return _get_method_wrapper(cls=self, method='PATCH',
                                    status_code=status_code,
                                    dependencies=dependencies,
                                    path=path,
-                                   model=model)
+                                   model=model,
+                                   openapi_extra=openapi_extra)
 
 
 def _get_method_wrapper(cls: Endpoint, method: Method,
@@ -120,14 +131,16 @@ def _get_method_wrapper(cls: Endpoint, method: Method,
                         status_code: StatusCode = None,
                         dependencies: Dependencies = None,
                         path: str | None = None,
-                        model: Type[B] | None = None):
+                        model: Type[B] | None = None,
+                        openapi_extra: dict[str, Any] | None = None):
     def _set_method_wrapper(func: EnpointFn):
         _meta: MethodMeta = {
             'fn': func,
             'status_code': status_code,
             'dependencies': dependencies,
             'path': path,
-            'response_model': model
+            'response_model': model,
+            'openapi_extra': openapi_extra
         }
         cls._methods[method] = _meta
 
@@ -169,7 +182,8 @@ def add_endpoint(path: str,
                              name=getdoc(_fn),
                              response_model=_response_model,
                              status_code=_status_code,
-                             dependencies=[*(_dependencies or []), *(dependencies or [])])
+                             dependencies=[*(_dependencies or []), *(dependencies or [])],
+                             openapi_extra=_meta.get('openapi_extra'))
 
 
 @dataclass(init=False)
@@ -187,11 +201,15 @@ class JSONSerialResponse(JSONResponse):
         ).encode("utf-8")
 
 
-class CamelCaseModel(BaseModel):
-    class Config:
-        alias_generator = to_camel_case
-        allow_population_by_field_name = True
-
+if pydantic.__version__ < '2.0.0':
+    class CamelCaseModel(BaseModel): # type: ignore
+        class Config:
+            alias_generator = to_camel_case
+            allow_population_by_field_name = True
+else:
+    class CamelCaseModel(BaseModel):
+        def __init__(self):
+            raise NotImplementedError('Please use pydantic < 2.0.0')
 
 def check_status(resp, status: int = starlette.status.HTTP_200_OK):
     assert resp.status_code == status, json.dumps(resp.json(), indent=4)

{tracktolib-0.38.1.dist-info → tracktolib-0.39.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tracktolib
-Version: 0.38.1
+Version: 0.39.1
 Summary: Utility library for python
 Home-page: https://github.com/tracktor/tracktolib
 License: MIT

{tracktolib-0.38.1.dist-info → tracktolib-0.39.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
 tracktolib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tracktolib/api.py,sha256=2hAtq7Cjx7GqiqO2ATJfiCK3rM66kDfLKMBVZ_I8ttU,6743
+tracktolib/api.py,sha256=RWXlSeARebJKI3PP5j5HC1l3XXcGok7QFwEnWB1ZDrU,7803
 tracktolib/http.py,sha256=GHWvD9rHTF1oAnG6Ptna_oeMDcwCxYYpPxLOME-oZ3Y,1574
 tracktolib/logs.py,sha256=nbI9LO2NaXDAv9aoUkad54V_x8B8zga-MwqUfYb-pkk,2523
 tracktolib/pg/__init__.py,sha256=Fb56_-_9MftglwiC99kWU1-_IMNJjh7WFkVWBK60U_U,297
@@ -13,7 +13,7 @@ tracktolib/s3/minio.py,sha256=gkpI2bvVwfXaAFY7M9e0Mlv82x0PjuriS8LdRmEJtMo,1258
 tracktolib/s3/s3.py,sha256=4AUtdkFO-T6Zx0cpo1Hw6yl31Namdbt6DnDVuF2M-sQ,4059
 tracktolib/tests.py,sha256=GycCVRUOdzazxqBXKBWWzhrlkV8yyD7NaActz_A2WXU,590
 tracktolib/utils.py,sha256=kU35LCJaFDNg0IR9ekAi4uEJ3unZCC3VPzqgkctt06Q,5437
-tracktolib-0.38.1.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
-tracktolib-0.38.1.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
-tracktolib-0.38.1.dist-info/METADATA,sha256=9ncsbslfc85lQxE-nFrexezfnhL87xCjqf0bp8xDaRg,3563
-tracktolib-0.38.1.dist-info/RECORD,,
+tracktolib-0.39.1.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.39.1.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
+tracktolib-0.39.1.dist-info/METADATA,sha256=W1pdt_tOL2AJ9AAfdP-mehFvK3EfmeocAv1F9lfLd9I,3563
+tracktolib-0.39.1.dist-info/RECORD,,