3 # General purpose package removal tool for ftpmaster
4 # Copyright (C) 2000, 2001 James Troup <james@nocrew.org>
5 # $Id: melanie,v 1.20 2001-11-19 03:56:29 rmurray Exp $
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.
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.
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
21 # X-Listening-To: Astronomy, Metallica - Garage Inc.
23 ################################################################################
25 import commands, os, pg, pwd, re, string, sys, tempfile
26 import utils, db_access
27 import apt_pkg, apt_inst;
29 ################################################################################
31 re_strip_source_version = re.compile (r'\s+.*$');
33 ################################################################################
38 ################################################################################
40 def usage (exit_code=0):
41 print """Usage: melanie [OPTIONS] PACKAGE[...]
42 Remove PACKAGE(s) from suite(s).
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
56 ARCH, BUG#, COMPONENT and SUITE can be space seperated lists, e.g.
57 --architecture=\"m68k i386\""""
61 ################################################################################
63 # "That's just fucking great! Game over, man! What the fuck are we
67 print "Continue (y/N)? ",
68 answer = string.lower(utils.our_raw_input());
73 ################################################################################
78 Cnf = utils.get_conf()
80 Arguments = [('h',"help","Melanie::Options::Help"),
81 ('a',"architecture","Melanie::Options::Architecture", "HasArg"),
82 ('b',"binary", "Melanie::Options::Binary-Only"),
83 ('c',"component", "Melanie::Options::Component", "HasArg"),
84 ('C',"carbon-copy", "Melanie::Options::Carbon-Copy", "HasArg"), # Bugs to Cc
85 ('d',"done","Melanie::Options::Done", "HasArg"), # Bugs fixed
86 ('m',"reason", "Melanie::Options::Reason", "HasArg"), # Hysterical raisins; -m is old-dinstall option for rejection reason
87 ('n',"no-action","Melanie::Options::No-Action"),
88 ('p',"partial", "Melanie::Options::Partial"),
89 ('s',"suite","Melanie::Options::Suite", "HasArg"),
90 ('S',"source-only", "Melanie::Options::Source-Only"),
93 for i in ["help", "architecture", "binary-only", "component", "carbon-copy", "done", "reason", "no-action", "partial", "source-only" ]:
94 if not Cnf.has_key("Melanie::Options::%s" % (i)):
95 Cnf["Melanie::Options::%s" % (i)] = "";
96 if not Cnf.has_key("Melanie::Options::Suite"):
97 Cnf["Melanie::Options::Suite"] = "unstable";
99 arguments = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
100 Options = Cnf.SubTree("Melanie::Options")
105 projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
106 db_access.init(Cnf, projectB);
108 # Sanity check options
110 utils.fubar("need at least one package name as an argument.");
111 if Options["Architecture"] and Options["Source-Only"]:
112 utils.fubar("can't use -a/--architecutre and -S/--source-only options simultaneously.");
113 if Options["Binary-Only"] and Options["Source-Only"]:
114 utils.fubar("can't use -b/--binary-only and -S/--source-only options simultaneously.");
115 if Options.has_key("Carbon-Copy") and not Options.has_key("Done"):
116 utils.fubar("can't use -C/--carbon-copy without also using -d/--done option.");
117 if Options["Architecture"] and not Options["Partial"]:
118 utils.warn("-a/--architecture implies -p/--partial.");
119 Options["Partial"] = "true";
121 # Process -C/--carbon-copy
123 # Accept 3 types of arguments (space separated):
124 # 1) a number - assumed to be a bug number, i.e. nnnnn@bugs.debian.org
125 # 2) the keyword 'package' - cc's $arch@packages.debian.org for every argument
126 # 3) contains a '@' - assumed to be an email address, used unmofidied
129 for copy_to in string.split(Options.get("Carbon-Copy")):
130 if utils.str_isnum(copy_to):
131 carbon_copy.append(copy_to + "@" + Cnf["Dinstall::BugServer"]);
132 elif copy_to == 'package':
133 for package in arguments:
134 carbon_copy.append(package + "@" + Cnf["Dinstall::PackagesServer"]);
136 carbon_copy.append(copy_to);
138 utils.fubar("Invalid -C/--carbon-copy argument '%s'; not a bug number, 'package' or email address." % (copy_to));
141 if Options["Binary-Only"]:
145 quoted_packages_list = []
146 for package in arguments:
147 quoted_packages_list.append("'%s'" % package)
148 packages[package] = "";
149 con_packages = "AND %s IN (%s)" % (field, string.join(quoted_packages_list, ", "));
152 for suite in string.split(Options["Suite"]):
154 if not Options["No-Action"] and suite == "stable":
155 print "**WARNING** About to remove from the stable suite!"
156 print "This should only be done just prior to a (point) release and not at"
157 print "any other time."
159 elif not Options["No-Action"] and suite == "testing":
160 print "**WARNING About to remove from the testing suite!"
161 print "There's no need to do this normally as removals from unstable will"
162 print "propogate to testing automagically."
165 suite_id = db_access.get_suite_id(suite);
167 utils.warn("suite '%s' not recognised." % (suite));
169 suites.append(suite);
170 suite_ids_list.append(suite_id);
172 suites_list = string.join(suites, ", ")
173 con_suites = "AND su.id IN (%s)" % string.join(map(str, suite_ids_list), ", ");
175 utils.fubar("No correct suite given.");
176 if Options["Component"]:
177 components_ids_list = [];
178 for component in string.split(Options["Component"]):
179 component_id = db_access.get_component_id(component);
180 if component_id == -1:
181 utils.warn("component '%s' not recognised." % (component));
183 components_ids_list.append(component_id);
184 con_components = "AND c.id IN (%s)" % string.join(map(str, components_ids_list), ", ");
185 over_con_components = "AND component IN (%s)" % string.join(map(str, components_ids_list), ", ");
188 over_con_components = "";
190 if Options["Architecture"]:
192 for architecture in string.split(Options["Architecture"]):
193 architecture_id = db_access.get_architecture_id(architecture);
194 if architecture_id == -1:
195 utils.warn("architecture '%s' not recognised." % (architecture));
197 arch_ids_list.append(architecture_id);
199 con_architectures = "AND a.id IN (%s)" % string.join(map(str, arch_ids_list), ", ");
201 utils.fubar("No correct architecture given");
203 con_architectures = "";
211 # We have 3 modes of package selection: binary-only, source-only
212 # and source+binary. The first two are trivial and obvious; the
213 # latter is a nasty mess, but very nice from a UI perspective so
214 # we try to support it.
216 if Options["Binary-Only"]:
218 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));
219 for i in q.getresult():
223 source_packages = {};
224 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));
225 for i in q.getresult():
226 source_packages[i[2]] = i[:2];
227 to_remove.append(i[2:]);
228 if not Options["Source-Only"]:
230 binary_packages = {};
231 # First get a list of binary package names we suspect are linked to the source
232 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));
233 for i in q.getresult():
234 binary_packages[i[0]] = "";
235 # Then parse each .dsc that we found earlier to see what binary packages it thinks it produces
236 for i in source_packages.keys():
237 filename = string.join(source_packages[i], '/');
239 dsc = utils.parse_changes(filename, 0);
240 except utils.cant_open_exc:
241 utils.warn("couldn't open '%s'." % (filename));
243 for package in string.split(dsc.get("binary"), ','):
244 package = string.strip(package);
245 binary_packages[package] = "";
246 # Then for each binary package: find any version in
247 # unstable, check the Source: field in the deb matches our
248 # source package and if so add it to the list of packages
250 for package in binary_packages.keys():
251 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));
252 for i in q.getresult():
253 filename = string.join(i[:2], '/');
254 control = apt_pkg.ParseSection(apt_inst.debExtractControl(utils.open_file(filename)))
255 source = control.Find("Source", control.Find("Package"));
256 source = re_strip_source_version.sub('', source);
257 if source_packages.has_key(source):
258 to_remove.append(i[2:]);
261 # If we don't have a reason; spawn an editor so the user can add one
262 # Write the rejection email out as the <foo>.reason file
263 if not Options["Reason"] and not Options["No-Action"]:
264 temp_filename = tempfile.mktemp();
265 fd = os.open(temp_filename, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0700);
267 editor = os.environ.get("EDITOR","vi")
268 result = os.system("%s %s" % (editor, temp_filename))
270 utils.fubar ("vi invocation failed for `%s'!" % (temp_filename), result)
271 file = utils.open_file(temp_filename);
272 for line in file.readlines():
273 Options["Reason"] = Options["Reason"] + line;
274 os.unlink(temp_filename);
276 # Generate the summary of what's to be removed
283 maintainers[maintainer] = "";
284 if not d.has_key(package):
286 if not d[package].has_key(version):
287 d[package][version] = [];
288 if architecture not in d[package][version]:
289 d[package][version].append(architecture);
291 maintainer_list = [];
292 for maintainer_id in maintainers.keys():
293 maintainer_list.append(db_access.get_maintainer(maintainer_id));
297 for package in packages:
298 versions = d[package].keys();
300 for version in versions:
301 summary = summary + "%10s | %10s | %s\n" % (package, version, string.join(d[package][version], ", "));
302 print "Will remove the following packages from %s:" % (suites_list);
305 print "Maintainer: %s" % string.join(maintainer_list, ", ")
307 print "Will also close bugs: "+Options["Done"];
309 print "Will also send CCs to:" + string.join(carbon_copy, ", ")
311 print "------------------- Reason -------------------"
312 print Options["Reason"];
313 print "----------------------------------------------"
316 # If -n/--no-action, drop out here
317 if Options["No-Action"]:
322 whoami = utils.whoami();
323 date = commands.getoutput('date -R');
325 # Log first; if it all falls apart I want a record that we at least tried.
326 logfile = utils.open_file(Cnf["Melanie::LogFile"], 'a');
327 logfile.write("=========================================================================\n");
328 logfile.write("[Date: %s] [ftpmaster: %s]\n" % (date, whoami));
329 logfile.write("Removed the following packages from %s:\n\n%s" % (suites_list, summary));
331 logfile.write("Closed bugs: %s\n" % (Options["Done"]));
332 logfile.write("\n------------------- Reason -------------------\n%s\n" % (Options["Reason"]));
333 logfile.write("----------------------------------------------\n");
336 dsc_type_id = db_access.get_override_type_id('dsc');
337 deb_type_id = db_access.get_override_type_id('deb');
339 # Do the actual deletion
342 projectB.query("BEGIN WORK");
347 for suite_id in suite_ids_list:
348 if architecture == "source":
349 projectB.query("DELETE FROM src_associations WHERE source = %s AND suite = %s" % (package_id, suite_id));
350 #print "DELETE FROM src_associations WHERE source = %s AND suite = %s" % (package_id, suite_id);
352 projectB.query("DELETE FROM bin_associations WHERE bin = %s AND suite = %s" % (package_id, suite_id));
353 #print "DELETE FROM bin_associations WHERE bin = %s AND suite = %s" % (package_id, suite_id);
354 # Delete from the override file
355 if not Options["Partial"]:
356 if architecture == "source":
357 type_id = dsc_type_id;
359 type_id = deb_type_id;
360 projectB.query("DELETE FROM override WHERE package = '%s' AND type = %s AND suite = %s %s" % (package, type_id, suite_id, over_con_components));
361 projectB.query("COMMIT WORK");
364 # Send the bug closing messages
367 Subst["__MELANIE_ADDRESS__"] = Cnf["Melanie::MyEmailAddress"];
368 Subst["__BUG_SERVER__"] = Cnf["Dinstall::BugServer"];
370 if Cnf.Find("Dinstall::Bcc") != "":
371 bcc.append(Cnf["Dinstall::Bcc"]);
372 if Cnf.Find("Melanie::Bcc") != "":
373 bcc.append(Cnf["Melanie::Bcc"]);
375 Subst["__BCC__"] = "Bcc: " + string.join(bcc, ", ");
377 Subst["__BCC__"] = "X-Filler: 42";
378 Subst["__CC__"] = "X-Melanie: $Revision: 1.20 $";
380 Subst["__CC__"] = Subst["__CC__"] + "\nCc: " + string.join(carbon_copy, ", ");
381 Subst["__SUITE_LIST__"] = suites_list;
382 Subst["__SUMMARY__"] = summary;
383 Subst["__ADMIN_ADDRESS__"] = Cnf["Dinstall::MyAdminAddress"];
384 Subst["__DISTRO__"] = Cnf["Dinstall::MyDistribution"];
385 Subst["__WHOAMI__"] = whoami;
386 whereami = utils.where_am_i();
387 Archive = Cnf.SubTree("Archive::%s" % (whereami));
388 Subst["__MASTER_ARCHIVE__"] = Archive["OriginServer"];
389 Subst["__PRIMARY_MIRROR__"] = Archive["PrimaryMirror"];
390 for bug in string.split(Options["Done"]):
391 Subst["__BUG_NUMBER__"] = bug;
392 mail_message = utils.TemplateSubst(Subst,open(Cnf["Dir::TemplatesDir"]+"/melanie.bug-close","r").read());
393 utils.send_mail (mail_message, "")
395 logfile.write("=========================================================================\n");
398 #######################################################################################
400 if __name__ == '__main__':