]> git.decadent.org.uk Git - dak.git/blobdiff - dak/control_overrides.py
Merge branch 'master' into content_generation
[dak.git] / dak / control_overrides.py
index cf33e26f05fa6a2273781708a0035714a2dbcf05..5d6ba46ce6eefe33bd2b1b1ef8b8e1691bcc56c9 100755 (executable)
@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Bulk manipulation of the overrides
+""" Bulk manipulation of the overrides """
 # Copyright (C) 2000, 2001, 2002, 2003, 2006  James Troup <james@nocrew.org>
 
 # This program is free software; you can redistribute it and/or modify
 ################################################################################
 
 # On 30 Nov 1998, James Troup wrote:
-# 
+#
 # > James Troup<2> <troup2@debian.org>
-# > 
+# >
 # >    James is a clone of James; he's going to take over the world.
 # >    After he gets some sleep.
-# 
+#
 # Could you clone other things too? Sheep? Llamas? Giant mutant turnips?
-# 
+#
 # Your clone will need some help to take over the world, maybe clone up an
 # army of penguins and threaten to unleash them on the world, forcing
 # governments to sway to the new James' will!
-# 
+#
 # Yes, I can envision a day when James' duplicate decides to take a horrific
 # vengance on the James that spawned him and unleashes his fury in the form
 # of thousands upon thousands of chickens that look just like Captin Blue
 # Eye! Oh the horror.
-# 
+#
 # Now you'll have to were name tags to people can tell you apart, unless of
 # course the new clone is truely evil in which case he should be easy to
 # identify!
-# 
+#
 # Jason
 # Chicken. Black. Helicopters.
 # Be afraid.
 ################################################################################
 
 import pg, sys, time
-import dak.lib.utils, dak.lib.database, dak.lib.logging
 import apt_pkg
+from daklib import utils
+from daklib import database
+from daklib import logging
+from daklib.regexes import re_comments
 
 ################################################################################
 
@@ -74,27 +77,29 @@ def usage (exit_code=0):
 
   -a, --add                add overrides (changes and deletions are ignored)
   -S, --set                set overrides
+  -C, --change             change overrides (additions and deletions are ignored)
   -l, --list               list overrides
 
   -q, --quiet              be less verbose
