added extraoverrides (for Task: fields)

[dak.git] / heidi

diff --git a/heidi b/heidi
index 309a4e94e4755802b6bdc3a0c92a9ccee96c9b69..9dff02447a58b21fbcad8879c08dcfa88c82209c 100755 (executable)
--- a/heidi
+++ b/heidi
@@ -1,8 +1,8 @@
 #!/usr/bin/env python
 
 # Manipulate suite tags
-# Copyright (C) 2000  James Troup <james@nocrew.org>
-# $Id: heidi,v 1.3 2001-01-10 06:08:03 troup Exp $
+# Copyright (C) 2000, 2001  James Troup <james@nocrew.org>
+# $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):
@@ -160,7 +166,7 @@ def main ():
 
     file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
 
-    projectB = pg.connect('projectb', 'localhost');
+    projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"],int(Cnf["DB::Port"]));
 
     db_access.init(Cnf, projectB);
 
@@ -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);