decompress stream if it is gzipped

eclecticiq · Oct 31, 2016 · 8bb39f9 · aukjan · Oct 31, 2016 · traut
1 parent a68ed4d
commit 8bb39f9
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/cabby/dispatcher.py b/cabby/dispatcher.py
@@ -4,6 +4,7 @@
 import sys
 
 import furl
+import gzip
 import requests
 from lxml import etree
 from six.moves import urllib
@@ -66,8 +67,8 @@ def send_taxii_request(session, url, request, taxii_binding=None,
 
         stream, headers = response.raw, response.headers
 
-    log.info("Response received for {} from {}"
-             .format(request.message_type, url))
+    if 'gzip' in headers.get('content-encoding', ''):
+        stream = gzip.GzipFile(fileobj=stream)
 
     gen = _parse_response(stream, headers, version=request.version)
     obj = next(gen)