Merge pull request #65 from ipinfo/umar/handle-timeout-error

handled timeout error

Author: UmanShahzad

ipinfo/handler.py

@@ -204,9 +204,12 @@ def getBatchDetails(
             chunk = lookup_addresses[i : i + batch_size]
 
             # lookup
-            response = requests.post(
-                url, json=chunk, headers=headers, **req_opts
-            )
+            try:
+                response = requests.post(
+                    url, json=chunk, headers=headers, **req_opts
+                )
+            except Exception as e:
+                return handler_utils.return_or_fail(raise_on_fail, e, result)
 
             # fail on bad status codes
             try:

ipinfo/handler_async.py

@@ -264,12 +264,15 @@ async def _do_batch_req(
         """
         Coroutine which will do the actual POST request for getBatchDetails.
         """
-        resp = await self.httpsess.post(
-            url,
-            data=json.dumps(chunk),
-            headers=headers,
-            timeout=timeout_per_batch,
-        )
+        try:
+            resp = await self.httpsess.post(
+                url,
+                data=json.dumps(chunk),
+                headers=headers,
+                timeout=timeout_per_batch,
+            )
+        except Exception as e:
+            return handler_utils.return_or_fail(raise_on_fail, e, None)
 
         # gather data
         try: