1 files changed, 9 insertions, 4 deletions
diff --git a/patchtracker/Templates.py b/patchtracker/Templates.py
index a30e2c0..42140a8 100755
--- a/patchtracker/Templates.py
+++ b/patchtracker/Templates.py
@@ -1,5 +1,6 @@
-from patchtracker import Conf
+from patchtracker import Conf, DB
 from patchtracker.DiffGzHandler import DiffGzHandler
+from patchtracker.DebTarHandler import DebTarHandler
 from patchtracker.SourceArchive import ReleaseList
 
 from Cheetah.Template import Template
@@ -34,11 +35,15 @@ class PackageVersTemplate(OurTemplate):
     self.src = srcpkg
     tpl=os.sep.join([Conf.template_dir, "package_vers.tmpl"])
     sl = {}
-    if srcpkg.diffgz_name:
-      dfile = os.sep.join([Conf.archive_root,srcpkg.loc,srcpkg.diffgz_name])
+    # XXX c/p from PatchCmd.make_diffhandler
+    db = DB.PatchTrackerDB()
+    dfile = db.findDiffGz(srcpkg.name,srcpkg.version)
+    if dfile:
       sl['diffhandler'] = DiffGzHandler(dfile)
     else:
-      sl['diffhandler'] = None
+      dfile = db.findDebTar(srcpkg.name, srcpkg.version)
+      if dfile:
+        sl['diffhandler'] = DebTarHandler(dfile)
     sl['crumbs'] = [("package/"+srcpkg.name,srcpkg.name),
                     ("package/"+srcpkg.name+"/"+srcpkg.version,
                      srcpkg.name+"/"+srcpkg.version)]