+  -n, --no-action          only list the action that would have been done
 
  starred (*) values are default"""
     sys.exit(exit_code)
 
 ################################################################################
 
-def process_file (file, suite, component, type, action):
-    suite_id = dak.lib.database.get_suite_id(suite)
+def process_file (file, suite, component, type, action, noaction=0):
+    suite_id = database.get_suite_id(suite)
     if suite_id == -1:
-        dak.lib.utils.fubar("Suite '%s' not recognised." % (suite))
+        utils.fubar("Suite '%s' not recognised." % (suite))
 
-    component_id = dak.lib.database.get_component_id(component)
+    component_id = database.get_component_id(component)
     if component_id == -1:
-        dak.lib.utils.fubar("Component '%s' not recognised." % (component))
+        utils.fubar("Component '%s' not recognised." % (component))
 
-    type_id = dak.lib.database.get_override_type_id(type)
+    type_id = database.get_override_type_id(type)
     if type_id == -1:
-        dak.lib.utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc.)" % (type))
+        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
@@ -114,13 +119,14 @@ def process_file (file, suite, component, type, action):
         original[i[0]] = i[1:]
 
     start_time = time.time()
-    projectB.query("BEGIN WORK")
+    if not noaction:
+        projectB.query("BEGIN WORK")
     for line in file.readlines():
-        line = dak.lib.utils.re_comments.sub('', line).strip()
+        line = re_comments.sub('', line).strip()
         if line == "":
             continue
 
-        maintainer_override = None
+        maintainer_override = ""
         if type == "dsc":
             split_line = line.split(None, 2)
             if len(split_line) == 2:
@@ -128,7 +134,7 @@ def process_file (file, suite, component, type, action):
             elif len(split_line) == 3:
                 (package, section, maintainer_override) = split_line
             else:
-                dak.lib.utils.warn("'%s' does not break into 'package section [maintainer-override]'." % (line))
+                utils.warn("'%s' does not break into 'package section [maintainer-override]'." % (line))
                 c_error += 1
                 continue
             priority = "source"
@@ -139,23 +145,23 @@ def process_file (file, suite, component, type, action):
             elif len(split_line) == 4:
                 (package, priority, section, maintainer_override) = split_line
             else:
-                dak.lib.utils.warn("'%s' does not break into 'package priority section [maintainer-override]'." % (line))
+                utils.warn("'%s' does not break into 'package priority section [maintainer-override]'." % (line))
                 c_error += 1
                 continue
 
-        section_id = dak.lib.database.get_section_id(section)
+        section_id = database.get_section_id(section)
         if section_id == -1:
-            dak.lib.utils.warn("'%s' is not a valid section. ['%s' in suite %s, component %s]." % (section, package, suite, component))
+            utils.warn("'%s' is not a valid section. ['%s' in suite %s, component %s]." % (section, package, suite, component))
             c_error += 1
             continue
-        priority_id = dak.lib.database.get_priority_id(priority)
+        priority_id = database.get_priority_id(priority)
         if priority_id == -1:
-            dak.lib.utils.warn("'%s' is not a valid priority. ['%s' in suite %s, component %s]." % (priority, package, suite, component))
+            utils.warn("'%s' is not a valid priority. ['%s' in suite %s, component %s]." % (priority, package, suite, component))
             c_error += 1
             continue
 
         if new.has_key(package):
-            dak.lib.utils.warn("Can't insert duplicate entry for '%s'; ignoring all but the first. [suite %s, component %s]" % (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 += 1
             continue
         new[package] = ""
@@ -163,8 +169,7 @@ def process_file (file, suite, component, type, action):
             (old_priority_id, old_section_id, old_maintainer_override, old_priority, old_section) = original[package]
             if action == "add" or old_priority_id == priority_id and \
                old_section_id == section_id and \
-               ((old_maintainer_override == maintainer_override) or \
-               (old_maintainer_override == "" and maintainer_override == None)):
+               old_maintainer_override == maintainer_override:
                 # If it's unchanged or we're in 'add only' mode, ignore it
                 c_skipped += 1
                 continue
@@ -172,8 +177,9 @@ def process_file (file, suite, component, type, action):
                 # If it's changed, delete the old one so we can
                 # reinsert it with the new information
                 c_updated += 1
-                projectB.query("DELETE FROM override WHERE suite = %s AND component = %s AND package = '%s' AND type = %s"
-                               % (suite_id, component_id, package, type_id))
+                if not noaction:
+                    projectB.query("DELETE FROM override WHERE suite = %s AND component = %s AND package = '%s' AND type = %s"
+                                   % (suite_id, component_id, package, type_id))
                 # Log changes
                 if old_priority_id != priority_id:
                     Logger.log(["changed priority",package,old_priority,priority])
@@ -182,64 +188,71 @@ def process_file (file, suite, component, type, action):
                 if old_maintainer_override != maintainer_override:
                     Logger.log(["changed maintainer override",package,old_maintainer_override,maintainer_override])
                 update_p = 1
+        elif action == "change":
+            # Ignore additions in 'change only' mode
+            c_skipped += 1
+            continue
         else:
             c_added += 1
             update_p = 0
 
-        if maintainer_override:
-            projectB.query("INSERT INTO override (suite, component, type, package, priority, section, maintainer) VALUES (%s, %s, %s, '%s', %s, %s, '%s')"
-                           % (suite_id, component_id, type_id, package, priority_id, section_id, maintainer_override))
-        else:
-            projectB.query("INSERT INTO override (suite, component, type, package, priority, section,maintainer) VALUES (%s, %s, %s, '%s', %s, %s, '')"
-                           % (suite_id, component_id, type_id, package, priority_id, section_id))
+        if not noaction:
+            if maintainer_override:
+                projectB.query("INSERT INTO override (suite, component, type, package, priority, section, maintainer) VALUES (%s, %s, %s, '%s', %s, %s, '%s')"
+                               % (suite_id, component_id, type_id, package, priority_id, section_id, maintainer_override))
+            else:
+                projectB.query("INSERT INTO override (suite, component, type, package, priority, section,maintainer) VALUES (%s, %s, %s, '%s', %s, %s, '')"
+                               % (suite_id, component_id, type_id, package, priority_id, section_id))
 
         if not update_p:
             Logger.log(["new override",suite,component,type,package,priority,section,maintainer_override])
 
-    if not action == "add":
+    if action == "set":
         # Delete any packages which were removed
         for package in original.keys():
             if not new.has_key(package):
-                projectB.query("DELETE FROM override WHERE suite = %s AND component = %s AND package = '%s' AND type = %s"
-                               % (suite_id, component_id, package, type_id))
+                if not noaction:
+                    projectB.query("DELETE FROM override WHERE suite = %s AND component = %s AND package = '%s' AND type = %s"
+                                   % (suite_id, component_id, package, type_id))
                 c_removed += 1
                 Logger.log(["removed override",suite,component,type,package])
 
-    projectB.query("COMMIT WORK")
+    if not noaction:
+        projectB.query("COMMIT WORK")
     if not Cnf["Control-Overrides::Options::Quiet"]:
         print "Done in %d seconds. [Updated = %d, Added = %d, Removed = %d, Skipped = %d, Errors = %d]" % (int(time.time()-start_time), c_updated, c_added, c_removed, c_skipped, c_error)
     Logger.log(["set complete",c_updated, c_added, c_removed, c_skipped, c_error])
 
 ################################################################################
 
-def list(suite, component, type):
-    suite_id = dak.lib.database.get_suite_id(suite)
+def list_overrides(suite, component, type):
+    suite_id = database.get_suite_id(suite)
     if suite_id == -1:
-        dak.lib.utils.fubar("Suite '%s' not recognised." % (suite))
+        utils.fubar("Suite '%s' not recognised." % (suite))
 
-    component_id = dak.lib.database.get_component_id(component)
+    component_id = database.get_component_id(component)
     if component_id == -1:
-        dak.lib.utils.fubar("Component '%s' not recognised." % (component))
+        utils.fubar("Component '%s' not recognised." % (component))
 
-    type_id = dak.lib.database.get_override_type_id(type)
+    type_id = database.get_override_type_id(type)
     if type_id == -1:
-        dak.lib.utils.fubar("Type '%s' not recognised. (Valid types are deb, udeb and dsc)" % (type))
+        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))
         for i in q.getresult():
-            print dak.lib.utils.result_join(i)
+            print utils.result_join(i)
     else:
         q = projectB.query("SELECT o.package, p.priority, s.section, o.maintainer, p.level FROM override o, priority p, section s WHERE o.suite = %s AND o.component = %s AND o.type = %s AND o.priority = p.id AND o.section = s.id ORDER BY s.section, p.level, o.package" % (suite_id, component_id, type_id))
         for i in q.getresult():
-            print dak.lib.utils.result_join(i[:-1])
+            print utils.result_join(i[:-1])
 
 ################################################################################
 
 def main ():
     global Cnf, projectB, Logger
 
-    Cnf = dak.lib.utils.get_conf()
+    Cnf = utils.get_conf()
     Arguments = [('a', "add", "Control-Overrides::Options::Add"),
                  ('c', "component", "Control-Overrides::Options::Component", "HasArg"),
                  ('h', "help", "Control-Overrides::Options::Help"),
@@ -247,18 +260,20 @@ def main ():
                  ('q', "quiet", "Control-Overrides::Options::Quiet"),
                  ('s', "suite", "Control-Overrides::Options::Suite", "HasArg"),
                  ('S', "set", "Control-Overrides::Options::Set"),
+                 ('C', "change", "Control-Overrides::Options::Change"),
+                 ('n', "no-action", "Control-Overrides::Options::No-Action"),
                  ('t', "type", "Control-Overrides::Options::Type", "HasArg")]
 
     # Default arguments
-    for i in [ "add", "help", "list", "quiet", "set" ]:
-       if not Cnf.has_key("Control-Overrides::Options::%s" % (i)):
-           Cnf["Control-Overrides::Options::%s" % (i)] = ""
+    for i in [ "add", "help", "list", "quiet", "set", "change", "no-action" ]:
+        if not Cnf.has_key("Control-Overrides::Options::%s" % (i)):
+            Cnf["Control-Overrides::Options::%s" % (i)] = ""
     if not Cnf.has_key("Control-Overrides::Options::Component"):
-       Cnf["Control-Overrides::Options::Component"] = "main"
+        Cnf["Control-Overrides::Options::Component"] = "main"
     if not Cnf.has_key("Control-Overrides::Options::Suite"):
-       Cnf["Control-Overrides::Options::Suite"] = "unstable"
+        Cnf["Control-Overrides::Options::Suite"] = "unstable"
     if not Cnf.has_key("Control-Overrides::Options::Type"):
-       Cnf["Control-Overrides::Options::Type"] = "deb"
+        Cnf["Control-Overrides::Options::Type"] = "deb"
 
     file_list = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv)
 
@@ -266,32 +281,39 @@ def main ():
         usage()
 
     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]))
-    dak.lib.database.init(Cnf, projectB)
+    database.init(Cnf, projectB)
 
     action = None
-    for i in [ "add", "list", "set" ]:
+    for i in [ "add", "list", "set", "change" ]:
         if Cnf["Control-Overrides::Options::%s" % (i)]:
             if action:
-                dak.lib.utils.fubar("Can not perform more than one action at once.")
+                utils.fubar("Can not perform more than one action at once.")
             action = i
 
-    (suite, component, type) = (Cnf["Control-Overrides::Options::Suite"],
-                                Cnf["Control-Overrides::Options::Component"],
-                                Cnf["Control-Overrides::Options::Type"])
+    (suite, component, otype) = (Cnf["Control-Overrides::Options::Suite"],
+                                 Cnf["Control-Overrides::Options::Component"],
+                                 Cnf["Control-Overrides::Options::Type"])
 
     if action == "list":
-        list(suite, component, type)
+        list_overrides(suite, component, otype)
     else:
-        Logger = dak.lib.logging.Logger(Cnf, "control-overrides")
+        if database.get_suite_untouchable(suite):
+            utils.fubar("%s: suite is untouchable" % suite)
+
+        noaction = 0
+        if Cnf["Control-Overrides::Options::No-Action"]:
+            utils.warn("In No-Action Mode")
+            noaction = 1
+
+        Logger = logging.Logger(Cnf, "control-overrides", noaction)
         if file_list:
-            for file in file_list:
-                process_file(dak.lib.utils.open_file(file), suite, component, type, action)
+            for f in file_list:
+                process_file(utils.open_file(f), suite, component, otype, action, noaction)
         else:
-            process_file(sys.stdin, suite, component, type, action)
+            process_file(sys.stdin, suite, component, otype, action, noaction)
         Logger.close()
 
 #######################################################################################
 
 if __name__ == '__main__':
     main()
-