contrib/rh-bkr: decompress mainpage if it's gzipped

Will be useful for compressed rpmmd.

1 files changed, 6 insertions, 0 deletions

diff --git a/contrib/rh-bkr/bkr.py b/contrib/rh-bkr/bkr.py
index 5c6bb18288..3125561dc2 100755
--- a/contrib/rh-bkr/bkr.py
+++ b/contrib/rh-bkr/bkr.py
@@ -383,6 +383,12 @@ class UploadFile_ParseWebsite(UploadFile):
         page = p.read()
         p.close()
 
+        if re.match('.*\.gz$', self._mainpage):
+            import gzip, StringIO
+            p = StringIO.StringIO(page)
+            page = gzip.GzipFile(fileobj=p).read()
+            p.close()
+
         urls = list(self.parse_urls(page))
         if not urls:
             self.raise_no_urls()