new ReleaseList class to order project releases

2 files changed, 15 insertions, 7 deletions

diff --git a/patchtracker/SourceArchive.py b/patchtracker/SourceArchive.py
index 832b315..099b506 100755
--- a/patchtracker/SourceArchive.py
+++ b/patchtracker/SourceArchive.py
@@ -144,6 +144,18 @@ class SourcePackageIndex:
   def indices(self):
     return sorted(self.pkgs.iterkeys())
 
+class ReleaseList:
+
+  def __init__(self, project):
+    self.project = project
+    # XXX this is hard coded for now...
+    self.relorder = {}
+    self.relorder["debian"] = ["etch", "lenny", "sid", "experimental"]
+
+  def __iter__(self):
+    for r in self.relorder[self.project]:
+      yield r
+
 # basic module tests
 if __name__ == "__main__":
   import Conf
diff --git a/patchtracker/Templates.py b/patchtracker/Templates.py
index b940670..89f7471 100755
--- a/patchtracker/Templates.py
+++ b/patchtracker/Templates.py
@@ -1,5 +1,6 @@
 from patchtracker import Conf
 from patchtracker.DiffGzHandler import DiffGzHandler
+from patchtracker.SourceArchive import ReleaseList
 
 from Cheetah.Template import Template
 from Cheetah.Compiler import Compiler
@@ -48,10 +49,7 @@ class LetterTocTemplate(OurTemplate):
   def __init__(self, letter, collection):
     self.pkgs = collection
     self.idx = letter
-    self.dists = {}
-    for name,packagelist in collection.iteritems():
-      for d in packagelist.iterkeys():
-        self.dists[d] = True
+    self.dists = ReleaseList("debian")
     sl = {}
     sl['crumbs'] = [("index/"+letter,"index for "+letter)]
     tpl = os.sep.join([Conf.template_dir, "letter_toc.tmpl"])
@@ -61,12 +59,10 @@ class SearchResultsTemplate(OurTemplate):
   def __init__(self, search, collection):
     self.pkgs = {}
     self.idx = search
-    self.dists = {}
+    self.dists = ReleaseList("debian")
     for idx in collection.indices():
       for name,packagelist in collection.getletter(idx).iteritems():
         self.pkgs[name] = packagelist
-        for d in packagelist.iterkeys():
-          self.dists[d] = True
     tpl = os.sep.join([Conf.template_dir, "searchresults.tmpl"])
     OurTemplate.__init__(self, file=tpl)