diff --git a/setup.py b/setup.py --- a/setup.py +++ b/setup.py @@ -52,7 +52,7 @@ include_package_data=True, entry_points=''' [console_scripts] - swh-scheduler=swh.scheduler.cli:cli + swh-scheduler=swh.scheduler.cli:main ''', classifiers=[ "Programming Language :: Python :: 3", diff --git a/swh/scheduler/cli.py b/swh/scheduler/cli.py --- a/swh/scheduler/cli.py +++ b/swh/scheduler/cli.py @@ -661,5 +661,9 @@ GHTorrentConsumer(backend, **ght_config).run() +def main(): + return cli(auto_envvar_prefix='SWH_SCHEDULER') + + if __name__ == '__main__': - cli() + main()