sectracker.repo.Config: new class, for loading config.json

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

3 files changed, 120 insertions, 2 deletions

diff --git a/data/config.json b/data/config.json
new file mode 100644
index 0000000000..bba9f3c7ce
--- /dev/null
+++ b/data/config.json
@@ -0,0 +1,53 @@
+{
+  "repositories" : {
+    "etch": "http://mirror.1und1.de/debian/dists/etch",
+    "etch-security" :
+      "http://security.debian.org/debian-security/dists/etch/updates",
+    "etch-proposed-updates" :
+      "http://mirror.1und1.de/debian/dists/etch-proposed-updates",
+
+    "lenny":  "http://mirror.1und1.de/debian/dists/lenny",
+    "lenny-security" :
+      "http://security.debian.org/debian-security/dists/lenny/updates",
+    "lenny-proposed-updates" :
+      "http://mirror.1und1.de/debian/dists/lenny-proposed-updates",
+
+    "squeeze":  "http://mirror.1und1.de/debian/dists/squeeze",
+    "squeeze-security" :
+      "http://security.debian.org/debian-security/dists/squeeze/updates",
+    "squeeze-proposed-updates" :
+      "http://mirror.1und1.de/debian/dists/squeeze-proposed-updates",
+
+    "sid" : "http://mirror.1und1.de/debian/dists/sid"
+  },
+
+  "distributions" : {
+    "etch" : {
+      "members" : {
+	"unsupported" : ["etch", "etch-security"],
+	"optional" : ["etch-proposed-updates"]
+      },
+      "release" : "oldstable"
+    },
+    "lenny" : {
+      "members" : {
+	"supported" : ["lenny", "lenny-security"],
+	"optional" : ["lenny-proposed-updates"]
+      },
+      "release" : "stable"
+    },
+    "squeeze" : {
+      "members" : {
+	"supported" : ["squeeze", "squeeze-security"],
+	"optional" : ["squeeze-proposed-updates"]
+      },
+      "release" : "testing"
+    },
+    "sid" : {
+      "members" : {
+	"supported" : ["sid"]
+      },
+      "release" : "unstable"
+    }
+  }
+}
diff --git a/lib/python/sectracker/repo.py b/lib/python/sectracker/repo.py
index 2b9c9d5fc0..7dbce2effb 100644
--- a/lib/python/sectracker/repo.py
+++ b/lib/python/sectracker/repo.py
@@ -18,6 +18,7 @@
 from __future__ import with_statement
 
 import bz2 as _bz2
+import cjson as _cjson
 import hashlib as _hashlib
 import gzip as _gzip
 import os as _os
@@ -199,7 +200,10 @@ class RepoCollection(object):
         with file(self._relname(name)) as f:
             return _parserelease(name, f)
 
-    def filemap(self):
+    def filemap(self, load=False):
+        """Returns dictionaries mapping repositories to components to files.
+        If load is true, the files are loaded using the source packages
+        parser."""
         d = {}
         for name in self.repos:
             rel = self.release(name)
@@ -216,7 +220,10 @@ class RepoCollection(object):
                     self.warn("file %s for %s/%s not present" %
                               (listname, name, comp))
                     continue
-                comps[comp] = listname
+                if load:
+                    comps[comp] = _parsers.sourcepackages(listname)
+                else:
+                    comps[comp] = listname
             d[name] = comps
         return d
 
@@ -245,3 +252,51 @@ class RepoCollection(object):
     def warn(self, msg):
         if self.verbose:
             print msg
+
+class Config(object):
+    def __init__(self, config, root):
+        with file(config) as f:
+            self.config = _cjson.decode(f.read())
+        self.repositories = self.config["repositories"]
+        self.distributions = self.config["distributions"]
+        self.releases = {}
+
+        self.collection = RepoCollection(root)
+        for k,v in self.repositories.items():
+            self.collection.add(k, v)
+
+        for d, dobj in self.distributions.items():
+            for m, mobj in dobj.get("members", {}).items():
+                for mem in mobj:
+                    if mem not in self.repositories:
+                        raise ValueError(
+                            "distributions[%r][%r] (%r) not a valid repository"
+                            % (d, m, mem))
+            if dobj.has_key("release"):
+                rel = dobj["release"]
+                if rel in self.releases:
+                    raise ValueError(
+                        "distributions[%r] is duplicate of %r (previous was %r)"
+                        % (d, rel, self.releases[rel]))
+                self.releases[rel] = d
+
+    def update(self):
+        self.collection.update()
+
+    def releasepackageversions(self):
+        """Returns dictionaries mapping release codenames to packages
+        to a set of versions."""
+        fm = self.collection.filemap(load=True)
+        r = {}
+        for d, dobj in self.distributions.items():
+            pkgver = {}
+            for mobj in dobj.get("members", {}).values():
+                for mem in mobj:
+                    for comps in fm[mem].values():
+                        for src in comps.values():
+                            if src.name in pkgver:
+                                pkgver[src.name].add(src.version)
+                            else:
+                                pkgver[src.name] = set((src.version,))
+            r[d] = pkgver
+        return r
diff --git a/lib/python/sectracker_test/test_repo.py b/lib/python/sectracker_test/test_repo.py
index 4ffe523388..740f466d68 100644
--- a/lib/python/sectracker_test/test_repo.py
+++ b/lib/python/sectracker_test/test_repo.py
@@ -15,6 +15,7 @@
 # along with this program; if not, write to the Free Software
 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
+import os.path
 import shutil
 import tempfile
 
@@ -41,3 +42,12 @@ try:
     assert "bash" in o["bash"].binary
 finally:
     shutil.rmtree(tmp)
+
+if not os.path.exists("sectracker_test/tmp"):
+    os.makedirs("sectracker_test/tmp")
+c = Config("../../data/config.json", "sectracker_test/tmp/repo")
+c.update()
+rpv = c.releasepackageversions()
+assert "sid" in rpv
+assert "bash" in rpv["sid"]
+