Question

我有一个python类Something。我想创建一个方法borrowed，它会检查blue的实例变量Something是否为None。

如何在实例方法上创建@check_none以便检查实例变量？在装饰器函数中使用self不起作用;（

示例：

def check_token(func):
    def inner(*args, **kwargs):
        if self.token == None:
            raise ValueError
        else:
            return func(*args, **kwargs)
    return inner

class Something(object):
   def __init__(self, token=None):
      self.token = token

   @check_token
   def testfunction(self):
      print "Test"

产生global name 'self' is not defined错误。

Answer 1

你的内在功能没有self参数;添加它并传递它：

def check_token(func):
    def inner(self, *args, **kwargs):
        if self.token is None:
            raise ValueError
        else:
            return func(self, *args, **kwargs)
    return inner

inner函数在装饰时替换原始方法，并在相同的self参数中传递。

或者，您可以使用args[0].token，因为self只是第一个位置参数。

请注意，我已将== None测试替换为推荐的is None测试。

演示：

>>> def check_token(func):
...     def inner(self, *args, **kwargs):
...         if self.token is None:
...             raise ValueError
...         else:
...             return func(self, *args, **kwargs)
...     return inner
... 
>>> class Something(object):
...    def __init__(self, token=None):
...       self.token = token
...    @check_token
...    def testfunction(self):
...       print "Test"
... 
>>> Something().testfunction()
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 4, in inner
ValueError
>>> Something('token').testfunction()
Test

装饰器检查实例变量的方法

1 个答案: