python使用装饰器为方法添加属性

Question

我正在尝试使用装饰器向我的方法添加一个属性，以确定它是否已被调用。我不知道为什么会出现这个错误：TypeError: bet() missing 1 required positional argument: 'betsize'。

文件game.py

class Game(object):
    def __init__(self):
        ..

    @AddCalled
    def bet(self, betsize):
        print(betsize)

文件 addcalled.py ：

class AddCalled(object):
    def __init__(self, f):
        self.f = f
        self.called = False

    def __call__(self, *args, **kwargs):
        self.f(*args, **kwargs)
        self.called = True

我从 Qt 信号（按钮单击）中获得下注大小。目标槽函数是 game.py 中的 bet()

文件 mainui.py

class MainWindow(QtGui.QMainWindow, Ui_MainWindow):
    ..

    self.connect(self.ui.btnBet, SIGNAL("clicked()"), self.bet)

    def bet(self):
        #assuming 'game' is an instance of Game()
        self.game.bet(self.ui.betLine.text())

现在每次点击下注按钮时都会出现此错误

Traceback (most recent call last):
  File "/home/njl/projet/mainui.py", line 27, in bet
    self.game.bet(self.ui.betLine.text())
  File "/home/njl/projet/addcalled.py", line 15, in __call__
    self.f(*args, **kwargs)
TypeError: bet() missing 1 required positional argument: 'betsize'

我将不胜感激。简直卡在里面了。

Answer 1

您的装饰器没有正确地将 self 传递给包装函数。

通常，self 是由函数对象的描述符机制添加的，该机制将它们即时转换为带有 self 滚动的方法对象。但是您将方法替换为 AddDecorator 对象，该对象没有实现描述符协议，所以当你的 AddDecorator 包装器被调用时，它不知道它被调用的是什么实例（即 self 应该是什么）。

你可以在你的 AddDecorator 类上实现描述符协议，方法是给它一个__get__ 方法。或者你可以使用一个函数编写一个普通的装饰器：

def addCalled(func):
    def newFunc(*args, **kwargs):
        newFunc.called = True
        func(*args, **kwargs)
    newFunc.called = False
    return newFunc

Answer 2

当你写作时：

@AddCalled
def bet(self, betsize):
    print(betsize)

这相当于：

def bet(self, betsize):
    print(betsize)

bet = AddCalled(bet)

因此，传递给AddCalled 的参数是bet 函数。