]> git.decadent.org.uk Git - dak.git/blobdiff - natalie.py
added extraoverrides (for Task: fields)
[dak.git] / natalie.py
index 8206ddfe8715705457f1432def9614a9559fe1e7..760a574f1c0c539e640edbd183afa7c56177b49b 100755 (executable)
@@ -1,8 +1,8 @@
 #!/usr/bin/env python
 
 # Manipulate override files
-# Copyright (C) 2000  James Troup <james@nocrew.org>
-# $Id: natalie.py,v 1.1 2001-01-10 05:58:26 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;
 
@@ -31,27 +31,43 @@ projectB = None;
 
 ################################################################################
 
-def init():
+def usage (exit_code):
+    print """Usage: natalie.py [OPTIONS]
+  -D, --debug=VALUE        debug
+  -h, --help               this help
+  -V, --version            retrieve version
+  -c, --component=CMPT     list/set overrides by component
+                                  (contrib,*main,non-free)
+  -s, --suite=SUITE        list/set overrides by suite 
+                                  (experimental,stable,testing,*unstable)
+  -t, --type=TYPE          list/set overrides by type
+                                  (*deb,dsc,udeb)
+  -S, --set                set overrides from stdin
+  -l, --list               list overrides on stdout
+
+ starred (*) values are default"""
+    sys.exit(exit_code)
+
+################################################################################
+
+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
@@ -85,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";
@@ -96,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] = "";
@@ -153,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));
@@ -194,14 +207,16 @@ def main ():
                  ('t',"type","Natalie::Options::Type", "HasArg")];
     file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
 
+    if Cnf["Natalie::Options::Help"]:
+        usage(0);
+
     init();
 
     action = None;
     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"])