source: trunk/scripts/outage/outage-mail @ 2878

Last change on this file since 2878 was 2276, checked in by price, 16 years ago

outage-mail: send xvm@ just one message for the whole batch

File size: 2.4 KB
Line 
1#!/usr/bin/python
2"""
3
4VM names on stdin.
5"""
6
7import smtplib
8import optparse
9import sys
10from email.mime.text import MIMEText
11
12from invirt import database
13
14def send_mail(smtp, opts, message, vm):
15    contact = database.Machine.query.filter_by(name=vm).first().contact
16    if '@' not in contact:
17        contact += '@mit.edu'
18    msg = MIMEText(message % vm)
19    msg['To'] = contact
20    msg['From'] = opts.from_addr
21    msg['Reply-To'] = 'XVM <xvm@mit.edu>'
22    msg['Subject'] = opts.subject % vm
23    smtp.sendmail(opts.from_addr,
24        [contact],
25        msg.as_string())
26    return msg.as_string()
27
28def send_summary(smtp, opts, messages):
29    msg = MIMEText('\n\n\n'.join(messages))
30    msg['To'] = 'xvm@mit.edu'
31    msg['From'] = opts.from_addr
32    msg['Subject'] = ("xvm outage-mail summary (%d): %s"
33                      % (len(messages), opts.subject))
34    smtp.sendmail(opts.from_addr,
35        ['xvm@mit.edu'],
36        msg.as_string())
37
38def main(argv):
39    parser = optparse.OptionParser(
40        usage = '%prog {-m message, -f from} [options] <vm-list',
41        description = __doc__.strip())
42    parser.add_option('-m', '--message',
43            type = 'string',
44            dest = 'message',
45            help = 'filename with body of message')
46    parser.add_option('-s', '--subject',
47            type = 'string',
48            dest = 'subject',
49            default = '[xvm] Unexpected reboot of your VM %s',
50            help = 'subject line of message; %s for VM name')
51    parser.add_option('-f', '--from',
52            type = 'string',
53            dest = 'from_addr',
54            help = 'for From: and envelope-from; first word is %(sig)s')
55    parser.add_option('--host',
56            type = 'string',
57            dest = 'host',
58            help = 'host that failed; %(host)s in message')
59    parser.add_option('--time',
60            type = 'string',
61            dest = 'time',
62            help = 'time of failure; %(time)s in message')
63    opts, args = parser.parse_args()
64
65    if len(args) or not opts.message or not opts.from_addr:
66        parser.print_help(sys.stderr)
67        return 2
68
69    opts.sig = opts.from_addr.split()[0]
70    message = file(opts.message).read() % opts.__dict__
71
72    vms = sys.stdin.read().split()
73
74    database.connect()
75    s = smtplib.SMTP()
76    s.connect()
77
78    messages = []
79    for vm in vms:
80        messages.append(send_mail(s, opts, message, vm))
81    send_summary(s, opts, messages)
82
83    s.close()
84
85if __name__ == '__main__':
86    sys.exit(main(sys.argv))
Note: See TracBrowser for help on using the repository browser.