cli: Require PID type for reindexing

* Adds the required `pid-type` parameter to the reindex that specifies
  the type of PID.

examples/app-fixtures.sh

@@ -13,5 +13,5 @@ export FLASK_APP=app.py
 
 # Load fixtures
 flask fixtures records
-flask index reindex --yes-i-know
+flask index reindex --yes-i-know -t recid
 flask index run

invenio_indexer/cli.py

@@ -67,13 +67,19 @@ def run(delayed, concurrency):
 @click.option('--yes-i-know', is_flag=True, callback=abort_if_false,
               expose_value=False,
               prompt='Do you really want to reindex all records?')
+@click.option('-t', '--pid-type', multiple=True, required=True)
 @with_appcontext
-def reindex():
-    """Reindex all records."""
+def reindex(pid_type, ids=None):
+    """Reindex all records.
+
+    :param pid_type: Pid type.
+    """
     click.secho('Sending records to indexing queue ...', fg='green')
 
     query = (x[0] for x in PersistentIdentifier.query.filter_by(
-        pid_type='recid', object_type='rec', status=PIDStatus.REGISTERED
+        object_type='rec', status=PIDStatus.REGISTERED
+    ).filter(
+        PersistentIdentifier.pid_type.in_(pid_type)
     ).values(
         PersistentIdentifier.object_uuid
     ))

tests/test_cli.py

@@ -87,7 +87,9 @@ def test_reindex(app, script_info):
                             obj=script_info)
         assert 0 == res.exit_code
 
-        res = runner.invoke(cli.reindex, ['--yes-i-know'], obj=script_info)
+        res = runner.invoke(cli.reindex,
+                            ['--yes-i-know', '-t', 'recid'],
+                            obj=script_info)
         assert 0 == res.exit_code
         res = runner.invoke(cli.run, [], obj=script_info)
         assert 0 == res.exit_code
@@ -103,7 +105,9 @@ def test_reindex(app, script_info):
         db.session.commit()
         # Destroy the index and reindex
         list(current_search.delete(ignore=[404]))
-        res = runner.invoke(cli.reindex, ['--yes-i-know'], obj=script_info)
+        res = runner.invoke(cli.reindex,
+                            ['--yes-i-know', '-t', 'recid'],
+                            obj=script_info)
         assert 0 == res.exit_code
         res = runner.invoke(cli.run, [], obj=script_info)
         assert 0 == res.exit_code