From 3235ac5f59bf823d2ae032da527b6898893defeb Mon Sep 17 00:00:00 2001
From: Eliot Berriot <contact@eliotberriot.com>
Date: Wed, 23 Jan 2019 14:43:41 +0100
Subject: [PATCH] Added CLI to poll all domains
---
funkwhale_network/cli.py | 10 +++++++++-
1 file changed, 9 insertions(+), 1 deletion(-)
diff --git a/funkwhale_network/cli.py b/funkwhale_network/cli.py
index b3e9de0..93ba295 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:
--
GitLab