mirror of
https://github.com/Steffo99/async-chain.git
synced 2024-11-25 06:24:22 +00:00
20 lines
454 B
Python
20 lines
454 B
Python
import functools
|
|
from .chain import ChainStart
|
|
|
|
|
|
class FunctionWrapper:
|
|
def __init__(self, func):
|
|
self._func = func
|
|
|
|
def __repr__(self):
|
|
return self._func.__name__
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
return self._func(*args, **kwargs)
|
|
|
|
|
|
def method_deco(func):
|
|
@functools.wraps(func)
|
|
def decorated(*args, **kwargs):
|
|
return ChainStart(start=FunctionWrapper(func))(*args, **kwargs)
|
|
return decorated
|