用于刷新光标实例的 Python 装饰器

Question

我有一种在数据库中保存数据的方法，还有一个装饰器来管理连接，但我不知道如何使其工作。

保存方法：

class DA_Row(DABase):

    @DABase.connectAndDisconnect
    def save(self):
        """
        Guarda el spin en la base de datos
        """
        self.__cursor.callproc('sp_insert_row', (
                                 "value 1",
                                 "value 2"
                                 )
        )

我这里继承的类带有一个不起作用的函数装饰器。

class DABase():

    def __init__(self):
        self.__cursor = None

    @staticmethod
    def connectAndDisconnect(func):
        def deco(*args):
            returnValue = None
            self.DBconnect()
            try:
                self.__cursor = self.db.cursor()
                returnValue = func(*args)
            finally:
                self.desconectarDB()

            return returnValue
        return deco
....

显示此...

如何从装饰器重新定义 DABase.__cursor ？

如果不可能，如何以不同的方式解决这个问题？

谢谢您的宝贵时间！

Answer 1

如果您显示您收到的错误，这将会有所帮助。但是，我可以猜测...

装饰类的方法很困难。 connectAndDisconnect 应该如何知道 self 应该是什么？ connectAndDisconnect 是基类的一个静态方法，在创建派生类时调用该方法，早于派生类的任何实例获得创建的。

有一个技巧可以让装饰器弄清楚 self 应该是什么，但这是一个复杂的技巧，而且很脆弱，我将在最后解释。诀窍是，使用类作为装饰器，并使该类成为描述符（即定义__get__），以便您有机会确定self 应该是。在您的情况下，它看起来像：

class DABase(object):
  def __init__(self):
    self.__cursor = None

  class connectAndDisconnect(object):
    def __init__(self, method):
      self._method = method # method is the thing being decorated
                            # note that it's an *unbound* method
      self._instance = None # no way to know what the instance is yet

    def __get__(self, instance, owner):
      'This will be called when the decorated method is accessed'
      self._instance = instance
      return self

    def __call__(self, *args):
      'This is where the actual decoration takes place'
      returnValue = None

      # 'self' is the connectAndDisconnect object. 'self._instance' is the decorated object.
      self._instance.DBConnect()
      try:
        self._instance.__cursor = self._instance.db.cursor()
        # Since self._method is unbound, we have to pass the instance explicitly
        returnValue = self._method(self._instance, *args)
      finally:
        self._instance.desconectarDB()
      return returnValue

派生类未更改：

class DA_Row(DABase):
  @DABase.connectAndDisconnect
  def save(self):
    # ...

现在 DA_Row.save 实际上是 connectAndDisconnect 类的实例。如果d是一个DA_Row对象并且有人调用d.save()，首先发生的事情是connectAndDisconnect.__get__< /code> 被调用，因为有人尝试访问 d.save。这会将 _instance 变量设置为等于 d。然后调用 connectAndDisconnect.__call__ 并进行实际的装饰。

大多数时候这都是有效的。但它很脆弱。 仅如果您以“正常”方式（即通过实例）调用save，它才有效。如果您尝试做一些有趣的事情，例如调用 DA_Row.save(d)，它不会工作，因为 connectAndDisconnect.__get__ 不会能够弄清楚实例应该是什么。

Answer 2

self 只是一个像其他东西一样的名称，它并不像 Java 的 this 那样神奇地出现。您需要将其添加到您的装饰器中。试试这个：

    @staticmethod
    def connectAndDisconnect(func):
        # deco will be a method, so it needs self (ie a DA_Row instance)
        def deco(self, *args):
            returnValue = None
            self.DBconnect()
            try:
                self.__cursor = self.db.cursor()
                # func was supposed to be a method to, so it needs self
                returnValue = func(self, *args)
            finally:
                self.desconectarDB()

            return returnValue
        return deco