X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=natalie.py;h=760a574f1c0c539e640edbd183afa7c56177b49b;hb=4a37809c0bcc3a451d3d51b29e3fcc3cfe805097;hp=e92d6d7d676bdc97355ea7d19a4740ceedbf41db;hpb=779bb0691d139ff2487383c6aa9d87b149bad294;p=dak.git

diff --git a/natalie.py b/natalie.py
index e92d6d7d..760a574f 100755
--- a/natalie.py
+++ b/natalie.py
@@ -1,8 +1,8 @@
 #!/usr/bin/env python
 
 # Manipulate override files
-# Copyright (C) 2000  James Troup <james@nocrew.org>
-# $Id: natalie.py,v 1.2 2001-01-27 00:09:36 troup Exp $
+# Copyright (C) 2000, 2001  James Troup <james@nocrew.org>
+# $Id: natalie.py,v 1.6 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
@@ -20,7 +20,7 @@
 
 ################################################################################
 
-import pg, string, sys, time
+import errno, os, pg, pwd, string, sys, time
 import utils, db_access
 import apt_pkg;
 
@@ -50,27 +50,24 @@ def usage (exit_code):
 
 ################################################################################
 
-def init():
+def init ():
     global projectB;
     
-    projectB = pg.connect('projectb', 'localhost');
+    projectB = pg.connect('projectb', None);
     db_access.init(Cnf, projectB);
 
 def process_file (file, suite, component, type, action):
     suite_id = db_access.get_suite_id(suite);
     if suite_id == -1:
-        sys.stderr.write("Suite '%s' not recognised.\n" % (suite));
-        sys.exit(2);
+        utils.fubar("Suite '%s' not recognised." % (suite));
 
     component_id = db_access.get_component_id(component);
     if component_id == -1:
-        sys.stderr.write("Component '%s' not recognised.\n" % (component));
-        sys.exit(2);
+        utils.fubar("Component '%s' not recognised." % (component));
 
     type_id = db_access.get_override_type_id(type);
     if type_id == -1:
-        sys.stderr.write("Type '%s' not recognised. (Valid types are deb, udeb and dsc.)\n" % (type));
-        sys.exit(2);
+        utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc.)" % (type));
 
     # --set is done mostly internal for performance reasons; most
     # invocations of --set will be updates and making people wait 2-3
@@ -104,7 +101,7 @@ def process_file (file, suite, component, type, action):
             elif len(split_line) == 3:
                 (package, section, maintainer_override) = split_line;
             else:
-                sys.stderr.write("W: '%s' does not break into 'package section [maintainer override]'.\n" % (line));
+                utils.warn("'%s' does not break into 'package section [maintainer-override]'." % (line));
                 c_error = c_error + 1;
                 continue;
             priority = "source";
@@ -115,23 +112,23 @@ def process_file (file, suite, component, type, action):
             elif len(split_line) == 4:
                 (package, priority, section, maintainer_override) = split_line;
             else:
-                sys.stderr.write("W: '%s' does not break into 'package priority section [maintainer override]'.\n" % (line));
+                utils.warn("'%s' does not break into 'package priority section [maintainer-override]'." % (line));
                 c_error = c_error + 1;
                 continue;
 
         section_id = db_access.get_section_id(section);
         if section_id == -1:
-            sys.stderr.write("W: '%s' is not a valid section. ['%s' in suite %s, component %s].\n" % (section, package, suite, component));
+            utils.warn("'%s' is not a valid section. ['%s' in suite %s, component %s]." % (section, package, suite, component));
             c_error = c_error + 1;
             continue;
         priority_id = db_access.get_priority_id(priority);
         if priority_id == -1:
-            sys.stderr.write("W: '%s' is not a valid priority. ['%s' in suite %s, component %s].\n" % (priority, package, suite, component));
+            utils.warn("'%s' is not a valid priority. ['%s' in suite %s, component %s]." % (priority, package, suite, component));
             c_error = c_error + 1;
             continue;
 
         if new.has_key(package):
-            sys.stderr.write("W: Can't insert duplicate entry for '%s'; ignoring all but the first. [suite %s, component %s]\n" % (package, suite, component));
+            utils.warn("Can't insert duplicate entry for '%s'; ignoring all but the first. [suite %s, component %s]" % (package, suite, component));
             c_error = c_error + 1;
             continue;
         new[package] = "";
@@ -172,18 +169,15 @@ def process_file (file, suite, component, type, action):
 def list(suite, component, type):
     suite_id = db_access.get_suite_id(suite);
     if suite_id == -1:
-        sys.stderr.write("Suite '%s' not recognised.\n" % (suite));
-        sys.exit(2);
+        utils.fubar("Suite '%s' not recognised." % (suite));
 
     component_id = db_access.get_component_id(component);
     if component_id == -1:
-        sys.stderr.write("Component '%s' not recognised.\n" % (component));
-        sys.exit(2);
+        utils.fubar("Component '%s' not recognised." % (component));
 
     type_id = db_access.get_override_type_id(type);
     if type_id == -1:
-        sys.stderr.write("Type '%s' not recognised. (Valid types are deb, udeb and dsc.)\n" % (type));
-        sys.exit(2);
+        utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc)" % (type));
 
     if type == "dsc":
         q = projectB.query("SELECT o.package, s.section, o.maintainer FROM override o, section s WHERE o.suite = %s AND o.component = %s AND o.type = %s AND o.section = s.id ORDER BY s.section, o.package" % (suite_id, component_id, type_id));
@@ -222,8 +216,7 @@ def main ():
     for i in [ "list", "set" ]:
         if Cnf["Natalie::Options::%s" % (i)]:
             if action != None:
-                sys.stderr.write("Can not perform more than one action at once.\n");
-                sys.exit(2);
+                utils.fubar("Can not perform more than one action at once.");
             action = i;
 
     (suite, component, type) = (Cnf["Natalie::Options::Suite"], Cnf["Natalie::Options::Component"], Cnf["Natalie::Options::Type"])