Идеи, предложенные здесь, превосходны, но имеют некоторые недостатки:
inspect.getouterframes и args[0].__class__.__name__ не подходят для простых функций и статических методов. __get__ должен быть в классе, который отклоняется @wraps. @wraps сама должна лучше скрывать следы.
Итак, я объединил некоторые идеи из этой страницы, ссылки, документы и мою собственную голову,
и, наконец, нашел решение, в котором отсутствуют все три вышеупомянутых недостатка.
В результате method_decorator:
- Знает класс, к которому привязан декорированный метод.
- Скрывает следы декоратора, отвечая на системные атрибуты более правильно, чем
functools.wraps().
- Покрыт юнит-тестами для привязки несвязанных методов экземпляра, методов класса, статических методов и простых функций.
Использование:
pip install method_decorator
from method_decorator import method_decorator
class my_decorator(method_decorator):
# ...
См. полные юнит-тесты для деталей использования .
А вот код класса method_decorator:
class method_decorator(object):
def __init__(self, func, obj=None, cls=None, method_type='function'):
# These defaults are OK for plain functions
# and will be changed by __get__() for methods once a method is dot-referenced.
self.func, self.obj, self.cls, self.method_type = func, obj, cls, method_type
def __get__(self, obj=None, cls=None):
# It is executed when decorated func is referenced as a method: cls.func or obj.func.
if self.obj == obj and self.cls == cls:
return self # Use the same instance that is already processed by previous call to this __get__().
method_type = (
'staticmethod' if isinstance(self.func, staticmethod) else
'classmethod' if isinstance(self.func, classmethod) else
'instancemethod'
# No branch for plain function - correct method_type for it is already set in __init__() defaults.
)
return object.__getattribute__(self, '__class__')( # Use specialized method_decorator (or descendant) instance, don't change current instance attributes - it leads to conflicts.
self.func.__get__(obj, cls), obj, cls, method_type) # Use bound or unbound method with this underlying func.
def __call__(self, *args, **kwargs):
return self.func(*args, **kwargs)
def __getattribute__(self, attr_name): # Hiding traces of decoration.
if attr_name in ('__init__', '__get__', '__call__', '__getattribute__', 'func', 'obj', 'cls', 'method_type'): # Our known names. '__class__' is not included because is used only with explicit object.__getattribute__().
return object.__getattribute__(self, attr_name) # Stopping recursion.
# All other attr_names, including auto-defined by system in self, are searched in decorated self.func, e.g.: __module__, __class__, __name__, __doc__, im_*, func_*, etc.
return getattr(self.func, attr_name) # Raises correct AttributeError if name is not found in decorated self.func.
def __repr__(self): # Special case: __repr__ ignores __getattribute__.
return self.func.__repr__()