handle missing api header

2018-10-27 14:59:15 -05:00 · 2018-10-27 14:59:15 -05:00 · 4e86604692
commit 4e86604692
parent c073020b80
1 changed files with 4 additions and 1 deletions
--- a/onionr/onionrutils.py
+++ b/onionr/onionrutils.py
@ -616,8 +616,11 @@ class OnionrUtils:
            proxies = {'http': 'socks4a://127.0.0.1:' + str(port), 'https': 'socks4a://127.0.0.1:' + str(port)}
            r = requests.get(url, headers=headers, proxies=proxies, allow_redirects=False, timeout=(15, 30))
            # Check server is using same API version as us
+            try:
                if r.headers['api'] != str(API_VERSION):
                    raise onionrexceptions.InvalidAPIVersion
+            except KeyError:
+                raise onionrexceptions.InvalidAPIVersion
            retData = r.text
        except KeyboardInterrupt:
            raise KeyboardInterrupt