u-toolkit 0.1.5__py3-none-any.whl → 0.1.6__py3-none-any.whl

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.5.dist-info → u_toolkit-0.1.6.dist-info}/METADATA

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

{u_toolkit-0.1.5.dist-info → u_toolkit-0.1.6.dist-info}/RECORD

@@ -13,7 +13,7 @@ u_toolkit/signature.py,sha256=-Q6n28PYBYYdd2OXBKESeVkL2rYpV6EaY3IVwQmzezQ,2161
 u_toolkit/fastapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 u_toolkit/fastapi/cbv.py,sha256=gtNhV8mQAEQNf9uIwyjlrA-ZbVwLeJenJ-f2G2Co1SE,10808
 u_toolkit/fastapi/config.py,sha256=kGpokR9XXr1KxMA1GVKYkCdKwqIQAIwOJ-v6sGbqzAQ,267
-u_toolkit/fastapi/exception.py,sha256=5E4wAJYwp0RJ4SEBVkchOgrgfwCgniQ8Mtg1O5sWUXE,3288
+u_toolkit/fastapi/exception.py,sha256=A9T21rZi7OR-gLeG1ZnFy8Xc8Sxz5Qq9doGAcZHkJ5k,4028
 u_toolkit/fastapi/helpers.py,sha256=BCMMLxa1c6BMA_rKq-hCi0iyEjrR3Z5rPMeTvgaVJB0,447
 u_toolkit/fastapi/lifespan.py,sha256=W1TwWymW7xtmntx59QBC4LQ6xQr3L7yuMMGj4U8hhTQ,1813
 u_toolkit/fastapi/pagination.py,sha256=yOgEDUT04m_mZ0cPliuDbUHLFnmxGAmr5PyZlwfjT_s,1940
@@ -30,7 +30,7 @@ u_toolkit/sqlalchemy/type_vars.py,sha256=m2VeV41CBIK_1QX3w2kgz-n556sILAGZ-Kaz3TD
 u_toolkit/sqlalchemy/orm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 u_toolkit/sqlalchemy/orm/fields.py,sha256=3zoYil23I6YLtc_59aHDt9w5l1NBTkePT9AfXI3DMiY,593
 u_toolkit/sqlalchemy/orm/models.py,sha256=V8vf4ps3phAmwxyaFYK7pw8Igz7h097o4QBjKB0gwC8,705
-u_toolkit-0.1.5.dist-info/METADATA,sha256=4JnI6t-zFqp7sGsXFTxSX_JvpnwspQVEmcXJ1PbHe_g,365
-u_toolkit-0.1.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-u_toolkit-0.1.5.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
-u_toolkit-0.1.5.dist-info/RECORD,,
+u_toolkit-0.1.6.dist-info/METADATA,sha256=aqibymMM4uylGpJTWYMXaaRHYYCurP7x1pMrAUqHiIM,365
+u_toolkit-0.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+u_toolkit-0.1.6.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
+u_toolkit-0.1.6.dist-info/RECORD,,