如何在Python装饰器中访问构造函数的变量？

问题原因 #

你遇到的NameError本质是：类定义是在模块加载阶段执行的，此时还没有创建任何MainWindow的实例，self这个变量只在实例方法的内部作用域中存在，所以在类定义时直接用self.my_thread作为装饰器参数，Python根本找不到这个名字。

解决方案 #

下面提供两种实用的解决思路：

方法1：修改装饰器，通过属性名动态获取实例变量 #

让装饰器接收实例的属性名称字符串，在方法执行时，从实例（也就是方法的第一个参数self）中动态取出对应的变量。

修改后的装饰器代码：

def thread_switch_decorator(attr_name):
    def decorator(function):
        def wrapper(*args, **kwargs):
            # 取出方法的第一个参数（即实例self）
            self_instance = args[0]
            # 通过属性名获取对应的线程对象
            thread = getattr(self_instance, attr_name)
            thread.stop()
            result = function(*args, **kwargs)
            thread.start()
            return result
        return wrapper
    return decorator

类中的使用方式：

class MainWindow(QMainWindow):
    def __init__(self):
        self.my_thread = initiate_connect()
    
    # 传入实例属性的名称字符串
    @thread_switch_decorator(attr_name='my_thread')
    def home_page(self):
        pass

方法2：在__init__中动态装饰方法 #

如果不想修改原有装饰器，可以在类的构造函数中完成装饰——因为__init__是实例创建时执行的，此时self已经存在，可以直接拿到self.my_thread。

需要注意用functools.wraps保留原方法的元信息（比如函数名、文档字符串），避免后续出现问题：

from functools import wraps

# 保留你原来的装饰器不变
def thread_switch_decorator(argument):
    def decorator(function):
        @wraps(function)
        def wrapper(*args, **kwargs):
            argument.stop()
            result = function(*args, **kwargs)
            argument.start()
            return result
        return wrapper
    return decorator

class MainWindow(QMainWindow):
    def __init__(self):
        self.my_thread = initiate_connect()
        # 在实例初始化时，动态装饰home_page方法
        self.home_page = thread_switch_decorator(argument=self.my_thread)(self.home_page)
    
    def home_page(self):
        pass

两种方法的对比 #

• 方法1更优雅，适合多个方法需要使用同一个装饰器的场景，只需要传入属性名即可。
• 方法2不需要修改原有装饰器，适合临时适配的场景，但如果有多个方法需要装饰，代码会稍显繁琐。

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