dak/override.py

   1 #!/usr/bin/env python
   2
   3 """ Microscopic modification and query tool for overrides in projectb """
   4 # Copyright (C) 2004, 2006  Daniel Silverstone <dsilvers@digital-scurf.org>
   5
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20
  21 ################################################################################
  22 ## So line up your soldiers and she'll shoot them all down
  23 ## Coz Alisha Rules The World
  24 ## You think you found a dream, then it shatters and it seems,
  25 ## That Alisha Rules The World
  26 ################################################################################
  27
  28 import os
  29 import sys
  30 import apt_pkg
  31
  32 from daklib.config import Config
  33 from daklib.dbconn import *
  34 from daklib import daklog
  35 from daklib import utils
  36
  37 ################################################################################
  38
  39 # Shamelessly stolen from 'dak rm'. Should probably end up in utils.py
  40 def game_over():
  41     answer = utils.our_raw_input("Continue (y/N)? ").lower()
  42     if answer != "y":
  43         print "Aborted."
  44         sys.exit(1)
  45
  46
  47 def usage (exit_code=0):
  48     print """Usage: dak override [OPTIONS] package [section] [priority]
  49 Make microchanges or microqueries of the binary overrides
  50
  51   -h, --help                 show this help and exit
  52   -c, --check                check override compliance
  53   -d, --done=BUG#            send priority/section change as closure to bug#
  54   -n, --no-action            don't do anything
  55   -s, --suite                specify the suite to use
  56 """
  57     sys.exit(exit_code)
  58
  59 def check_override_compliance(package, priority, suite, cnf, session):
  60     print "Checking compliance with related overrides..."
  61
  62     depends = set()
  63     rdepends = set()
  64     components = get_component_names(session)
  65     arches = set([x.arch_string for x in get_suite_architectures(suite)])
  66     arches -= set(["source", "all"])
  67     for arch in arches:
  68         for component in components:
  69             Packages = utils.get_packages_from_ftp(cnf['Dir::Root'], suite, component, arch)
  70             while Packages.Step():
  71                 package_name = Packages.Section.Find("Package")
  72                 dep_list = Packages.Section.Find("Depends")
  73                 if dep_list:
  74                     if package_name == package:
  75                         for d in apt_pkg.ParseDepends(dep_list):
  76                             for i in d:
  77                                 depends.add(i[0])
  78                     else:
  79                         for d in apt_pkg.ParseDepends(dep_list):
  80                             for i in d:
  81                                 if i[0] == package:
  82                                     rdepends.add(package_name)
  83
  84     query = """SELECT o.package, p.level, p.priority
  85                FROM override o
  86                JOIN suite s ON s.id = o.suite
  87                JOIN priority p ON p.id = o.priority
  88                WHERE s.suite_name = '%s'
  89                AND o.package in ('%s')""" \
  90                % (suite, "', '".join(depends.union(rdepends)))
  91     packages = session.execute(query)
  92
  93     excuses = []
  94     for p in packages:
  95         if p[0] == package or not p[1]:
  96             continue
  97         if p[0] in depends:
  98             if priority.level < p[1]:
  99                 excuses.append("%s would have priority %s, its dependency %s has priority %s" \
 100                       % (package, priority.priority, p[0], p[2]))
 101         if p[0] in rdepends:
 102             if priority.level > p[1]:
 103                 excuses.append("%s would have priority %s, its reverse dependency %s has priority %s" \
 104                       % (package, priority.priority, p[0], p[2]))
 105
 106     if excuses:
 107         for ex in excuses:
 108             print ex
 109     else:
 110         print "Proposed override change complies with Debian Policy"
 111
 112 def main ():
 113     cnf = Config()
 114
 115     Arguments = [('h',"help","Override::Options::Help"),
 116                  ('c',"check","Override::Options::Check"),
 117                  ('d',"done","Override::Options::Done", "HasArg"),
 118                  ('n',"no-action","Override::Options::No-Action"),
 119                  ('s',"suite","Override::Options::Suite", "HasArg"),
 120                  ]
 121     for i in ["help", "check", "no-action"]:
 122         if not cnf.has_key("Override::Options::%s" % (i)):
 123             cnf["Override::Options::%s" % (i)] = ""
 124     if not cnf.has_key("Override::Options::Suite"):
 125         cnf["Override::Options::Suite"] = "unstable"
 126
 127     arguments = apt_pkg.ParseCommandLine(cnf.Cnf, Arguments, sys.argv)
 128     Options = cnf.SubTree("Override::Options")
 129
 130     if Options["Help"]:
 131         usage()
 132
 133     session = DBConn().session()
 134
 135     if not arguments:
 136         utils.fubar("package name is a required argument.")
 137
 138     package = arguments.pop(0)
 139     suite = Options["Suite"]
 140     if arguments and len(arguments) > 2:
 141         utils.fubar("Too many arguments")
 142
 143     if arguments and len(arguments) == 1:
 144         # Determine if the argument is a priority or a section...
 145         arg = arguments.pop()
 146         q = session.execute("""
 147         SELECT ( SELECT COUNT(*) FROM section WHERE section = :arg ) AS secs,
 148                ( SELECT COUNT(*) FROM priority WHERE priority = :arg ) AS prios
 149                """, {'arg': arg})
 150         r = q.fetchall()
 151         if r[0][0] == 1:
 152             arguments = (arg, ".")
 153         elif r[0][1] == 1:
 154             arguments = (".", arg)
 155         else:
 156             utils.fubar("%s is not a valid section or priority" % (arg))
 157
 158     # Retrieve current section/priority...
 159     oldsection, oldsourcesection, oldpriority = None, None, None
 160     for packagetype in ['source', 'binary']:
 161         eqdsc = '!='
 162         if packagetype == 'source':
 163             eqdsc = '='
 164         q = session.execute("""
 165     SELECT priority.priority AS prio, section.section AS sect, override_type.type AS type
 166       FROM override, priority, section, suite, override_type
 167      WHERE override.priority = priority.id
 168        AND override.type = override_type.id
 169        AND override_type.type %s 'dsc'
 170        AND override.section = section.id
 171        AND override.package = :package
 172        AND override.suite = suite.id
 173        AND suite.suite_name = :suite
 174         """ % (eqdsc), {'package': package, 'suite': suite})
 175
 176         if q.rowcount == 0:
 177             continue
 178         if q.rowcount > 1:
 179             utils.fubar("%s is ambiguous. Matches %d packages" % (package,q.rowcount))
 180
 181         r = q.fetchone()
 182         if packagetype == 'binary':
 183             oldsection = r[1]
 184             oldpriority = r[0]
 185         else:
 186             oldsourcesection = r[1]
 187             oldpriority = 'source'
 188
 189     if not oldpriority and not oldsourcesection:
 190         utils.fubar("Unable to find package %s" % (package))
 191
 192     if oldsection and oldsourcesection and oldsection != oldsourcesection:
 193         # When setting overrides, both source & binary will become the same section
 194         utils.warn("Source is in section '%s' instead of '%s'" % (oldsourcesection, oldsection))
 195
 196     if not oldsection:
 197         oldsection = oldsourcesection
 198
 199     if not arguments:
 200         print "%s is in section '%s' at priority '%s'" % (
 201             package, oldsection, oldpriority)
 202         sys.exit(0)
 203
 204     # At this point, we have a new section and priority... check they're valid...
 205     newsection, newpriority = arguments
 206
 207     if newsection == ".":
 208         newsection = oldsection
 209     if newpriority == ".":
 210         newpriority = oldpriority
 211
 212     s = get_section(newsection, session)
 213     if s is None:
 214         utils.fubar("Supplied section %s is invalid" % (newsection))
 215     newsecid = s.section_id
 216
 217     p = get_priority(newpriority, session)
 218     if p is None:
 219         utils.fubar("Supplied priority %s is invalid" % (newpriority))
 220     newprioid = p.priority_id
 221
 222     if newpriority == oldpriority and newsection == oldsection:
 223         print "I: Doing nothing"
 224         sys.exit(0)
 225
 226     if oldpriority == 'source' and newpriority != 'source':
 227         utils.fubar("Trying to change priority of a source-only package")
 228
 229     if Options["Check"] and newpriority != oldpriority:
 230         check_override_compliance(package, p, suite, cnf, session)
 231
 232     # If we're in no-action mode
 233     if Options["No-Action"]:
 234         if newpriority != oldpriority:
 235             print "I: Would change priority from %s to %s" % (oldpriority,newpriority)
 236         if newsection != oldsection:
 237             print "I: Would change section from %s to %s" % (oldsection,newsection)
 238         if Options.has_key("Done"):
 239             print "I: Would also close bug(s): %s" % (Options["Done"])
 240
 241         sys.exit(0)
 242
 243     if newpriority != oldpriority:
 244         print "I: Will change priority from %s to %s" % (oldpriority,newpriority)
 245
 246     if newsection != oldsection:
 247         print "I: Will change section from %s to %s" % (oldsection,newsection)
 248
 249     if not Options.has_key("Done"):
 250         pass
 251         #utils.warn("No bugs to close have been specified. Noone will know you have done this.")
 252     else:
 253         print "I: Will close bug(s): %s" % (Options["Done"])
 254
 255     game_over()
 256
 257     Logger = daklog.Logger("override")
 258
 259     dsc_otype_id = get_override_type('dsc').overridetype_id
 260
 261     # We're already in a transaction
 262     # We're in "do it" mode, we have something to do... do it
 263     if newpriority != oldpriority:
 264         session.execute("""
 265         UPDATE override
 266            SET priority = :newprioid
 267          WHERE package = :package
 268            AND override.type != :otypedsc
 269            AND suite = (SELECT id FROM suite WHERE suite_name = :suite)""",
 270            {'newprioid': newprioid, 'package': package,
 271             'otypedsc':  dsc_otype_id, 'suite': suite})
 272
 273         Logger.log(["changed priority", package, oldpriority, newpriority])
 274
 275     if newsection != oldsection:
 276         q = session.execute("""
 277         UPDATE override
 278            SET section = :newsecid
 279          WHERE package = :package
 280            AND suite = (SELECT id FROM suite WHERE suite_name = :suite)""",
 281            {'newsecid': newsecid, 'package': package,
 282             'suite': suite})
 283
 284         Logger.log(["changed section", package, oldsection, newsection])
 285
 286     session.commit()
 287
 288     if Options.has_key("Done"):
 289         if not cnf.has_key("Dinstall::BugServer"):
 290             utils.warn("Asked to send Done message but Dinstall::BugServer is not configured")
 291             Logger.close()
 292             return
 293
 294         Subst = {}
 295         Subst["__OVERRIDE_ADDRESS__"] = cnf["Dinstall::MyEmailAddress"]
 296         Subst["__BUG_SERVER__"] = cnf["Dinstall::BugServer"]
 297         bcc = []
 298         if cnf.Find("Dinstall::Bcc") != "":
 299             bcc.append(cnf["Dinstall::Bcc"])
 300         if bcc:
 301             Subst["__BCC__"] = "Bcc: " + ", ".join(bcc)
 302         else:
 303             Subst["__BCC__"] = "X-Filler: 42"
 304         if cnf.has_key("Dinstall::PackagesServer"):
 305             Subst["__CC__"] = "Cc: " + package + "@" + cnf["Dinstall::PackagesServer"] + "\nX-DAK: dak override"
 306         else:
 307             Subst["__CC__"] = "X-DAK: dak override"
 308         Subst["__ADMIN_ADDRESS__"] = cnf["Dinstall::MyAdminAddress"]
 309         Subst["__DISTRO__"] = cnf["Dinstall::MyDistribution"]
 310         Subst["__WHOAMI__"] = utils.whoami()
 311         Subst["__SOURCE__"] = package
 312
 313         summary = "Concerning package %s...\n" % (package)
 314         summary += "Operating on the %s suite\n" % (suite)
 315         if newpriority != oldpriority:
 316             summary += "Changed priority from %s to %s\n" % (oldpriority,newpriority)
 317         if newsection != oldsection:
 318             summary += "Changed section from %s to %s\n" % (oldsection,newsection)
 319         Subst["__SUMMARY__"] = summary
 320
 321         template = os.path.join(cnf["Dir::Templates"], "override.bug-close")
 322         for bug in utils.split_args(Options["Done"]):
 323             Subst["__BUG_NUMBER__"] = bug
 324             mail_message = utils.TemplateSubst(Subst, template)
 325             utils.send_mail(mail_message)
 326             Logger.log(["closed bug", bug])
 327
 328     Logger.close()
 329
 330 #################################################################################
 331
 332 if __name__ == '__main__':
 333     main()