Используйте декоратор класса для реализации поздней инициализации

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 более надежным, сделав его создание дескрипторов, так что он может вести себя корректно для функций и статических методы, а также методы класса и экземпляра.

Answer 2

мне нравится подход sr2222 в использовании свойства для получения соединения, однако здесь подход с декораторами, которые могут быть полезными или, по крайней мере, информативными (использование functools.wraps() не является обязательным):

import functools 

def require_connection(f): 
    @functools.wraps(f) 
    def wrapped(self, *args, **kwargs): 
     self.connect() 
     return f(self, *args, **kwargs) 
    return wrapped 

class MyClass(object): 
    def __init__(self): 
     self.conn = None 

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

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

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

Answer 3

Подобно sr2222, но называя это тем, что это такое: a cached_property.

Код более компактный, использует многоразовые строительные блоки и, на мой взгляд, более читаемый.

class MyClass(object): 

    @cached_property 
    def conn(self): 
     return ConnectToDatabase() 

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

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

Определение cached_propertyhere найдено.