1 files changed, 9 insertions, 9 deletions
diff --git a/gen-patch-info.py b/gen-patch-info.py
index 3b5847b..17b2a2a 100755
--- a/gen-patch-info.py
+++ b/gen-patch-info.py
@@ -8,14 +8,14 @@ import gzip
 import errno
 import simplejson
 
+os.environ['DJANGO_SETTINGS_MODULE'] = 'settings'
+import patchtracker.models
+
 import patchtracker.Conf as Conf
-from patchtracker.SourceArchive import Archive, SourcePackage
-from patchtracker.DB import PatchTrackerDB
+from patchtracker.SourceArchive import Archive
 from patchtracker.PtsIndex import PtsIndexFile
 
 if __name__ == '__main__':
-  db = PatchTrackerDB()
-  os.system("cheetah compile templates/skeleton")
   opts,args = getopt.getopt(sys.argv[1:], "ais:p:")
   suites = None
   packages = None
@@ -43,15 +43,15 @@ if __name__ == '__main__':
   print a
   for s in a.suites(filter=suites):
     print "suite: ",s
-    db.saveSuite(s)
+    suite = patchtracker.models.RepositorySuite.objects.get(name=s)
     for c in a.components(s):
       print "\tcomponent:",c
-      db.saveComponent(c)
+      component = patchtracker.models.RepositoryComponent.objects.get(name=c)
       for p in a.sourcepackages(s, c, filter=packages):
         print "\t\tpackage:",p
-        db.saveSourcePackage(p)
-        db.relateSourcePackage(name=p.name, version=p.version, suite=s,
-                               component=c)
+        m = patchtracker.models.SourcePackageMapping.objects.get(suite=suite, component=component, package__name=p.name)
+        m.package = p
+        m.save()
         if gen_pts_index:
           pts_idx.add(p)