u-toolkit 0.1.14__py3-none-any.whl → 0.1.17__py3-none-any.whl

u_toolkit/decorators.py

@@ -53,19 +53,25 @@ def define_method_handler(
 
 _P = ParamSpec("_P")
 
+_T2 = TypeVar("_T2")
+
 
 def create_decorator(
-    handle: Callable[_P] | None = None,
+    handle: Callable[_P, _T2] | None = None,
+    use_handle_return: bool | None = None,
 ):
     def decorator(fn: Callable[_P, _T]):
         if inspect.iscoroutinefunction(fn):
 
             @wraps(fn)
             async def async_wrapper(*args, **kwargs):
+                handle_result = None
                 if inspect.iscoroutinefunction(handle):
-                    await handle(*args, **kwargs)
+                    handle_result = await handle(*args, **kwargs)
                 elif handle:
-                    handle(*args, **kwargs)
+                    handle_result = handle(*args, **kwargs)
+                if use_handle_return:
+                    return handle_result
                 return await fn(*args, **kwargs)
 
             return async_wrapper
@@ -73,7 +79,10 @@ def create_decorator(
         @wraps(fn)
         def wrapper(*args: _P.args, **kwargs: _P.kwargs):
             if handle:
-                handle(*args, **kwargs)
+                handle_result = handle(*args, **kwargs)
+                if use_handle_return:
+                    return handle_result
+
             return fn(*args, **kwargs)
 
         return wrapper

u_toolkit/fastapi/cbv.py

@@ -163,21 +163,31 @@ class CBVRoutesInfo(TypedDict):
     deprecated: NotRequired[bool | None]
 
 
+CBVRoutesInfoT = TypeVar("CBVRoutesInfoT", bound=CBVRoutesInfo)
+
+
 class CBVRouteInfo(CBVRoutesInfo, Generic[_T]):
     methods: NotRequired[list[RequestMethod] | None]
     response_model: NotRequired[type[_T] | None]
     status: NotRequired[int | None]
+    summary: NotRequired[str | None]
+    description: NotRequired[str | None]
+    name: NotRequired[str | None]
 
 
-class CBV:
+class CBV(Generic[CBVRoutesInfoT]):
     def __init__(self, router: APIRouter | None = None) -> None:
         self.router = router or APIRouter()
 
-        self._state: dict[type, dict[_FnName, CBVRouteInfo]] = {}
-        self._cls_routes_extra: dict[
-            type, tuple[CBVRoutesInfo, Callable[[type[_T]], _T] | None]  # type: ignore
+        self.state: dict[type, dict[_FnName, CBVRouteInfo]] = {}
+        self.routes_extra: dict[
+            type,
+            tuple[
+                CBVRoutesInfoT | None,
+                Callable[[type[_T]], _T] | None,  # type: ignore
+            ],
         ] = {}
-        self._initialed_state: dict[type[_T], _T] = {}  # type: ignore
+        self.initialed_state: dict[type[_T], _T] = {}  # type: ignore
 
     def create_route(
         self,
@@ -187,41 +197,44 @@ class CBV:
         method: RequestMethod,
         method_name: str,
     ):
-        class_routes_info = self._cls_routes_extra[cls][0]
+        class_routes_info = self.routes_extra[cls][0] or {}
 
         class_tags = class_routes_info.get("tags") or []
         endpoint_tags: list[str | Enum] = (
-            self._state[cls][method_name].get("tags") or []
+            self.state[cls][method_name].get("tags") or []
         )
         tags = class_tags + endpoint_tags
 
         class_dependencies = class_routes_info.get("dependencies") or []
         endpoint_dependencies = (
-            self._state[cls][method_name].get("dependencies") or []
+            self.state[cls][method_name].get("dependencies") or []
         )
         dependencies = class_dependencies + endpoint_dependencies
 
         class_responses = class_routes_info.get("responses") or []
         endpoint_responses = (
-            self._state[cls][method_name].get("responses") or []
+            self.state[cls][method_name].get("responses") or []
         )
         responses = build_responses(*class_responses, *endpoint_responses)
 
-        status_code = self._state[cls][method_name].get("status")
+        status_code = self.state[cls][method_name].get("status")
 
-        deprecated = self._state[cls][method_name].get(
+        deprecated = self.state[cls][method_name].get(
             "deprecated", class_routes_info.get("deprecated")
         )
 
-        response_model = self._state[cls][method_name].get("response_model")
+        response_model = self.state[cls][method_name].get("response_model")
 
         endpoint_methods = [
             i.upper()
-            for i in (self._state[cls][method_name].get("methods") or [method])
+            for i in (self.state[cls][method_name].get("methods") or [method])
         ]
 
-        path = self._state[cls][method_name].get("path") or path
+        path = self.state[cls][method_name].get("path") or path
 
+        summary = self.state[cls][method_name].get("summary")
+        description = self.state[cls][method_name].get("description")
+        name = self.state[cls][method_name].get("name")
         return self.router.api_route(
             path,
             methods=endpoint_methods,
@@ -231,9 +244,12 @@ class CBV:
             responses=responses,
             status_code=status_code,
             deprecated=deprecated,
+            summary=summary,
+            description=description,
+            name=name,
         )
 
-    def info(
+    def info(  # noqa: PLR0913
         self,
         *,
         path: str | None = None,
@@ -242,10 +258,13 @@ class CBV:
         dependencies: list | None = None,
         responses: list[Response] | None = None,
         response_model: type[_T] | None = None,
+        summary: str | None = None,
+        description: str | None = None,
+        name: str | None = None,
         status: int | None = None,
         deprecated: bool | None = None,
     ):
-        state = self._state
+        state = self.state
         initial_state = self._initial_state
         data = CBVRouteInfo(
             path=path,
@@ -256,6 +275,9 @@ class CBV:
             response_model=response_model,
             status=status,
             deprecated=deprecated,
+            summary=summary,
+            description=description,
+            name=name,
         )
 
         def handle(params: DefineMethodParams):
@@ -268,22 +290,20 @@ class CBV:
         return define_method_handler(handle)
 
     def _initial_state(self, cls: type[_T]) -> _T:
-        if result := self._initialed_state.get(cls):
+        if result := self.initialed_state.get(cls):
             return cast(_T, result)
 
         default_data = {}
         for endpoint in iter_endpoints(cls):
             default_data[endpoint.original_handle_name] = {}
 
-        self._state.setdefault(cls, default_data)
+        self.state.setdefault(cls, default_data)
         result = self._build_cls(cls)
-        self._initialed_state[cls] = result
+        self.initialed_state[cls] = result
         return result
 
     def _build_cls(self, cls: type[_T]) -> _T:
-        if cls in self._cls_routes_extra and (
-            build := self._cls_routes_extra[cls][1]
-        ):
+        if cls in self.routes_extra and (build := self.routes_extra[cls][1]):
             return build(cls)  # type: ignore
         return cls()
 
@@ -366,24 +386,24 @@ class CBV:
     def __call__(
         self,
         *,
-        info: CBVRoutesInfo | None = None,
+        info: CBVRoutesInfoT | None = None,
         build: Callable[[type[_T]], _T] | None = None,
     ) -> Callable[[type[_T]], type[_T]]: ...
 
     def __call__(self, *args, **kwargs):
-        info: CBVRoutesInfo = {}
+        info = None
         build: Callable | None = None
 
         def decorator(cls: type[_T]) -> type[_T]:
             instance = self._initial_state(cls)
-            self._cls_routes_extra[cls] = info, build
+            self.routes_extra[cls] = info, build
 
             decorator = self.__create_class_dependencies_injector(cls)
 
             def valid_method(
                 name: str, _handle: Callable
             ) -> None | list[_MethodInfo]:
-                if (cls_state := self._state.get(cls)) and (
+                if (cls_state := self.state.get(cls)) and (
                     method_state := cls_state.get(name)
                 ):
                     methods: list[RequestMethod] = (
@@ -414,7 +434,7 @@ class CBV:
         if args:
             return decorator(args[0])
 
-        info.update(kwargs.get("info") or {})
+        info = kwargs.get("info") or None
         build = kwargs.get("build")
 
         return decorator

{u_toolkit-0.1.14.dist-info → u_toolkit-0.1.17.dist-info}/METADATA

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

{u_toolkit-0.1.14.dist-info → u_toolkit-0.1.17.dist-info}/RECORD

@@ -1,7 +1,7 @@
 u_toolkit/__init__.py,sha256=7f_bFg1UERA4gyh3HjCAjOZo2cBUgElpwgX5x-Xk2WA,238
 u_toolkit/alias_generators.py,sha256=KmPL1ViGJjptONffZUAX4ENvkldrwDylm_Ly5RYE8b4,963
 u_toolkit/datetime.py,sha256=GOG0xa6yKeqvFXJkImK5Pt7wsPXnHMlNKju8deniGJ4,644
-u_toolkit/decorators.py,sha256=3AQ_NMT7YfKDNFBX07chghktwhTBlIcTjAzjgVMmfNI,2082
+u_toolkit/decorators.py,sha256=rBZfXtWVfja6YMDqF9izjAiabRVyp4pAWOlwnN0HkUw,2396
 u_toolkit/enum.py,sha256=2yWmK8V1q0P5S3ltBED-TF2H09BmRbsBuqu3Th3G1NE,862
 u_toolkit/function.py,sha256=GAE6TIm5jpemUiNUPjzk7pIHrxwSSX7sHl1V5E735dE,252
 u_toolkit/helpers.py,sha256=K3FCz93K1nT4o7gWKVpMKy3k3BnirTPCSb8F8EFUrWk,112
@@ -11,7 +11,7 @@ u_toolkit/object.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 u_toolkit/path.py,sha256=IkyIHcU9hKBCOZfF30FrKf4CfL-MH91fjeYF9EY7eos,128
 u_toolkit/signature.py,sha256=-Q6n28PYBYYdd2OXBKESeVkL2rYpV6EaY3IVwQmzezQ,2161
 u_toolkit/fastapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-u_toolkit/fastapi/cbv.py,sha256=q9A4HKguOVBfLeZmhnY9kfiP6FfAViZUo52m-5aKZhU,12219
+u_toolkit/fastapi/cbv.py,sha256=vRYGeopzduO7Gmf5O5oz76zGOYqFtD8aGAe1oCcK8R4,12910
 u_toolkit/fastapi/config.py,sha256=kGpokR9XXr1KxMA1GVKYkCdKwqIQAIwOJ-v6sGbqzAQ,267
 u_toolkit/fastapi/exception.py,sha256=P9apEEQvbBMmffiHX57768gmemMeAy69-nb-1JQZAZE,4624
 u_toolkit/fastapi/helpers.py,sha256=BCMMLxa1c6BMA_rKq-hCi0iyEjrR3Z5rPMeTvgaVJB0,447
@@ -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.14.dist-info/METADATA,sha256=-mn37kFXx78jKuagV5ZFLhtjjvIV2wx3dfB3_EhWpjU,366
-u_toolkit-0.1.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-u_toolkit-0.1.14.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
-u_toolkit-0.1.14.dist-info/RECORD,,
+u_toolkit-0.1.17.dist-info/METADATA,sha256=uQ1VONPPf54w3xIyljJ9-j8Mz3YMQdes1CksSF26ZB8,366
+u_toolkit-0.1.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+u_toolkit-0.1.17.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
+u_toolkit-0.1.17.dist-info/RECORD,,