u-toolkit 0.1.4__tar.gz → 0.1.6__tar.gz

{u_toolkit-0.1.4 → u_toolkit-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: u-toolkit
-Version: 0.1.4
+Version: 0.1.6
 Summary: Add your description here
 Requires-Python: >=3.11
 Requires-Dist: pydantic>=2.11.3

{u_toolkit-0.1.4 → u_toolkit-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "u-toolkit"
-version = "0.1.4"
+version = "0.1.6"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"

{u_toolkit-0.1.4 → u_toolkit-0.1.6}/src/u_toolkit/fastapi/cbv.py

@@ -13,7 +13,11 @@ from u_toolkit.fastapi.helpers import get_depend_from_annotation, is_depend
 from u_toolkit.fastapi.responses import Response, build_responses
 from u_toolkit.helpers import is_annotated
 from u_toolkit.merge import deep_merge_dict
-from u_toolkit.signature import update_parameters, with_parameter
+from u_toolkit.signature import (
+    list_parameters,
+    update_parameters,
+    with_parameter,
+)
 
 
 class EndpointsClassInterface(Protocol):
@@ -304,7 +308,9 @@ class CBV:
             for name, dep in iter_dependencies(cls)
         ]
 
-        update_parameters(collect_cls_dependencies, *parameters)
+        has_cls_deps = bool(parameters)
+        if has_cls_deps:
+            update_parameters(collect_cls_dependencies, *parameters)
 
         def new_fn(method_name, kwargs):
             instance = self._build_cls(cls)
@@ -320,11 +326,14 @@ class CBV:
             sign_cls_fn = partial(cls_fn)
             update_wrapper(sign_cls_fn, cls_fn)
 
-            parameters, *_ = with_parameter(
-                sign_cls_fn,
-                name=collect_cls_dependencies.__name__,
-                default=Depends(collect_cls_dependencies),
-            )
+            if has_cls_deps:
+                parameters, *_ = with_parameter(
+                    sign_cls_fn,
+                    name=collect_cls_dependencies.__name__,
+                    default=Depends(collect_cls_dependencies),
+                )
+            else:
+                parameters = list_parameters(sign_cls_fn)
 
             update_parameters(sign_cls_fn, *(parameters[1:]))
 

{u_toolkit-0.1.4 → u_toolkit-0.1.6}/src/u_toolkit/fastapi/exception.py

@@ -1,12 +1,20 @@
 from collections.abc import Callable, Sequence
 from typing import Generic, Literal, Protocol, cast
 
-from fastapi import status
+from fastapi import Response, status
+from fastapi.responses import JSONResponse, ORJSONResponse
+from fastapi.utils import is_body_allowed_for_status_code
 from pydantic import BaseModel, create_model
 
 from u_toolkit.pydantic.type_vars import BaseModelT
 
 
+try:
+    import orjson
+except ImportError:  # pragma: nocover
+    orjson = None  # type: ignore
+
+
 class WrapperError(BaseModel, Generic[BaseModelT]):  # type: ignore
     @classmethod
     def create(
@@ -113,3 +121,23 @@ class NamedHTTPError(Exception, Generic[BaseModelT]):
     @classmethod
     def response_schema(cls):
         return {cls.status: {"model": cls.response_class()}}
+
+
+def named_http_error_handler(_, exc: NamedHTTPError):
+    headers = exc.headers
+
+    if not is_body_allowed_for_status_code(exc.status):
+        return Response(status_code=exc.status, headers=headers)
+
+    if orjson:
+        return ORJSONResponse(
+            exc.data.model_dump(exclude_none=True),
+            status_code=exc.status,
+            headers=headers,
+        )
+
+    return JSONResponse(
+        exc.data.model_dump(exclude_none=True),
+        status_code=exc.status,
+        headers=headers,
+    )

{u_toolkit-0.1.4 → u_toolkit-0.1.6}/src/u_toolkit/signature.py

@@ -1,6 +1,5 @@
 import inspect
 from collections.abc import Callable, Sequence
-from functools import partial
 from typing import Annotated, Any, overload
 
 
@@ -57,9 +56,9 @@ def update_signature(
     return_annotation: type | None = None,
 ):
     signature = inspect.signature(fn)
-    if parameters:
+    if parameters is not None:
         signature = signature.replace(parameters=parameters)
-    if return_annotation:
+    if return_annotation is not None:
         signature = signature.replace(return_annotation=return_annotation)
 
     setattr(fn, "__signature__", signature)