From eb1a47579f915c516e74c2df060ae0c4e548bc50 Mon Sep 17 00:00:00 2001
From: Jesse Keating <jkeating at redhat dot com>
Date: Thu, 21 Feb 2008 18:04:39 -0500
Subject: [PATCH] Handle gzipped comps files.

---
 Changelog             | 3 +++
 src/pypungi/gather.py | 7 ++++++-
 2 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/Changelog b/Changelog
index c1d26cae..d4465d75 100644
--- a/Changelog
+++ b/Changelog
@@ -1,3 +1,6 @@
+* Thu Feb 21 2008 Jesse Keating <jkeating@redhat.com>
+- Handle gzipped comps files.
+
 * Fri Feb 08 2008 Jesse Keating <jkeating@redhat.com>
 - Fix comps mashup to be more lenient with the open/close of <comps
 
diff --git a/src/pypungi/gather.py b/src/pypungi/gather.py
index 0ed10436..d7f0c81b 100755
--- a/src/pypungi/gather.py
+++ b/src/pypungi/gather.py
@@ -16,6 +16,7 @@ import yum
 import os
 import shutil
 import sys
+import gzip
 import pypungi
 import logging
 import urlgrabber.progress
@@ -389,7 +390,11 @@ class Gather(pypungi.PungiBase):
                 self.logger.warn("No group data found for %s" % repo.id)
                 pass
             else:
-                compslines = open(groupfile, 'r').readlines()
+                # Check to see if we got a gzipped comps file
+                if groupfile.endswith('.gz'):
+                    compslines = gzip.open(groupfile, 'r').readlines()
+                else:
+                    compslines = open(groupfile, 'r').readlines()
                 for line in compslines:
                     if line.startswith('</comps'):
                         end = compslines.index(line)