lauren

   1 #!/usr/bin/env python
   2
   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 $
   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 import os, pg, sys;
  24 import db_access, katie, logging, utils;
  25 import apt_pkg;
  26
  27 ################################################################################
  28
  29 # Globals
  30 lauren_version = "$Revision: 1.4 $";
  31
  32 Cnf = None;
  33 Options = None;
  34 projectB = None;
  35 Katie = None;
  36 Logger = None;
  37
  38 ################################################################################
  39
  40 def usage(exit_code=0):
  41     print """Usage: lauren .CHANGES[...]
  42 Manually reject the .CHANGES file(s).
  43
  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."""
  47     sys.exit(exit_code)
  48
  49 ################################################################################
  50
  51 def main():
  52     global Cnf, Logger, Options, projectB, Katie;
  53
  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)] = "";
  61
  62     arguments = apt_pkg.ParseCommandLine(Cnf, Arguments, sys.argv);
  63
  64     Options = Cnf.SubTree("Lauren::Options");
  65     if Options["Help"]:
  66         usage();
  67     if not arguments:
  68         utils.fubar("need at least one .changes filename as an argument.");
  69
  70     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
  71     db_access.init(Cnf, projectB);
  72
  73     Katie = katie.Katie(Cnf);
  74     Logger = Katie.Logger = logging.Logger(Cnf, "lauren");
  75
  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"]);
  79     else:
  80         Katie.Subst["__BCC__"] = bcc;
  81
  82     for arg in arguments:
  83         arg = utils.validate_changes_file_arg(arg);
  84         Katie.pkg.changes_file = arg;
  85         Katie.init_vars();
  86         cwd = os.getcwd();
  87         os.chdir(Cnf["Suite::Proposed-Updates::CopyKatie"]);
  88         Katie.update_vars();
  89         os.chdir(cwd);
  90         Katie.update_subst();
  91
  92         print arg
  93         done = 0;
  94         prompt = "Manual reject, [S]kip, Quit ?";
  95         while not done:
  96             answer = "XXX";
  97
  98             while prompt.find(answer) == -1:
  99                 answer = utils.our_raw_input(prompt);
 100                 m = katie.re_default_answer.search(prompt);
 101                 if answer == "":
 102                     answer = m.group(1)
 103                 answer = answer[:1].upper()
 104
 105             if answer == 'M':
 106                 aborted = reject(Options["Manual-Reject"]);
 107                 if not aborted:
 108                     done = 1;
 109             elif answer == 'S':
 110                 done = 1;
 111             elif answer == 'Q':
 112                 sys.exit(0)
 113
 114     Logger.close();
 115
 116 ################################################################################
 117
 118 def reject (reject_message = ""):
 119     files = Katie.pkg.files;
 120     dsc = Katie.pkg.dsc;
 121     changes_file = Katie.pkg.changes_file;
 122
 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")
 128         answer = 'E';
 129         while answer == 'E':
 130             os.system("%s %s" % (editor, temp_filename))
 131             file = utils.open_file(temp_filename);
 132             reject_message = "".join(file.readlines());
 133             file.close();
 134             print "Reject message:";
 135             print utils.prefix_multi_line_string(reject_message,"  ", include_blank_lines=1);
 136             prompt = "[R]eject, Edit, Abandon, Quit ?"
 137             answer = "XXX";
 138             while prompt.find(answer) == -1:
 139                 answer = utils.our_raw_input(prompt);
 140                 m = katie.re_default_answer.search(prompt);
 141                 if answer == "":
 142                     answer = m.group(1);
 143                 answer = answer[:1].upper();
 144         os.unlink(temp_filename);
 145         if answer == 'A':
 146             return 1;
 147         elif answer == 'Q':
 148             sys.exit(0);
 149
 150     print "Rejecting.\n"
 151
 152     # Reject the .changes file
 153     Katie.force_reject([changes_file]);
 154
 155     # Setup the .reason file
 156     reason_filename = changes_file[:-8] + ".reason";
 157     reject_filename = Cnf["Dir::Queue::Reject"] + '/' + reason_filename;
 158
 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);
 164
 165     # Build up the rejection email
 166     user_email_address = utils.whoami() + " <%s>" % (Cnf["Dinstall::MyAdminAddress"]);
 167
 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");
 174
 175     # Write the rejection email out as the <foo>.reason file
 176     os.write(reject_fd, reject_mail_message);
 177     os.close(reject_fd);
 178
 179     # Remove the packages from proposed-updates
 180     suite_id = db_access.get_suite_id('proposed-updates');
 181
 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));
 189             ql = q.getresult();
 190             if not ql:
 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));
 199             ql = q.getresult();
 200
 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.
 206             if not ql:
 207                 utils.warn("reject: Couldn't find %s_%s_%s in binaries table." % (package, version, architecture));
 208             else:
 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");
 212
 213     # Send the rejection mail if appropriate
 214     if not Options["No-Mail"]:
 215         utils.send_mail(reject_mail_message);
 216
 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);
 220
 221     Logger.log(["rejected", changes_file]);
 222     return 0;
 223
 224 ################################################################################
 225
 226 if __name__ == '__main__':
 227     main();