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