diff --git a/swh/journal/cli.py b/swh/journal/cli.py --- a/swh/journal/cli.py +++ b/swh/journal/cli.py @@ -13,7 +13,7 @@ from swh.storage import get_storage from swh.objstorage import get_objstorage -from swh.journal.client import JournalClient +from swh.journal.client import JournalClient, ACCEPTED_OBJECT_TYPES from swh.journal.replay import process_replay_objects from swh.journal.replay import process_replay_objects_content from swh.journal.backfill import JournalBackfiller @@ -50,7 +50,8 @@ ctx.obj['config'] = conf -def get_journal_client(ctx, brokers, prefix, group_id): +def get_journal_client(ctx, brokers, prefix, group_id, + object_types=ACCEPTED_OBJECT_TYPES): conf = ctx.obj['config'] if brokers is None: brokers = conf.get('journal', {}).get('brokers') @@ -65,7 +66,9 @@ if group_id is None: group_id = conf.get('journal', {}).get('group_id') - return JournalClient(brokers=brokers, group_id=group_id, prefix=prefix) + return JournalClient( + brokers=brokers, group_id=group_id, prefix=prefix, + object_types=object_types) @cli.command()