Mercurial > treepkg
view bin/sendnotificationmails.py @ 138:1bcdc8934b27
In PBuilder.run_script create the log file before starting pbuilder to
make sure the file is owned by the user running treepkg and not by root.
author | Bernhard Herzog <bh@intevation.de> |
---|---|
date | Mon, 02 Jun 2008 14:55:34 +0000 |
parents | aea6b97e7c68 |
children | df3065e4c76b |
line wrap: on
line source
#! /usr/bin/python2.4 # Copyright (C) 2008 by Intevation GmbH # Authors: # Bernhard Herzog <bh@intevation.de> # # This program is free software under the GPL (>=v2) # Read the file COPYING coming with the software for details. """Send pending notification mails""" import os import smtplib import email import email.Utils from optparse import OptionParser from ConfigParser import SafeConfigParser import treepkgcmd from treepkg.readconfig import read_config_section from treepkg.run import capture_output from treepkg.cmdexpand import cmdexpand notification_desc = ["build_user", "build_host", "build_listpending", "notification_template", "smtp_host", ("smtp_port", int), ] def read_config(filename): parser = SafeConfigParser() parser.read([filename]) return read_config_section(parser, "notification", notification_desc) def parse_commandline(): parser = OptionParser() parser.set_defaults(config_file=os.path.join(treepkgcmd.topdir, "notification.cfg")) parser.add_option("--config-file", help=("The configuration file." " Default notification.cfg")) return parser.parse_args() def send_mail(config, raw_message): msg = email.message_from_string(raw_message) sender = email.Utils.parseaddr(msg["From"])[1] recipients = [addr[1] for addr in email.Utils.getaddresses(msg.get_all("To", []) + msg.get_all("Cc", []))] server = smtplib.SMTP(config["smtp_host"], config["smtp_port"]) server.sendmail(sender, recipients, raw_message) server.quit() def send_notification_mails(config_filename): config = read_config(config_filename) template = open(config["notification_template"]).read() lines = capture_output(cmdexpand("ssh $build_user$@$build_host" " $build_listpending", **config)) for line in lines.splitlines(): words = line.split() if len(words) == 3: status, track, revision = words values = config.copy() values.update(locals()) send_mail(config, template % values) def main(): options, args = parse_commandline() send_notification_mails(options.config_file) main()