python - 如何向 Python Locust 添加自定义命令行参数？

Question

如何向 Python Locust 0.7.2 添加自定义命令行参数，然后检索其值？

我想指定一个数字：

locust --my-argument 6

然后在我的脚本中，我想检索该值，例如：

my_argument = ??.options.my_argument

我在 locust/main.py 中找到了这个

def main():
    parser, options, arguments = parse_options()

但似乎 parse_options() 函数不会从外部查找任何其他参数。

这可能吗？

谢谢。

Answer 1

您可以使用环境。

c:\locust>set env_test = ekomut
c:\locust>locust ...

import os
....
 def on_start(self):
        print("Variable:"+os.environ.get("env_test "))

Answer 2

您可以编辑文件（或 fork 项目）以启用其他可选参数。毕竟，它是开源的并且具有许可的许可证。