X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=heidi;h=9dff02447a58b21fbcad8879c08dcfa88c82209c;hb=89ada75793da6403e6ceb74a66a119621f19bba4;hp=c2f5071bb6b2dae1317c1f563239d1a13916c4ef;hpb=0ae0a3896e77f8e07a43d2a8b368466f4edabedd;p=dak.git

diff --git a/heidi b/heidi
index c2f5071b..9dff0244 100755
--- a/heidi
+++ b/heidi
@@ -2,7 +2,7 @@
 
 # Manipulate suite tags
 # Copyright (C) 2000, 2001  James Troup <james@nocrew.org>
-# $Id: heidi,v 1.5 2001-03-20 00:28:11 troup Exp $
+# $Id: heidi,v 1.7 2001-06-22 22:53:14 troup Exp $
 
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -55,15 +55,19 @@ projectB = None;
 
 def process_file (file, suite_id, action):
 
+    lines = file.readlines();
+
+    projectB.query("BEGIN WORK");
+
     if action == "set":
         projectB.query("DELETE FROM bin_associations WHERE suite = %s" % (suite_id));
         projectB.query("DELETE FROM src_associations WHERE suite = %s" % (suite_id));
         action = "add";
         
-    for line in file.readlines():
+    for line in lines:
         split_line = string.split(string.strip(line[:-1]));
         if len(split_line) != 3:
-            sys.stderr.write("W: '%s' does not break into 'package version architecture'.\n" % (line[:-1]));
+            utils.warn("'%s' does not break into 'package version architecture'." % (line[:-1]));
             continue;
         
         (package, version, architecture) = split_line;
@@ -75,10 +79,10 @@ def process_file (file, suite_id, action):
 
         ql = q.getresult();
         if ql == []:
-            sys.stderr.write("W: Couldn't find '%s~%s~%s'.\n" % (package, version, architecture));
+            utils.warn("Couldn't find '%s~%s~%s'." % (package, version, architecture));
             continue;
         if len(ql) > 1:
-            sys.stderr.write("E: Found more than one match for '%s~%s~%s'.\n" % (package, version, architecture));
+            utils.warn("Found more than one match for '%s~%s~%s'." % (package, version, architecture));
             continue;
         id = ql[0][0];
 
@@ -93,13 +97,13 @@ def process_file (file, suite_id, action):
             # Take action
             if action == "add":
                 if assoication_id != None:
-                    sys.stderr.write("W: '%s~%s~%s' already exists in suite %s.\n" % (package, version, architecture, suite_id));
+                    utils.warn("'%s~%s~%s' already exists in suite %s." % (package, version, architecture, suite_id));
                     continue;
                 else:
                     q = projectB.query("INSERT INTO src_associations (suite, source) VALUES (%s, %s)" % (suite_id, id));
             elif action == "remove":
                 if assoication_id == None:
-                    sys.stderr.write("W: '%s~%s~%s' doesn't exist in suite %s.\n" % (package, version, architecture, suite_id));
+                    utils.warn("'%s~%s~%s' doesn't exist in suite %s." % (package, version, architecture, suite_id));
                     continue;
                 else:
                     q = projectB.query("DELETE FROM src_associations WHERE id = %s" % (assoication_id));
@@ -114,17 +118,19 @@ def process_file (file, suite_id, action):
             # Take action
             if action == "add":
                 if assoication_id != None:
-                    sys.stderr.write("W: '%s~%s~%s' already exists in suite %s.\n" % (package, version, architecture, suite_id));
+                    utils.warn("'%s~%s~%s' already exists in suite %s." % (package, version, architecture, suite_id));
                     continue;
                 else:
                     q = projectB.query("INSERT INTO bin_associations (suite, bin) VALUES (%s, %s)" % (suite_id, id));
             elif action == "remove":
                 if assoication_id == None:
-                    sys.stderr.write("W: '%s~%s~%s' doesn't exist in suite %s.\n" % (package, version, architecture, suite_id));
+                    utils.warn("'%s~%s~%s' doesn't exist in suite %s." % (package, version, architecture, suite_id));
                     continue;
                 else:
                     q = projectB.query("DELETE FROM bin_associations WHERE id = %s" % (assoication_id));
-              
+
+    projectB.query("COMMIT WORK");
+
 #######################################################################################
 
 def get_list (suite_id):
@@ -170,24 +176,20 @@ def main ():
         suite = Cnf["Heidi::Options::%s" % (i)];
         if suite !="":
             if not Cnf.has_key("Suite::%s" % (suite)):
-                sys.stderr.write("Unknown suite %s.\n" %(suite));
-                sys.exit(2);
+                utils.fubar("Unknown suite '%s'." %(suite));
             else:
                 suite_id = db_access.get_suite_id(suite);
                 if action != None:
-                    sys.stderr.write("Can only do one action at a time.\n");
-                    sys.exit(2);
+                    utils.fubar("Can only perform one action at a time.");
                 action = i;
 
     # Need an action...
     if action == None:
-        sys.stderr.write("No action specified.\n");
-        sys.exit(2);
+        utils.fubar("No action specified.");
 
     # Safety/Sanity check
     if action == "set" and suite != "testing":
-        sys.stderr.write("Will not reset a suite other than testing...\n");
-        sys.exit(2);
+        utils.fubar("Will not reset a suite other than testing.");
 
     if action == "list":
         get_list(suite_id);