在Python裝飾器中獲取類

Question

在這段代碼中：

def online_only(func, self):
    def f(*args, **kwargs):
        if self.running:
            return func(*args, **kwargs)
        else:
            return False
    return f
class VM(object):
   @property
   def running(self):
       return True
   @property
   @online_only
   def diskinfo(self):
       return True

我希望diskinfo只在VM.running返回True時運行。 如何才能通過online_only閱讀self.running？

Answer 1

self作為第一個參數傳遞給包裝函數，所以只需要在f處理第一個參數：

def online_only(func):
    def f(self, *args, **kwargs):
        if self.running:
            return func(self, *args, **kwargs)
        else:
            return False
    return f

Answer 2

1. 你不能在def online_only(func, self)有兩個參數？ 它會引發TypeError，因此將其更改為def online_only(func)
2. 包裝函數的第一個參數是self，你可以使用它，例如

---

def online_only(func):
    def f(self):
        if self.running:
            return func(self)
        else:
            return False
    return f

class VM(object):
    @property
    def running(self):
        return True

    @property
    @online_only
    def diskinfo(self):
        return True

print VM().diskinfo