diff --git a/funkwhale_network/cli.py b/funkwhale_network/cli.py index b3e9de019dc3b7a5c2c249dc43ad8178f813bff3..93ba2956d917124d57b961f2a222cb4543af77ec 100644 --- a/funkwhale_network/cli.py +++ b/funkwhale_network/cli.py @@ -92,7 +92,7 @@ async def launch_domain_poll(pool, session, domain): @cli.command() -@click.argument("domain", type=str, nargs=-1, required=True) +@click.argument("domain", type=str, nargs=-1) @async_command async def poll(domain): """ @@ -101,8 +101,16 @@ async def poll(domain): from . import crawler from . import db from . import settings + from . import worker pool = await db.get_pool(settings.DB_DSN) + if not domain: + click.echo("Polling all domains…") + crawler = worker.Crawler( + redis_settings=arq.RedisSettings(**settings.REDIS_CONFIG) + ) + return await crawler.poll_all() + try: kwargs = crawler.get_session_kwargs() async with aiohttp.ClientSession(**kwargs) as session: