python - 刺槐。从 python 代码中读取 --host 值

Question

蝗虫 --no-web --client=1 --hatch-rate=1 --num-request=2 --host= http://localhost

我想读取 HTTPLocust 类中 cmd 行中提供的 --host 值。我知道我可以使用主机属性进行直接分配，但我不想要它。我想从 HTTPLocust 类中的 cmd 行读取值。我正在构建自定义日志并希望将该值传递给日志。我尝试了 HTTPLocust.host 但没有返回。

我还想从 python 代码中读取 --web-port 。

Answer 1

新答案

有一个比下面我最初的解决方案更简单的解决方案。每个 TaskSet 都有一个 locust 属性，该属性链接回其父 Locust locustinstance ，所以像这样的东西将完全满足您的需要:

class UserBehaviour(TaskSet):
    def __init__(self, parent):
        super().__init__(parent)
        print(self.locust.host)

旧答案

查看code for HttpSession ，看来 base_url 就是你想要的。

这样的东西应该给你当前的主机，要么是默认的，要么是在命令行上指定的:

class HostGetter(locust.TaskSet):
    @locust.Task()
    def get_host(self):
        print(self.client.base_url)