From 1be03569faa7e248da60e8e1dcd20cc4e6c45d71 Mon Sep 17 00:00:00 2001 From: Nikolai Kristiansen Date: Wed, 15 Jun 2016 22:13:27 +0200 Subject: [PATCH] Add timeout to logstash post --- upsc_to_logstash.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/upsc_to_logstash.py b/upsc_to_logstash.py index 1332044..daf1315 100755 --- a/upsc_to_logstash.py +++ b/upsc_to_logstash.py @@ -59,8 +59,14 @@ if __name__ == '__main__': 'status': get_status(sys.argv[1]) } json_output = json.dumps(output, indent=4) - r = requests.post( - settings.LOGSTASH_URL, - data=json_output, headers={'content-type':'application/json'}, - auth=(settings.LOGSTASH_USER, settings.LOGSTASH_PASSWORD) - ) + try: + r = requests.post( + settings.LOGSTASH_URL, + data=json_output, headers={'content-type':'application/json'}, + auth=(settings.LOGSTASH_USER, settings.LOGSTASH_PASSWORD), + timeout=10 + ) + except requests.exceptions.Timeout as e: + print('Could not post data to logstash: {0}'.format(str(e))) + exit(2) +