为什么在使用 class 作为装饰器时 'self' 不在 args 中？

Question

Class method decorator with self arguments? (direct link to answer: ) 的最佳答案描述了装饰器在装饰方法时如何访问对象的属性。

我试图将它应用到我自己的代码中，但意识到它仅在使用函数作为装饰器时才有效。出于某种原因，当使用 class 作为装饰器时，参数元组不再包含 self 对象。为什么会发生这种情况，还有其他方法可以访问包装方法的 self?

演示：

import functools


class DecoratorClass:
    def __init__(self, func):
        functools.update_wrapper(self, func)
        self.func = func                    
                        
    def __call__(self, *args, **kwargs):
        print("args:", args)            
        print("kwargs:", kwargs)
        print("Counting with:", args[0].name)
        self.func(*args, **kwargs)           
                                  

def decorator_func(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        print("args:", args)     
        print("kwargs:", kwargs)
        print("Counting with:", args[0].name)
        func(*args, **kwargs)                
    return wrapper           
                  

class Counter:
    def __init__(self, name, count):
        self.name = name            
        self.count = count
                          
    @decorator_func
    def count_a(self):
        for i in range(self.count):
            print(i)               
                    
    @DecoratorClass
    def count_b(self):
        for i in range(self.count):
            print(i)               
                    

c = Counter('my counter', 3)
                            
c.count_a()
print()    
c.count_b()

输出：

args: (<__main__.Counter object at 0x7f4491a21970>,)
kwargs: {}
Counting with: my counter
0
1
2

args: ()
kwargs: {}
Traceback (most recent call last):
  File "./test2.py", line 48, in <module>
    c.count_b()
  File "./test2.py", line 14, in __call__
    print("Counting with:", args[0].name)
IndexError: tuple index out of range

Answer 1

这是一个已知问题（请参阅）。我实际上运行在我的一个项目中实施 class 装饰器时遇到了同样的问题。

为了解决这个问题，我将以下方法添加到我的 class - 您也可以将其添加到您的 DecoratorClass，然后一切正常。

def __get__(self, instance, owner):
    """
    Fix: make our decorator class a decorator, so that it also works to
    decorate instance methods.
    
    """
    from functools import partial
    return partial(self.__call__, instance)

此外，请查看链接的 SO 答案以了解为什么会这样。

为什么在使用 class 作为装饰器时 'self' 不在 args 中？

Why is 'self' not in args when using a class as a decorator?

python

methods

decorator