如何向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

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