3 # Manually reject packages for proprosed-updates
4 # Copyright (C) 2001, 2002, 2003, 2004 James Troup <james@nocrew.org>
5 # $Id: lauren,v 1.4 2004-04-01 17:13:11 troup 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 ################################################################################
24 import db_access, katie, logging, utils;
27 ################################################################################
30 lauren_version = "$Revision: 1.4 $";
38 ################################################################################
40 def usage(exit_code=0):
41 print """Usage: lauren .CHANGES[...]
42 Manually reject the .CHANGES file(s).
44 -h, --help show this help and exit.
45 -m, --message=MSG use this message for the rejection.
46 -s, --no-mail don't send any mail."""
49 ################################################################################
52 global Cnf, Logger, Options, projectB, Katie;
54 Cnf = utils.get_conf();
55 Arguments = [('h',"help","Lauren::Options::Help"),
56 ('m',"manual-reject","Lauren::Options::Manual-Reject", "HasArg"),
57 ('s',"no-mail", "Lauren::Options::No-Mail")];
58 for i in [ "help", "manual-reject", "no-mail" ]:
59 if not Cnf.has_key("Lauren::Options::%s" % (i)):
60 Cnf["Lauren::Options::%s" % (i)] = "";
62 arguments = apt_pkg.ParseCommandLine(Cnf, Arguments, sys.argv);
64 Options = Cnf.SubTree("Lauren::Options");
68 utils.fubar("need at least one .changes filename as an argument.");
70 projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
71 db_access.init(Cnf, projectB);
73 Katie = katie.Katie(Cnf);
74 Logger = Katie.Logger = logging.Logger(Cnf, "lauren");
76 bcc = "X-Katie: lauren %s" % (lauren_version);
77 if Cnf.has_key("Dinstall::Bcc"):
78 Katie.Subst["__BCC__"] = bcc + "\nBcc: %s" % (Cnf["Dinstall::Bcc"]);
80 Katie.Subst["__BCC__"] = bcc;
83 arg = utils.validate_changes_file_arg(arg);
84 Katie.pkg.changes_file = arg;
87 os.chdir(Cnf["Suite::Proposed-Updates::CopyKatie"]);
94 prompt = "Manual reject, [S]kip, Quit ?";
98 while prompt.find(answer) == -1:
99 answer = utils.our_raw_input(prompt);
100 m = katie.re_default_answer.search(prompt);
103 answer = answer[:1].upper()
106 aborted = reject(Options["Manual-Reject"]);
116 ################################################################################
118 def reject (reject_message = ""):
119 files = Katie.pkg.files;
121 changes_file = Katie.pkg.changes_file;
123 # If we weren't given a manual rejection message, spawn an editor
124 # so the user can add one in...
125 if not reject_message:
126 temp_filename = utils.temp_filename();
127 editor = os.environ.get("EDITOR","vi")
130 os.system("%s %s" % (editor, temp_filename))
131 file = utils.open_file(temp_filename);
132 reject_message = "".join(file.readlines());
134 print "Reject message:";
135 print utils.prefix_multi_line_string(reject_message," ", include_blank_lines=1);
136 prompt = "[R]eject, Edit, Abandon, Quit ?"
138 while prompt.find(answer) == -1:
139 answer = utils.our_raw_input(prompt);
140 m = katie.re_default_answer.search(prompt);
143 answer = answer[:1].upper();
144 os.unlink(temp_filename);
152 # Reject the .changes file
153 Katie.force_reject([changes_file]);
155 # Setup the .reason file
156 reason_filename = changes_file[:-8] + ".reason";
157 reject_filename = Cnf["Dir::Queue::Reject"] + '/' + reason_filename;
159 # If we fail here someone is probably trying to exploit the race
160 # so let's just raise an exception ...
161 if os.path.exists(reject_filename):
162 os.unlink(reject_filename);
163 reject_fd = os.open(reject_filename, os.O_RDWR|os.O_CREAT|os.O_EXCL, 0644);
165 # Build up the rejection email
166 user_email_address = utils.whoami() + " <%s>" % (Cnf["Dinstall::MyAdminAddress"]);
168 Katie.Subst["__REJECTOR_ADDRESS__"] = user_email_address;
169 Katie.Subst["__MANUAL_REJECT_MESSAGE__"] = reject_message;
170 Katie.Subst["__STABLE_REJECTOR__"] = Cnf["Lauren::StableRejector"];
171 Katie.Subst["__MORE_INFO_URL__"] = Cnf["Lauren::MoreInfoURL"];
172 Katie.Subst["__CC__"] = "Cc: " + Cnf["Dinstall::MyEmailAddress"];
173 reject_mail_message = utils.TemplateSubst(Katie.Subst,Cnf["Dir::Templates"]+"/lauren.stable-rejected");
175 # Write the rejection email out as the <foo>.reason file
176 os.write(reject_fd, reject_mail_message);
179 # Remove the packages from proposed-updates
180 suite_id = db_access.get_suite_id('proposed-updates');
182 projectB.query("BEGIN WORK");
183 # Remove files from proposed-updates suite
184 for file in files.keys():
185 if files[file]["type"] == "dsc":
186 package = dsc["source"];
187 version = dsc["version"]; # NB: not files[file]["version"], that has no epoch
188 q = projectB.query("SELECT id FROM source WHERE source = '%s' AND version = '%s'" % (package, version));
191 utils.fubar("reject: Couldn't find %s_%s in source table." % (package, version));
192 source_id = ql[0][0];
193 projectB.query("DELETE FROM src_associations WHERE suite = '%s' AND source = '%s'" % (suite_id, source_id));
194 elif files[file]["type"] == "deb":
195 package = files[file]["package"];
196 version = files[file]["version"];
197 architecture = files[file]["architecture"];
198 q = projectB.query("SELECT b.id FROM binaries b, architecture a WHERE b.package = '%s' AND b.version = '%s' AND (a.arch_string = '%s' OR a.arch_string = 'all') AND b.architecture = a.id" % (package, version, architecture));
201 # Horrible hack to work around partial replacement of
202 # packages with newer versions (from different source
203 # packages). This, obviously, should instead check for a
204 # newer version of the package and only do the
205 # warn&continue thing if it finds one.
207 utils.warn("reject: Couldn't find %s_%s_%s in binaries table." % (package, version, architecture));
209 binary_id = ql[0][0];
210 projectB.query("DELETE FROM bin_associations WHERE suite = '%s' AND bin = '%s'" % (suite_id, binary_id));
211 projectB.query("COMMIT WORK");
213 # Send the rejection mail if appropriate
214 if not Options["No-Mail"]:
215 utils.send_mail(reject_mail_message);
217 # Finally remove the .katie file
218 katie_file = os.path.join(Cnf["Suite::Proposed-Updates::CopyKatie"], os.path.basename(changes_file[:-8]+".katie"));
219 os.unlink(katie_file);
221 Logger.log(["rejected", changes_file]);
224 ################################################################################
226 if __name__ == '__main__':