Python类内部装饰器调用需传实例？求优化实现方案

问题描述 #

我编写了一段带类内装饰器的Python代码，用于模拟过期超时场景下的自动刷新逻辑。代码能实现预期效果，但调用实例方法时必须手动传入自身实例（比如test.get_something(test)），想知道能否改成常规的test.get_something()调用方式？这种手动传参的做法是不是最佳实践？这个场景下的最优处理方式是什么？

原代码如下：

from functools import wraps  # 补充原代码遗漏的导入

class Custom():
    def __init__(self, var):
        self.var = var

    def renew(func):
        @wraps(func)
        def wrapper(self, *args, **kwargs):
            try:
                return func(*args, **kwargs)
            except:
                print('refreshing')
                self.refresh()
                return func(*args, **kwargs)
        return wrapper

    def refresh(self):
        self.var = 'refreshed'

    @renew
    def get_something(self):
        print(self.var)
        raise Exception

test = Custom('a')
test.get_something(test)  # 必须手动传实例

问题原因 #

问题出在装饰器renew的参数处理逻辑上：

• 用@renew修饰实例方法get_something时，renew接收的func是未绑定的原始函数（还未与实例关联）。
• 原代码中wrapper调用func(*args, **kwargs)时，没有把wrapper接收到的self传给func，导致func缺少必要的实例参数。
• 所以必须手动传入实例，让args包含这个实例才能让func正常执行，但这违背了Python实例方法的自动绑定规则。

解决方案：修正装饰器的参数传递 #

只需要调整装饰器内func的调用方式，把wrapper接收到的self传给func，就能实现无参调用：

from functools import wraps

class Custom():
    def __init__(self, var):
        self.var = var

    def renew(func):
        @wraps(func)
        def wrapper(self, *args, **kwargs):
            try:
                # 将wrapper的self传给未绑定的func，完成实例绑定
                return func(self, *args, **kwargs)
            except Exception:  # 替换裸except，捕获具体异常类型更安全
                print('refreshing')
                self.refresh()
                return func(self, *args, **kwargs)
        return wrapper

    def refresh(self):
        self.var = 'refreshed'

    @renew
    def get_something(self):
        print(self.var)
        raise Exception

test = Custom('a')
test.get_something()  # 现在无需传参，正常调用

关于传参方式与最优实践 #

• 手动传实例绝非最佳实践：这种方式完全不符合Python语法规范，会降低代码可读性，还容易引发参数传递错误。
• 最优处理方式：
  1. 优先采用上述修正参数传递的方案，这是最简洁直接的修复方式。
  2. 避免使用裸except，应该捕获具体的异常类型（比如你模拟的超时/过期异常），防止隐藏未知bug。
  3. 如果后续需要更复杂的装饰逻辑，可考虑将renew定义为类方法或静态方法，但当前场景下无需额外改动。

内容的提问来源于stack exchange，提问作者bbd108