diff --git a/workers/sample_generator.py b/workers/sample_generator.py index d86d5f3dc64d57cdb6d9cfb64a55d81c826c0e76..82dd4e90691d64e5a761e3050f8e90bde084ff4c 100644 --- a/workers/sample_generator.py +++ b/workers/sample_generator.py @@ -30,7 +30,10 @@ def callback(channel, method, properties, body): #data = res.json() # get sample records from the ingest index - source_es = Elasticsearch([cfg['reindexer']['source_url']], timeout=60) + if 'source_url' in cfg['reindexer'].keys(): + source_es = Elasticsearch([cfg['reindexer']['source_url']], timeout=60) + else: + source_es = Elasticsearch([cfg['reindexer']['destination_url']], timeout=60) the_query = dict() the_query['size'] = sample_size