sectracker.repo: only download Sources files

Also adjust RepoCollection.filemap() to focus on source packages,
and augment test case for it.


git-svn-id: svn+ssh://svn.debian.org/svn/secure-testing@14641 e39458fd-73e7-0310-bf30-c45bca0a0e42

2 files changed, 41 insertions, 41 deletions

diff --git a/lib/python/sectracker/repo.py b/lib/python/sectracker/repo.py
index cc0a3184ae..7318ed8b3d 100644
--- a/lib/python/sectracker/repo.py
+++ b/lib/python/sectracker/repo.py
@@ -156,25 +156,24 @@ class RepoCollection:
             rel = self.release(name)
             hashes = rel["sha256"]
             for comp in rel["components"]:
-                for arch in rel["architectures"]:
-                    plainpath = self._plainpath(comp, arch)
-                    plainurl = url + plainpath
-                    if not plainpath in hashes:
-                        self.warn("not downloaded because uncompressed version not present in Release file: " + plainurl)
-                        continue
-                    uncompressed_digest = hashes[plainpath]
-                    listname = self._listname(uncompressed_digest)
-                    if os.path.exists(listname):
-                        continue
-                    success = False
-                    for suffix, method in ((".bz2", downloadbz2),
-                                           (".gz", downloadgz)):
-                        if method(plainurl + suffix, listname,
-                                  uncompressed_digest):
-                            success = True
-                            break
-                    if not success:
-                        self.warn("download failed: " + plainurl)
+                plainpath = self._sourcepath(comp)
+                plainurl = url + plainpath
+                if not plainpath in hashes:
+                    self.warn("not downloaded because uncompressed version not present in Release file: " + plainurl)
+                    continue
+                uncompressed_digest = hashes[plainpath]
+                listname = self._listname(uncompressed_digest)
+                if os.path.exists(listname):
+                    continue
+                success = False
+                for suffix, method in ((".bz2", downloadbz2),
+                                       (".gz", downloadgz)):
+                    if method(plainurl + suffix, listname,
+                              uncompressed_digest):
+                        success = True
+                        break
+                if not success:
+                    self.warn("download failed: " + plainurl)
 
     def _updatelrelease(self, name):
         url = self.repos[name]
@@ -205,36 +204,29 @@ class RepoCollection:
         for name in self.repos:
             rel = self.release(name)
             hashes = rel["sha256"]
-            l = []
+            comps = {}
             for comp in rel["components"]:
-                for arch in rel["architectures"]:
-                    plainpath = self._plainpath(comp, arch)
-                    if not plainpath in hashes:
-                        self.warn("failed to find %s/%s/%s" % (name, comp, arch))
-                        continue
-                    digest = hashes[plainpath]
-                    listname = self._listname(digest)
-                    if not os.path.exists(listname):
-                        self.warn("file %s for %s/%s/%s not present" %
-                                  (listname, name, comp, arch))
-                        continue
-                    if arch == "source":
-                        method = _parsers.sourcepackages
-                    else:
-                        method = _parsers.binarypackages
-                    l.append((comp, arch, listname, method))
-            d[name] = l
+                plainpath = self._sourcepath(comp)
+                if not plainpath in hashes:
+                    self.warn("failed to find %s/%s" % (name, comp))
+                    continue
+                digest = hashes[plainpath]
+                listname = self._listname(digest)
+                if not os.path.exists(listname):
+                    self.warn("file %s for %s/%s not present" %
+                              (listname, name, comp))
+                    continue
+                comps[comp] = listname
+            d[name] = comps
         return d
 
     def _relname(self, name):
         return "%s/r_%s" % (self.root, name)
 
-    def _plainpath(self, comp, arch):
+    def _sourcepath(self, comp):
         # Hack to deal with the "updates/" special case.
         comp = comp.split("/")[-1]
-        if arch == "source":
-            return comp + "/source/Sources"
-        return "%s/binary-%s/Packages" % (comp, arch)
+        return comp + "/source/Sources"
 
     def _listname(self, digest):
         return "%s/h_%s" % (self.root, digest)
diff --git a/lib/python/sectracker_test/test_repo.py b/lib/python/sectracker_test/test_repo.py
index 79a4260c90..076d2bbd86 100644
--- a/lib/python/sectracker_test/test_repo.py
+++ b/lib/python/sectracker_test/test_repo.py
@@ -18,6 +18,7 @@
 import shutil
 
 from sectracker.repo import *
+import sectracker.parsers as p
 
 tmp = tempfile.mkdtemp()
 try:
@@ -30,5 +31,12 @@ try:
     r.add("squeeze", mirror + "debian/dists/squeeze")
     r.add("sid", mirror + "debian/dists/sid")
     r.update()
+    fm = r.filemap()
+    assert "sid" in fm
+    assert "main" in fm["sid"]
+    o = p.sourcepackages(fm["sid"]["main"])
+    assert "bash" in o
+    assert o["bash"].name == "bash"
+    assert "bash" in o["bash"].binary
 finally:
     shutil.rmtree(tmp)