u-toolkit 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

u_toolkit/fastapi/exception.py

@@ -1,5 +1,6 @@
+import inspect
 from collections.abc import Callable, Sequence
-from typing import Generic, Literal, Protocol, cast
+from typing import Any, Generic, Literal, Protocol, TypeVar, cast
 
 from fastapi import Response, status
 from fastapi.responses import JSONResponse, ORJSONResponse
@@ -10,12 +11,12 @@ from u_toolkit.pydantic.type_vars import BaseModelT
 
 
 try:
-    import orjson
+    import orjson  # type: ignore
 except ImportError:  # pragma: nocover
     orjson = None  # type: ignore
 
 
-class WrapperError(BaseModel, Generic[BaseModelT]):  # type: ignore
+class WrapperError(BaseModel, Generic[BaseModelT]):
     @classmethod
     def create(
         cls: type["WrapperError[BaseModelT]"],
@@ -24,7 +25,10 @@ class WrapperError(BaseModel, Generic[BaseModelT]):  # type: ignore
         raise NotImplementedError
 
 
-class EndpointError(WrapperError, Generic[BaseModelT]):
+WrapperErrorT = TypeVar("WrapperErrorT", bound=WrapperError)
+
+
+class EndpointError(WrapperError[BaseModelT], Generic[BaseModelT]):
     error: BaseModelT
 
     @classmethod
@@ -39,13 +43,17 @@ class HTTPErrorInterface(Protocol):
     def response_class(cls) -> type[BaseModel]: ...
 
 
-class NamedHTTPError(Exception, Generic[BaseModelT]):
+class NamedHTTPError(Exception, Generic[WrapperErrorT, BaseModelT]):
     status: int = status.HTTP_400_BAD_REQUEST
     code: str | None = None
-    targets: Sequence[str] | None = None
-    target_transform: Callable[[str], str] | None = None
+    targets: Sequence[Any] | None = None
+    target_transform: Callable[[Any], Any] | None = None
     message: str | None = None
-    wrapper_class: type[WrapperError[BaseModelT]] | None = EndpointError
+    wrapper: (
+        type[WrapperError[BaseModelT]]
+        | tuple[WrapperErrorT, Callable[[BaseModelT], WrapperErrorT]]
+        | None
+    ) = None
 
     @classmethod
     def error_name(cls):
@@ -87,7 +95,7 @@ class NamedHTTPError(Exception, Generic[BaseModelT]):
         target: str | None = None,
         headers: dict[str, str] | None = None,
     ) -> None:
-        kwargs = {
+        kwargs: dict[str, Any] = {
             "code": self.error_code(),
             "message": message or "operation failed",
         }
@@ -99,10 +107,13 @@ class NamedHTTPError(Exception, Generic[BaseModelT]):
             kwargs["message"] = kwargs["message"].format(target=target)
 
         self.model = self.model_class()(**kwargs)
+        create: Callable[[BaseModelT], BaseModel] | None = None
+        if inspect.isclass(self.wrapper):
+            create = self.wrapper.create
+        elif isinstance(self.wrapper, tuple):
+            create = self.wrapper[1]
         self.data: BaseModel = (
-            self.wrapper_class.create(self.model)
-            if self.wrapper_class is not None
-            else self.model
+            create(self.model) if create is not None else self.model
         )
 
         self.headers = headers
@@ -116,7 +127,16 @@ class NamedHTTPError(Exception, Generic[BaseModelT]):
     @classmethod
     def response_class(cls):
         model = cls.model_class()
-        return cls.wrapper_class if cls.wrapper_class is not None else model
+
+        if cls.wrapper:
+            wrapper: Any
+            if inspect.isclass(cls.wrapper):
+                wrapper = cls.wrapper
+            else:
+                wrapper = cls.wrapper[0]
+            return wrapper[model]
+
+        return model
 
     @classmethod
     def response_schema(cls):

{u_toolkit-0.1.7.dist-info → u_toolkit-0.1.9.dist-info}/METADATA

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

{u_toolkit-0.1.7.dist-info → u_toolkit-0.1.9.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=ON2yx7sIIlwvlk6VtBWL4iXOioL6qCZC7rlNFn_et0g,4064
+u_toolkit/fastapi/exception.py,sha256=P9apEEQvbBMmffiHX57768gmemMeAy69-nb-1JQZAZE,4624
 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.7.dist-info/METADATA,sha256=kfRXER25SA83kSQZagcTTNWSSm40zu5xA6w83cMBIeQ,365
-u_toolkit-0.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-u_toolkit-0.1.7.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
-u_toolkit-0.1.7.dist-info/RECORD,,
+u_toolkit-0.1.9.dist-info/METADATA,sha256=5Cqv8wacCiIyr68zBiVWhghEUxDQTOYP848G52cGaq0,365
+u_toolkit-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+u_toolkit-0.1.9.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
+u_toolkit-0.1.9.dist-info/RECORD,,