add --all option to mgagnome to checkout all Mageia packages

1 files changed, 22 insertions, 5 deletions

diff --git a/mgagnome b/mgagnome
index 690a27e..560891d 100755
--- a/mgagnome
+++ b/mgagnome
@@ -859,6 +859,7 @@ class Downstream(object):
 
         FILES = {}
         TARBALLS = {}
+        PACKAGES = set()
 
         for line in  contents:
             try:
@@ -867,6 +868,8 @@ class Downstream(object):
                 print >>sys.stderr, line
                 continue
 
+            PACKAGES.add(srpm)
+
             if '.tar' in filename:
                 r = self.re_file.match(filename)
                 if r:
@@ -887,9 +890,14 @@ class Downstream(object):
                 FILES[srpm] = set()
             FILES[srpm].add(filename)
 
+        self._packages = PACKAGES
         self.tarballs = TARBALLS
         self.files = FILES
 
+    @property
+    def packages(self):
+        return sorted(self._packages)
+
     _provides_cache = {}
     @classmethod
     def alternative_provides(cls, search_for):
@@ -975,10 +983,11 @@ def write_file(path, data):
         fdst.flush()
         os.rename(fdst.name, path)
 
-def cmd_co_multi(l):
-    package, module, package_version, spec_version, downstream_files = l
+def cmd_co_multi(args):
+    package, what_to_print = args
+
+    print what_to_print
 
-    print "%s => %s" % (module, package)
     try:
         Downstream.co(package)
     except subprocess.CalledProcessError:
@@ -986,7 +995,13 @@ def cmd_co_multi(l):
 
 def cmd_co(options, parser):
     p = multiprocessing.Pool(5)
-    p.map(cmd_co_multi, sorted(join_streams()))
+
+    if options.all:
+        packages = ((package, package) for package in Downstream().packages)
+    else:
+        packages = ((l[0], "%s => %s" % (l[0], l[1])) for l in sorted(join_streams()))
+
+    p.map(cmd_co_multi, packages)
 
 def join_streams(show_version=False, only_diff_version=False):
     root = os.path.expanduser(Downstream.PKGROOT)
@@ -1431,8 +1446,10 @@ def main():
     )
 
     subparser = subparsers.add_parser('co', help='checkout all GNOME packages')
+    subparser.add_argument("-a", "--all", action="store_true", dest="all",
+                                 help="Wait before trying to retrieve the new version")
     subparser.set_defaults(
-        func=cmd_co
+        func=cmd_co, all=False
     )
 
     subparser = subparsers.add_parser('dep3', help='add dep3 headers')