From 00a310a5454272bc3f3dd5dd3552b42b0e02dfb2 Mon Sep 17 00:00:00 2001 From: Harris Tzovanakis Date: Wed, 21 Feb 2018 21:04:34 +0100 Subject: [PATCH] cli: addition of queue option in ``run`` Signed-off-by: Harris Tzovanakis --- invenio_indexer/cli.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/invenio_indexer/cli.py b/invenio_indexer/cli.py index 938db63..c09a46a 100644 --- a/invenio_indexer/cli.py +++ b/invenio_indexer/cli.py @@ -49,16 +49,25 @@ def abort_if_false(ctx, param, value): @click.option( '--concurrency', '-c', default=1, type=int, help='Number of concurrent indexing tasks to start.') +@click.option('--queue', '-q', type=str, + help='Name of the celery queue used to put the tasks into.') @click.option('--version-type') @with_appcontext -def run(delayed, concurrency, version_type=None): +def run(delayed, concurrency, version_type=None, queue=None): """Run bulk record indexing.""" if delayed: click.secho( 'Starting {0} tasks for indexing records...'.format(concurrency), fg='green') + data = { + 'kwargs': { + 'version_type': version_type + } + } + if queue is not None: + data.update({'queue': queue}) for c in range(0, concurrency): - process_bulk_queue.delay(version_type=version_type) + process_bulk_queue.apply_async(**data) else: click.secho('Indexing records...', fg='green') RecordIndexer(version_type=version_type).process_bulk_queue()