Python类装饰器报错：Browser.check_if_logged()缺少必需位置参数'function'问题排查

我来帮你排查这个装饰器的问题，你的错误其实是几个常见的装饰器使用误区叠加导致的，咱们一步步拆解：

错误原因分析 #

1. 参数不匹配的核心问题：你把check_if_logged定义成了类的实例方法（需要self作为第一个参数），但在类定义阶段用@check_if_logged装饰to_do_login_erp时，Python会直接把被装饰的函数作为第一个参数传给check_if_logged——此时类还没实例化，根本没有self存在！这就导致函数期望接收2个参数（self和function）但只收到1个，直接触发TypeError。
2. 装饰器返回值错误：原代码最后返回wrapper()，这会直接执行wrapper函数并返回结果，而不是返回wrapper作为装饰后的函数，逻辑完全混乱。
3. 参数传递缺失：被装饰的to_do_login_erp是类方法，需要接收self参数，但原wrapper没有定义任何参数，无法把self传递给被装饰函数。
4. 异常处理不精准：用Exception捕获所有异常，无法区分「元素未找到」的正常状态检查和真正的错误。

修复方案（推荐版本） #

我把装饰器改成适配类方法的独立函数，同时修正所有问题，代码如下：

from functools import wraps
from selenium.common.exceptions import NoSuchElementException

# 独立的装饰器函数，适配类方法
def check_if_logged(function):
    '''
    装饰器函数，用于验证用户是否已登录。
    如果用户未登录，则在ERP系统执行登录操作；
    如果用户已登录，则直接跳过。
    '''
    @wraps(function)  # 保留原函数的元信息（名称、文档字符串等）
    def wrapper(self, *args, **kwargs):
        try:
            # 尝试找到登录按钮，判断是否需要登录
            try:
                self.browser.find_element(By.XPATH, '//*[text()="Login administrativo"]')
                print("用户未登录，执行登录流程...")
                # 传递所有参数给被装饰的登录函数
                return function(self, *args, **kwargs)
            except NoSuchElementException:
                # 未找到登录按钮，检查是否处于已登录的企业选择页面
                try:
                    self.browser.find_element(By.XPATH, '//*[text()="Selecione uma empresa"]')
                    print("用户已登录，跳过登录流程。")
                    return True
                except NoSuchElementException:
                    print("无法验证登录状态：未找到预期元素。")
                    return False
        except Exception as e:
            print('验证登录状态时出错:', e)
            return False
    # 返回wrapper函数本身，而不是执行它
    return wrapper

# 类内部使用装饰器的示例
class Browser:
    def __init__(self, browser):
        self.browser = browser

    @check_if_logged
    def to_do_login_erp(self):
        try:
            input_login = self.browser.find_element(By.XPATH, ERP_INPUT_LOGIN_USERNAME)
            input_login.send_keys(USER_NAME_LOGIN)
            input_password = self.browser.find_element(By.XPATH, ERP_INPUT_LOGIN_PASSWORD)
            input_password.send_keys(PASSWORD_LOGIN)
            btn_login = self.browser.find_element(By.XPATH, ERP_BTN_LOGIN)
            btn_login.click()
            return True
        except Exception as e:
            print('执行登录时出错: ', e)
            return False

关键修复点说明 #

• 参数匹配修正：通过wrapper(self, *args, **kwargs)接收类方法的self参数，既能访问实例的browser属性，又能传递所有参数给被装饰函数。
• 返回值修正：返回wrapper函数本身，让装饰器正确替换原函数，而不是提前执行。
• 精准异常处理：用NoSuchElementException捕获元素未找到的情况，区分正常状态检查和其他错误。
• 保留元信息：functools.wraps确保被装饰函数的名称、文档字符串等信息不丢失，方便调试和维护。

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