commonize all the diffhandler stuff into the SourcePackage model

1 files changed, 13 insertions, 0 deletions

diff --git a/patchtracker/models.py b/patchtracker/models.py
index 95a59df..3d4f582 100644
--- a/patchtracker/models.py
+++ b/patchtracker/models.py
@@ -1,5 +1,9 @@
+import os
 import re
 
+import Conf
+import DiffGzHandler, DebTarHandler
+
 from django.db import models
 from django.contrib import admin
 
@@ -24,6 +28,15 @@ class SourcePackage (models.Model):
     else:
       return "Native"
 
+  def diffhandler( self ):
+    """ Return a diffhandler object too the caller """
+    if self.diffgz_name:
+      dgz = os.sep.join([Conf.archive_root, self.loc, self.diffgz_name])
+      return DiffGzHandler.DiffGzHandler(dgz)
+    elif self.debtar_name:
+      dtar = os.sep.join([Conf.archive_root, self.loc, self.debtar_name])
+      return DebTarHandler.DebTarHandler(dtar)
+
   def anchor_name(self):
     """ output a version of the package name suitable for use as an
         anchor name (i.e. <a name="foo">) """