在Python中調用類變量的裝飾器

Question

可能我以錯誤的方式接近問題，並有一個更簡單的解決方案，但這是我的問題。

我有這樣定義的裝飾：

def my_decorator(argument): 
    def wrap(f): 
     def wrapped_f(*args, **kwargs): 
      .... # does something with argument 
     return wrapped_f 
    return wrap 

然後我定義一個類，它是這樣的：因爲在裝飾水平

class MyClass(object): 
    def __init__(argument): 
     self.argument = argument 

    @my_decorator(self.argument) # !this is an error! 
    def my_method(self): 
     .... # does something 

顯然我寫的是錯了，我無法訪問self，但解決此問題的正確方法是什麼？

Answer 1

你可以有裝飾接入self.argument內wrapped_f：

def mydecorator(f): 
    def wrapped_f(self, *args, **kwargs): 
     argument = self.argument 
     .... # does something with argument 
     return f(self, *args, **kwargs) 
    return wrapped_f 
return wrap 

如果變量被作爲視功能argument變化正在裝修過，你可以將它傳遞給裝飾作爲一個字符串，並使用getattr來得到它從self：

def my_decorator(arg_name): 
    def wrap(f): 
     def wrapped_f(self, *args, **kwargs): 
      argument = getattr(self, arg_name) 
      .... # does something with argument 
     return wrapped_f 
    return wrap 

class MyClass(object): 
    def __init__(self, argument): 
     self.argument = argument 

    @my_decorator("argument") 
    def my_method(self): 
     .... does something