为日志和结果动态生成目录

Question

我正在尝试配置 locust 以将我的日志文件存储到 logs/ 目录中。例如：

# example.conf
locustfile = example/locust.py
host = http://www.example.com
users = 10
spawn-rate = 10
logfile = logs/example.log

但是，如果 logs/ 目录不存在，locust 将失败。我试过用初始化事件侦听器添加目录：

@events.init.add_listener
def on_init(**kw):
    if 'logs' not in os.listdir():
        os.mkdir('logs')

但是locust还没走到这一步就失败了。有什么办法可以解决这个问题，目前的蝗虫构建方式？

Answer 1

只需将您的代码放在 locust 文件的 top/module 级别，而不是函数内部，它将在其他所有内容之前执行。

if 'logs' not in os.listdir():
    os.mkdir('logs')

Dynamically generating directories for logs and results