run locust 1.0.1 directly from .py file

Anyway I run locust 1.0.1 directly from .py file? test.py

class MainTaskSet(TaskSet):
	@task
	...

class WebsiteUser(FastHttpUser):
	tasks = [MainTaskSet]

if __name__ == '__main__':
	from locust.main import main
	main()

It used to work, but with 1.0.1 I got: ./test.py Could not find any locustfile! Ensure file ends in ‘.py’ and see --help for available options.

One way is to set env. var with current file name in if __name__ == '__main__' block. Another is to run it as library.

Is there is an easier way to do it? It’s somewhat related to run/debug from locustfile from vscode also…

[ecc256]

I have it already. Just missing the old way I used to run it, that’s all. BTW: It was a suggestion only, don’t take it personal, please! 😃

[cyberw]

I have it already. Just missing the old way I used to run it, that’s all. BTW: It was a suggestion only, don’t take it personal, please! 😃

I wasn’t taking it personally, sorry if I made it sound like I did 😃