如何在HttpUser调用的SequentialTaskSet任务间共享Header并修复类型错误？

核心错误分析 #

1. 类定义顺序颠倒：Python中必须先定义类才能引用，你在PrepTasks里先写了tasks = [QuickStartSequence]，但QuickStartSequence是后面才定义的内部类；同理QuickStartSequence的tasks引用Task1/2/3时，这三个类也还没定义，直接导致"not defined"错误。
2. TaskSet错误继承ApiUser：Locust的TaskSet是依附于User类的任务集合，不能直接继承ApiUser（它是HttpUser的子类）。TaskSet实例可以通过self.user访问所属的User对象，完全不需要继承。

修复后的完整代码 #

ApiUser代码（保持不变）

from locust import HttpUser
class ApiUser(HttpUser):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.users = None
        self.access_token = None
        self.refresh_token = None
        self.headers = None
        self.post_headers = None
    
    def post_helper(self, url: str, json: dict):
        """带错误处理的POST请求封装"""
    
    def get_helper(self, path: str):
        """带错误处理的GET请求封装"""

PrepTasks修复后代码

from api_user import ApiUser
from locust import SequentialTaskSet, task
from typing import Dict, Any

class PrepTasks(ApiUser):
    # 先定义所有内部Task类，再引用
    class Task1(SequentialTaskSet):
        @task
        def get_call1(self):
            # 通过self.user访问ApiUser的方法
            return self.user.get_helper("some-path/some-sub")

        @task
        def stop(self):
            self.interrupt()

    class Task2(SequentialTaskSet):
        @task
        def get_call2(self):
            return self.user.get_helper("some-path/some-sub")

        @task
        def stop(self):
            self.interrupt()

    class Task3(SequentialTaskSet):
        @task
        def get_call3(self):
            return self.user.get_helper("some-path/some-sub")

        @task
        def stop(self):
            self.interrupt()

    class QuickStartSequence(SequentialTaskSet):
        # 此时Task1/2/3已经定义，可以直接引用
        tasks = [Task1, Task2, Task3]

    # 最后引用QuickStartSequence
    tasks = [QuickStartSequence]

    def __init__(self, post_headers: Dict[str, Any], *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.post_headers = post_headers

关键修改说明 #

• 调整内部类顺序：把Task1/2/3、QuickStartSequence的定义放在PrepTasks的tasks引用之前，确保引用时类已经存在。
• 移除TaskSet对ApiUser的继承：所有Task类只继承SequentialTaskSet，通过self.user调用ApiUser的get_helper方法，也可以通过self.user.post_headers获取传递的Header参数。
• Header传递逻辑：PrepTasks的__init__接收post_headers并赋值给自身属性，后续所有TaskSet中的任务都能通过self.user.post_headers拿到这个Header，用于API请求的身份验证等场景。

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