Question

我有一个名为Server的类，可以启动和停止。除非启动服务器，否则不应调用某些方法，在这种情况下应引发NotConnectedException。有没有办法在类中的每个方法之前调用方法，并确定类变量_started是否设置为True？

我尝试使用装饰器，但装饰器功能无法访问类变量。我试图做这样的事情：

class Server(object):
    _started = False
    def started(self):
        if(self._started == False):
            raise NotConnectedException

    @started
    def doServerAction(self):
       ...

Answer 1

记住装饰：

@decorate
def foo(...):
    ...

完全等同于：

def foo(...):
    ...
foo = decorate(foo)

在函数上调用装饰器，因此调用第一个参数 self 是没有意义的。此外，在定义函数时调用函数，并且使用它返回的任何函数代替函数。因此，即使您的started装饰器没有通过尝试访问函数的AttributeError属性而抛出_started，它也会返回None，使您的所有方法都设置到None，因此甚至无法调用。

你想要的是这样的：

import functools

def started(func):
    @functools.wraps(func)
    def wrapper(self, *args, **kwargs):
        if not self._started:
            raise ...
        else:
            return func(self, *args, **kwargs)
    return wrapper

几乎所有装饰者都是这种形式;他们接受一个函数，创建一个包装器，在接收函数周围做一些事情，然后返回包装器。如果您最终在交互式解释器会话中使用此代码，那么在此使用functools.wraps会很方便。它会自动使用原始函数的名称和文档字符串更新wrapper函数，这使得修饰函数“看起来像”原始函数更多。

这是否在课堂内定义是无关紧要的。

方法调用前检查条件

1 个答案: