melanie

   1 #!/usr/bin/env python
   2
   3 # General purpose package removal tool for ftpmaster
   4 # Copyright (C) 2000, 2001  James Troup <james@nocrew.org>
   5 # $Id: melanie,v 1.18 2001-11-04 22:35:11 troup Exp $
   6
   7 # This program is free software; you can redistribute it and/or modify
   8 # it under the terms of the GNU General Public License as published by
   9 # the Free Software Foundation; either version 2 of the License, or
  10 # (at your option) any later version.
  11
  12 # This program is distributed in the hope that it will be useful,
  13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 # GNU General Public License for more details.
  16
  17 # You should have received a copy of the GNU General Public License
  18 # along with this program; if not, write to the Free Software
  19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  20
  21 # X-Listening-To: Astronomy, Metallica - Garage Inc.
  22
  23 ################################################################################
  24
  25 import commands, os, pg, pwd, re, string, sys, tempfile
  26 import utils, db_access
  27 import apt_pkg, apt_inst;
  28
  29 ################################################################################
  30
  31 re_strip_source_version = re.compile (r'\s+.*$');
  32
  33 ################################################################################
  34
  35 Cnf = None;
  36 projectB = None;
  37
  38 ################################################################################
  39
  40 def usage (exit_code=0):
  41     print """Usage: melanie [OPTIONS] PACKAGE[...]
  42 Remove PACKAGE(s) from suite(s).
  43
  44   -a, --architecture=ARCH    only act on this architecture
  45   -b, --binary               remove binaries only
  46   -c, --component=COMPONENT  act on this component
  47   -C, --carbon-copy=EMAIL    send a CC of removal message to EMAIL
  48   -d, --done=BUG#            send removal message as closure to bug#
  49   -m, --reason=MSG           reason for removal
  50   -n, --no-action            don't do anything
  51   -p, --partial              don't affect override files
  52   -s, --suite=SUITE          act on this suite
  53   -S, --source-only          remove source only
  54   -h, --help                 show this help and exit
  55
  56 ARCH, BUG#, COMPONENT and SUITE can be space seperated lists, e.g.
  57     --architecture=\"m68k i386\""""
  58
  59     sys.exit(exit_code)
  60
  61 ################################################################################
  62
  63 # "That's just fucking great!  Game over, man!  What the fuck are we
  64 #  going to do now?"
  65
  66 def game_over():
  67     print "Continue (y/N)? ",
  68     answer = string.lower(utils.our_raw_input());
  69     if answer != "y":
  70         print "Aborted."
  71         sys.exit(1);
  72
  73 ################################################################################
  74
  75 def main ():
  76     global Cnf, projectB;
  77
  78     apt_pkg.init();
  79
  80     Cnf = apt_pkg.newConfiguration();
  81     apt_pkg.ReadConfigFileISC(Cnf,utils.which_conf_file());
  82
  83     Arguments = [('h',"help","Melanie::Options::Help"),
  84                  ('a',"architecture","Melanie::Options::Architecture", "HasArg"),
  85                  ('b',"binary", "Melanie::Options::Binary-Only"),
  86                  ('c',"component", "Melanie::Options::Component", "HasArg"),
  87                  ('C',"carbon-copy", "Melanie::Options::Carbon-Copy", "HasArg"), # Bugs to Cc
  88                  ('d',"done","Melanie::Options::Done", "HasArg"), # Bugs fixed
  89                  ('m',"reason", "Melanie::Options::Reason", "HasArg"), # Hysterical raisins; -m is old-dinstall option for rejection reason
  90                  ('n',"no-action","Melanie::Options::No-Action"),
  91                  ('p',"partial", "Melanie::Options::Partial"),
  92                  ('s',"suite","Melanie::Options::Suite", "HasArg"),
  93                  ('S',"source-only", "Melanie::Options::Source-Only"),
  94                  ];
  95
  96     for i in ["help", "architecture", "binary", "component", "carbon-copy", "done", "reason", "no-action", "partial", "source-only" ]:
  97         Cnf["Melanie::Options::%s" % (i)] = "";
  98     Cnf["Melanie::Options::Suite"] = "unstable";
  99
 100     arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
 101     Options = Cnf.SubTree("Melanie::Options")
 102
 103     if Options["Help"]:
 104         usage();
 105
 106     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
 107     db_access.init(Cnf, projectB);
 108
 109     # Sanity check options
 110     if not arguments:
 111         utils.fubar("need at least one package name as an argument.");
 112     if Options["Architecture"] and Options["Source-Only"]:
 113         utils.fubar("can't use -a/--architecutre and -S/--source-only options simultaneously.");
 114     if Options["Binary-Only"] and Options["Source-Only"]:
 115         utils.fubar("can't use -b/--binary-only and -S/--source-only options simultaneously.");
 116     if Options.has_key("Carbon-Copy") and not Options.has_key("Done"):
 117         utils.fubar("can't use -C/--carbon-copy without also using -d/--done option.");
 118     if Options["Architecture"] and not Options["Partial"]:
 119         utils.warn("-a/--architecture implies -p/--partial.");
 120         Options["Partial"] = "true";
 121
 122     # Process -C/--carbon-copy
 123     #
 124     # Accept 3 types of arguments (space separated):
 125     #  1) a number - assumed to be a bug number, i.e. nnnnn@bugs.debian.org
 126     #  2) the keyword 'package' - cc's $arch@packages.debian.org for every argument
 127     #  3) contains a '@' - assumed to be an email address, used unmofidied
 128     #
 129     carbon_copy = [];
 130     for copy_to in string.split(Options.get("Carbon-Copy")):
 131         if utils.str_isnum(copy_to):
 132             carbon_copy.append(copy_to + "@" + Cnf["Dinstall::BugServer"]);
 133         elif copy_to == 'package':
 134             for package in arguments:
 135                 carbon_copy.append(package + "@" + Cnf["Dinstall::PackagesServer"]);
 136         elif '@' in copy_to:
 137             carbon_copy.append(copy_to);
 138         else:
 139             utils.fubar("Invalid -C/--carbon-copy argument '%s'; not a bug number, 'package' or email address." % (copy_to));
 140
 141     packages = {};
 142     if Options["Binary-Only"]:
 143         field = "b.package";
 144     else:
 145         field = "s.source";
 146     quoted_packages_list = []
 147     for package in arguments:
 148         quoted_packages_list.append("'%s'" % package)
 149         packages[package] = "";
 150     con_packages = "AND %s IN (%s)" % (field, string.join(quoted_packages_list, ", "));
 151     suites = [];
 152     suite_ids_list = [];
 153     for suite in string.split(Options["Suite"]):
 154
 155         if not Options["No-Action"] and suite == "stable":
 156             print "**WARNING** About to remove from the stable suite!"
 157             print "This should only be done just prior to a (point) release and not at"
 158             print "any other time."
 159             game_over();
 160         elif not Options["No-Action"] and suite == "testing":
 161             print "**WARNING About to remove from the testing suite!"
 162             print "There's no need to do this normally as removals from unstable will"
 163             print "propogate to testing automagically."
 164             game_over();
 165
 166         suite_id = db_access.get_suite_id(suite);
 167         if suite_id == -1:
 168             utils.warn("suite '%s' not recognised." % (suite));
 169         else:
 170             suites.append(suite);
 171             suite_ids_list.append(suite_id);
 172     if suite_ids_list:
 173         suites_list = string.join(suites, ", ")
 174         con_suites = "AND su.id IN (%s)" % string.join(map(str, suite_ids_list), ", ");
 175     else:
 176         utils.fubar("No correct suite given.");
 177     if Options["Component"]:
 178         components_ids_list = [];
 179         for component in string.split(Options["Component"]):
 180             component_id = db_access.get_component_id(component);
 181             if component_id == -1:
 182                 utils.warn("component '%s' not recognised." % (component));
 183             else:
 184                 components_ids_list.append(component_id);
 185         con_components = "AND c.id IN (%s)" % string.join(map(str, components_ids_list), ", ");
 186         over_con_components = "AND component IN (%s)" % string.join(map(str, components_ids_list), ", ");
 187     else:
 188         con_components = "";
 189         over_con_components = "";
 190
 191     if Options["Architecture"]:
 192         arch_ids_list = [];
 193         for architecture in string.split(Options["Architecture"]):
 194             architecture_id = db_access.get_architecture_id(architecture);
 195             if architecture_id == -1:
 196                 utils.warn("architecture '%s' not recognised." % (architecture));
 197             else:
 198                 arch_ids_list.append(architecture_id);
 199         if arch_ids_list:
 200             con_architectures = "AND a.id IN (%s)" % string.join(map(str, arch_ids_list), ", ");
 201         else:
 202             utils.fubar("No correct architecture given");
 203     else:
 204         con_architectures = "";
 205
 206
 207     print "Working...",
 208     sys.stdout.flush();
 209     to_remove = [];
 210     maintainers = {};
 211
 212     # We have 3 modes of package selection: binary-only, source-only
 213     # and source+binary.  The first two are trivial and obvious; the
 214     # latter is a nasty mess, but very nice from a UI perspective so
 215     # we try to support it.
 216
 217     if Options["Binary-Only"]:
 218         # Binary-only
 219         q = projectB.query("SELECT b.package, b.version, a.arch_string, b.id, b.maintainer FROM binaries b, bin_associations ba, architecture a, suite su, files f, location l, component c WHERE ba.bin = b.id AND ba.suite = su.id AND b.architecture = a.id AND b.file = f.id AND f.location = l.id AND l.component = c.id %s %s %s %s" % (con_packages, con_suites, con_components, con_architectures));
 220         for i in q.getresult():
 221             to_remove.append(i);
 222     else:
 223         # Source-only
 224         source_packages = {};
 225         q = projectB.query("SELECT l.path, f.filename, s.source, s.version, 'source', s.id, s.maintainer FROM source s, src_associations sa, suite su, files f, location l, component c WHERE sa.source = s.id AND sa.suite = su.id AND s.file = f.id AND f.location = l.id AND l.component = c.id %s %s %s" % (con_packages, con_suites, con_components));
 226         for i in q.getresult():
 227             source_packages[i[2]] = i[:2];
 228             to_remove.append(i[2:]);
 229         if not Options["Source-Only"]:
 230             # Source + Binary
 231             binary_packages = {};
 232             # First get a list of binary package names we suspect are linked to the source
 233             q = projectB.query("SELECT DISTINCT package FROM binaries WHERE EXISTS (SELECT s.source, s.version, l.path, f.filename FROM source s, src_associations sa, suite su, files f, location l, component c WHERE binaries.source = s.id AND sa.source = s.id AND sa.suite = su.id AND s.file = f.id AND f.location = l.id AND l.component = c.id %s %s %s)" % (con_packages, con_suites, con_components));
 234             for i in q.getresult():
 235                 binary_packages[i[0]] = "";
 236             # Then parse each .dsc that we found earlier to see what binary packages it thinks it produces
 237             for i in source_packages.keys():
 238                 filename = string.join(source_packages[i], '/');
 239                 try:
 240                     dsc = utils.parse_changes(filename, 0);
 241                 except utils.cant_open_exc:
 242                     utils.warn("couldn't open '%s'." % (filename));
 243                     continue;
 244                 for package in string.split(dsc.get("binary"), ','):
 245                     package = string.strip(package);
 246                     binary_packages[package] = "";
 247             # Then for each binary package: find any version in
 248             # unstable, check the Source: field in the deb matches our
 249             # source package and if so add it to the list of packages
 250             # to be removed.
 251             for package in binary_packages.keys():
 252                 q = projectB.query("SELECT l.path, f.filename, b.package, b.version, a.arch_string, b.id, b.maintainer FROM binaries b, bin_associations ba, architecture a, suite su, files f, location l, component c WHERE ba.bin = b.id AND ba.suite = su.id AND b.architecture = a.id AND b.file = f.id AND f.location = l.id AND l.component = c.id %s %s %s AND b.package = '%s'" % (con_suites, con_components, con_architectures, package));
 253                 for i in q.getresult():
 254                     filename = string.join(i[:2], '/');
 255                     control = apt_pkg.ParseSection(apt_inst.debExtractControl(utils.open_file(filename)))
 256                     source = control.Find("Source", control.Find("Package"));
 257                     source = re_strip_source_version.sub('', source);
 258                     if source_packages.has_key(source):
 259                         to_remove.append(i[2:]);
 260     print "done."
 261
 262     # If we don't have a reason; spawn an editor so the user can add one
 263     # Write the rejection email out as the <foo>.reason file
 264     if not Options["Reason"] and not Options["No-Action"]:
 265         temp_filename = tempfile.mktemp();
 266         fd = os.open(temp_filename, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0700);
 267         os.close(fd);
 268         editor = os.environ.get("EDITOR","vi")
 269         result = os.system("%s %s" % (editor, temp_filename))
 270         if result != 0:
 271             utils.fubar ("vi invocation failed for `%s'!" % (temp_filename), result)
 272         file = utils.open_file(temp_filename);
 273         for line in file.readlines():
 274             Options["Reason"] = Options["Reason"] + line;
 275         os.unlink(temp_filename);
 276
 277     # Generate the summary of what's to be removed
 278     d = {};
 279     for i in to_remove:
 280         package = i[0];
 281         version = i[1];
 282         architecture = i[2];
 283         maintainer = i[4];
 284         maintainers[maintainer] = "";
 285         if not d.has_key(package):
 286             d[package] = {};
 287         if not d[package].has_key(version):
 288             d[package][version] = [];
 289         if architecture not in d[package][version]:
 290             d[package][version].append(architecture);
 291
 292     maintainer_list = [];
 293     for maintainer_id in maintainers.keys():
 294         maintainer_list.append(db_access.get_maintainer(maintainer_id));
 295     summary = "";
 296     packages = d.keys();
 297     packages.sort();
 298     for package in packages:
 299         versions = d[package].keys();
 300         versions.sort();
 301         for version in versions:
 302             summary = summary + "%10s | %10s | %s\n" % (package, version, string.join(d[package][version], ", "));
 303     print "Will remove the following packages from %s:" % (suites_list);
 304     print
 305     print summary
 306     print "Maintainer: %s" % string.join(maintainer_list, ", ")
 307     if Options["Done"]:
 308         print "Will also close bugs: "+Options["Done"];
 309     if carbon_copy:
 310         print "Will also send CCs to:" + string.join(carbon_copy, ", ")
 311     print
 312     print "------------------- Reason -------------------"
 313     print Options["Reason"];
 314     print "----------------------------------------------"
 315     print
 316
 317     # If -n/--no-action, drop out here
 318     if Options["No-Action"]:
 319         sys.exit(0);
 320
 321     game_over();
 322
 323     whoami = utils.whoami();
 324     date = commands.getoutput('date -R');
 325
 326     # Log first; if it all falls apart I want a record that we at least tried.
 327     logfile = utils.open_file(Cnf["Melanie::LogFile"], 'a');
 328     logfile.write("=========================================================================\n");
 329     logfile.write("[Date: %s] [ftpmaster: %s]\n" % (date, whoami));
 330     logfile.write("Removed the following packages from %s:\n\n%s" % (suites_list, summary));
 331     if Options["Done"]:
 332         logfile.write("Closed bugs: %s\n" % (Options["Done"]));
 333     logfile.write("\n------------------- Reason -------------------\n%s\n" % (Options["Reason"]));
 334     logfile.write("----------------------------------------------\n");
 335     logfile.flush();
 336
 337     dsc_type_id = db_access.get_override_type_id('dsc');
 338     deb_type_id = db_access.get_override_type_id('deb');
 339
 340     # Do the actual deletion
 341     print "Deleting...",
 342     sys.stdout.flush();
 343     projectB.query("BEGIN WORK");
 344     for i in to_remove:
 345         package = i[0];
 346         architecture = i[2];
 347         package_id = i[3];
 348         for suite_id in suite_ids_list:
 349             if architecture == "source":
 350                 projectB.query("DELETE FROM src_associations WHERE source = %s AND suite = %s" % (package_id, suite_id));
 351                 #print "DELETE FROM src_associations WHERE source = %s AND suite = %s" % (package_id, suite_id);
 352             else:
 353                 projectB.query("DELETE FROM bin_associations WHERE bin = %s AND suite = %s" % (package_id, suite_id));
 354                 #print "DELETE FROM bin_associations WHERE bin = %s AND suite = %s" % (package_id, suite_id);
 355             # Delete from the override file
 356             if not Options["Partial"]:
 357                 if architecture == "source":
 358                     type_id = dsc_type_id;
 359                 else:
 360                     type_id = deb_type_id;
 361                 projectB.query("DELETE FROM override WHERE package = '%s' AND type = %s AND suite = %s %s" % (package, type_id, suite_id, over_con_components));
 362     projectB.query("COMMIT WORK");
 363     print "done."
 364
 365     # Send the bug closing messages
 366     if Options["Done"]:
 367         Subst = {};
 368         Subst["__MELANIE_ADDRESS__"] = Cnf["Melanie::MyEmailAddress"];
 369         Subst["__BUG_SERVER__"] = Cnf["Dinstall::BugServer"];
 370         bcc = [];
 371         if Cnf.Find("Dinstall::Bcc") != "":
 372             bcc.append(Cnf["Dinstall::Bcc"]);
 373         if Cnf.Find("Melanie::Bcc") != "":
 374             bcc.append(Cnf["Melanie::Bcc"]);
 375         if bcc:
 376             Subst["__BCC__"] = "Bcc: " + string.join(bcc, ", ");
 377         else:
 378             Subst["__BCC__"] = "X-Filler: 42";
 379         Subst["__CC__"] = "X-Melanie: $Revision: 1.18 $";
 380         if carbon_copy:
 381             Subst["__CC__"] = Subst["__CC__"] + "\nCc: " + string.join(carbon_copy, ", ");
 382         Subst["__SUITE_LIST__"] = suites_list;
 383         Subst["__SUMMARY__"] = summary;
 384         Subst["__ADMIN_ADDRESS__"] = Cnf["Dinstall::MyAdminAddress"];
 385         Subst["__DISTRO__"] = Cnf["Dinstall::MyDistribution"];
 386         Subst["__WHOAMI__"] = whoami;
 387         whereami = utils.where_am_i();
 388         Archive = Cnf.SubTree("Archive::%s" % (whereami));
 389         Subst["__MASTER_ARCHIVE__"] = Archive["OriginServer"];
 390         Subst["__PRIMARY_MIRROR__"] = Archive["PrimaryMirror"];
 391         for bug in string.split(Options["Done"]):
 392             Subst["__BUG_NUMBER__"] = bug;
 393             mail_message = utils.TemplateSubst(Subst,open(Cnf["Dir::TemplatesDir"]+"/melanie.bug-close","r").read());
 394             utils.send_mail (mail_message, "")
 395
 396     logfile.write("=========================================================================\n");
 397     logfile.close();
 398
 399 #######################################################################################
 400
 401 if __name__ == '__main__':
 402     main()
 403