diff options
| -rw-r--r-- | scripts/mongo_to_elasticsearch.py | 7 | ||||
| -rw-r--r-- | scripts/shared_utils.py | 4 |
2 files changed, 9 insertions, 2 deletions
diff --git a/scripts/mongo_to_elasticsearch.py b/scripts/mongo_to_elasticsearch.py index a569ac6..2ffbc17 100644 --- a/scripts/mongo_to_elasticsearch.py +++ b/scripts/mongo_to_elasticsearch.py @@ -395,7 +395,12 @@ def publish_mongo_data(output_destination): for mongo_json_line in fobj: test_result = json.loads(mongo_json_line) if modify_mongo_entry(test_result): - shared_utils.publish_json(test_result, es_creds, output_destination) + status, data = shared_utils.publish_json(test_result, es_creds, output_destination) + if status > 300: + project = test_result['project_name'] + case_name = test_result['case_name'] + logger.info('project {} case {} publish failed, due to [{}]' + .format(project, case_name, json.loads(data)['error']['reason'])) finally: if os.path.exists(tmp_filename): os.remove(tmp_filename) diff --git a/scripts/shared_utils.py b/scripts/shared_utils.py index 91ce38e..8bbbdbe 100644 --- a/scripts/shared_utils.py +++ b/scripts/shared_utils.py @@ -12,9 +12,11 @@ def publish_json(json_ojb, creds, output_destination): json_dump = json.dumps(json_ojb) if output_destination == 'stdout': print json_dump + return 200, None else: headers = urllib3.make_headers(basic_auth=creds) - http.request('POST', output_destination, headers=headers, body=json_dump) + result = http.request('POST', output_destination, headers=headers, body=json_dump) + return result.status, result.data def _get_nr_of_hits(elastic_json): |