u-toolkit 0.1.13__py3-none-any.whl → 0.1.15__py3-none-any.whl

u_toolkit/decorators.py

@@ -1,7 +1,7 @@
 import inspect
 from collections.abc import Callable
 from functools import wraps
-from typing import Generic, NamedTuple, TypeVar
+from typing import Generic, NamedTuple, ParamSpec, TypeVar
 
 
 _FnT = TypeVar("_FnT", bound=Callable)
@@ -49,3 +49,42 @@ def define_method_handler(
             handle(params)
 
     return Decorator
+
+
+_P = ParamSpec("_P")
+
+_T2 = TypeVar("_T2")
+
+
+def create_decorator(
+    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):
+                    handle_result = await handle(*args, **kwargs)
+                elif handle:
+                    handle_result = handle(*args, **kwargs)
+                if use_handle_return:
+                    return handle_result
+                return await fn(*args, **kwargs)
+
+            return async_wrapper
+
+        @wraps(fn)
+        def wrapper(*args: _P.args, **kwargs: _P.kwargs):
+            if handle:
+                handle_result = handle(*args, **kwargs)
+                if use_handle_return:
+                    return handle_result
+
+            return fn(*args, **kwargs)
+
+        return wrapper
+
+    return decorator

{u_toolkit-0.1.13.dist-info → u_toolkit-0.1.15.dist-info}/METADATA

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

{u_toolkit-0.1.13.dist-info → u_toolkit-0.1.15.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=1drP4aNDpGxoEGDdnJyWP3lIeMBPPfeWEUrG0J8CpzE,1330
+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
@@ -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.13.dist-info/METADATA,sha256=CumVE1W_bIzTbM-fYPfOcaBFmEcT1rgWVwXF288NuGQ,366
-u_toolkit-0.1.13.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-u_toolkit-0.1.13.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
-u_toolkit-0.1.13.dist-info/RECORD,,
+u_toolkit-0.1.15.dist-info/METADATA,sha256=9YW-ZY96UyD1LCdAKjIGAVWbvOM8mqaCvcP-152uIVU,366
+u_toolkit-0.1.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+u_toolkit-0.1.15.dist-info/entry_points.txt,sha256=hTfAYCd5vvRiqgnJk2eBsoRIiIVB9pK8WZm3Q3jjKFU,45
+u_toolkit-0.1.15.dist-info/RECORD,,