使用类装饰器来实现后期初始化

Question

我正在使用一些需要连接到数据库的类.只有在执行实际操作时才需要连接.我想延迟连接阶段,直到确实需要它为止.为此,我想做类似的事情:

class MyClass

    def __init__(self):
        self.conn = None

    def connect(self):
        if self.conn : return
        self.conn = ConnectToDatabase()

    @connect
    def do_something1(self):
        self.conn.do_something1()

    @connect
    def do_something2(self):
        self.conn.do_something2()

但我不知道如何connect为类定义装饰器.

我当然可以这样做:

    def do_something1(self):
        self.connect()
        self.conn.do_something1()

但使用装饰器似乎是一种更易读的解决方案.可能吗？

Answer 1

而不是尝试装饰需要连接的函数,而是使用属性来获取连接本身.

class MyClass(object):

    def __init__(self):
        self._conn = None

    @property
    def conn(self):
        if self._conn is None:
            self._conn = ConnectToDatabase()
        return self._conn

    def do_something1(self):
        self.conn.do_something1()

    def do_something2(self):
        self.conn.do_something2()

至于直装饰的例子,玩FJ的答案:

def prerequisite(prerequisite_function, *pre_args, **pre_kwargs):
    def wrapper(func):
        def wrapped(self, *args, **kwargs):
            prerequisite_function(self, *pre_args, **pre_kwargs)
            return func(self, *args, **kwargs)
        return wrapped
    return wrapper

 class MyClass(object):

     def __init__(self):
         self.conn = None

     def connect(self):
         if self.conn is None:
             self.conn = ConnectToDatabase()

     @prerequisite(connect)
     def do_something(self):
         self.conn.do_something()

您还可以prerequisite通过创建描述符使其更加健壮,以便它可以正常运行函数和静态方法以及类和实例方法.