u-toolkit 0.1.5__tar.gz → 0.1.7__tar.gz

{u_toolkit-0.1.5 → u_toolkit-0.1.7}/PKG-INFO

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

{u_toolkit-0.1.5 → u_toolkit-0.1.7}/pyproject.toml

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

{u_toolkit-0.1.5 → u_toolkit-0.1.7}/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(
@@ -75,13 +83,13 @@ class NamedHTTPError(Exception, Generic[BaseModelT]):
     def __init__(
         self,
         *,
-        message: str,
+        message: str | None = None,
         target: str | None = None,
         headers: dict[str, str] | None = None,
     ) -> None:
         kwargs = {
             "code": self.error_code(),
-            "message": message,
+            "message": message or "operation failed",
         }
 
         if target:
@@ -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,
+    )