small fixups to the pts index feature

1 files changed, 4 insertions, 1 deletions

diff --git a/gen-patch-info.py b/gen-patch-info.py
index ef1621e..ea9cdb7 100755
--- a/gen-patch-info.py
+++ b/gen-patch-info.py
@@ -43,6 +43,8 @@ if __name__ == '__main__':
     if e.errno != errno.EEXIST:
       raise e
 
+  pts_idx = PtsIndexFile()
+
   print a
   for s in a.suites(filter=suites):
     print "suite: ",s
@@ -55,9 +57,10 @@ if __name__ == '__main__':
         db.saveSourcePackage(p)
         db.relateSourcePackage(name=p.name, version=p.version, suite=s,
                                component=c)
+        if gen_pts_index:
+          pts_idx.add(p) 
 
   if gen_pts_index:
-    pts_idx = PtsIndexFile(a)
     pts_fh = gzip.GzipFile(Conf.pts_index_file, mode="wb")
     simplejson.dump(pts_idx, pts_fh)
     pts_fh.close()