use a "marked" field to catch stale/removed pkgs

1 files changed, 7 insertions, 2 deletions

diff --git a/patchtracker/DB.py b/patchtracker/DB.py
index 119d97a..a81fb19 100644
--- a/patchtracker/DB.py
+++ b/patchtracker/DB.py
@@ -126,10 +126,10 @@ class PatchTrackerDB:
      
   def relateSourcePackage(self, name, version, suite, component):
     q = "INSERT INTO package_rel_map \
-           (package_id,suite_id,component_id) \
+           (package_id,suite_id,component_id,marked) \
            VALUES ((SELECT id FROM packages WHERE name=? AND version=?), \
                   (SELECT id FROM suites WHERE name=?), \
-                  (SELECT id FROM components WHERE name=?))"
+                  (SELECT id FROM components WHERE name=?), 1)"
     cursor = self.db.cursor()
     cursor.execute(q, (name, version, suite, component))
 
@@ -144,6 +144,11 @@ class PatchTrackerDB:
       return None
 
   def finalize(self):
+    q1 = "DELETE FROM package_rel_map WHERE marked != 1"
+    cursor = self.db.cursor()
+    cursor.execute(q1)
+    q2 = "UPDATE package_rel_map SET marked=0"
+    cursor.execute(q2)
     self.db.commit()
 
   def __del__(self):