如何使用 class 方法的装饰器

Question

当我设计一个 class 时，我写了一个 class 成员 function 作为装饰器

    def switchWindow(self, win: str):
        def actual_decorator(func):
            self.browser.switch_to.window(self.windowsHandles[win])

            def inner():
                func()
            return inner
        return actual_decorator

装饰器用于在其他成员 function 运行之前更改环境，但是，当我在成员 function 上使用装饰器时，如下所示：

    @switchWindow(win="crop")
    def test_cropFunction(self):
        pass

IDE 引发如下错误：

    @switchWindow(win="crop") TypeError: switchWindow() missing 1 required positional argument: 'self'

我对这个错误感到困惑，但后来我通过在装饰器之前添加一个“自我”来重写装饰器，但这也是错误的：

    @self.switchWindow(win="crop") NameError: name 'self' is not defined

谁能帮我吗？

Answer 1

我将装饰器 function 从 class 中移出，并进行如下更改：

def switchWindow(win: str):
    def actual_decorator(func):
        print(win)

        def inner(self):
            self.browser.switch_to.window(self.windowsHandles[win])
            func(self)
        return inner
    return actual_decorator

现在它可以工作了！