Conditionally disable local reindexing

Author: dsotirho-ucsc

scripts/reindex.py

@@ -47,15 +47,22 @@
 parser.add_argument('--local',
                     default=False,
                     action='store_true',
-                    help='Do not offload the listing of subgraphs to the indexer Lambda function. When this option is '
-                         'used, this script queries the repository without partitioning, and the indexer notification '
-                         'endpoint is invoked for each subgraph individually and concurrently using worker threads. '
-                         'This is magnitudes slower than remote (i.e. partitioned) indexing. If this option is not '
-                         'used (the default), the set of subgraphs matching the query is partitioned using the '
-                         'partition prefix length configured for each of the catalog sources being reindexed. Each '
-                         'query partition is processed independently and remotely by the indexer lambda. The index '
-                         'Lambda function queries the repository for each partition and queues a notification for each '
-                         'matching subgraph in the partition.')
+                    help=(
+                             '' if config.enable_bundle_notifications else '**DISABLED** '
+                         ) + (
+                             'Do not offload the listing of subgraphs to the indexer Lambda function. When '
+                             'this option is used, this script queries the repository without partitioning, '
+                             'and the indexer notification endpoint is invoked for each subgraph '
+                             'individually and concurrently using worker threads. This is magnitudes slower '
+                             'than remote (i.e. partitioned) indexing. If this option is not used (the '
+                             'default), the set of subgraphs matching the query is partitioned using the '
+                             'partition prefix length configured for each of the catalog sources being '
+                             'reindexed. Each query partition is processed independently and remotely by '
+                             'the indexer lambda. The index Lambda function queries the repository for each '
+                             'partition and queues a notification for each matching subgraph in the '
+                             'partition.'
+                         )
+                    )
 parser.add_argument('--catalogs',
                     nargs='+',
                     metavar='NAME',
@@ -134,6 +141,10 @@ def main(argv: list[str]):
     every_source = '*' in source_globs
     sources_by_catalog = azul.sources_by_catalog(args.catalogs)
 
+    if not config.enable_bundle_notifications and args.local:
+        parser.error('Local reindexing is not available while bundle '
+                     'notifications are disabled.')
+
     if not every_source:
         if args.local:
             parser.error('Cannot specify sources when performing a local reindex')

src/azul/azulclient.py

@@ -128,6 +128,8 @@ def mirror_source_message(self,
         )
 
     def local_reindex(self, catalog: CatalogName, prefix: str) -> int:
+        if not config.enable_bundle_notifications:
+            raise AzulClientError('Bundle notifications are disabled')
         service = self.index_repository_service
         notifications: JSONs = [
             # Notifications sent organically by DSS had a different structure,
@@ -148,6 +150,8 @@ def index(self,
               notifications: Iterable[JSON],
               delete: bool = False
               ):
+        if not config.enable_bundle_notifications:
+            raise AzulClientError('Bundle notifications are disabled')
         errors = defaultdict[int, int](int)
         missing = []
         indexed = 0