outage-mail

   1 #!/usr/bin/python
   2 """
   3
   4 VM names on stdin.
   5 """
   6
   7 import smtplib
   8 import optparse
   9 import sys
  10 from email.mime.text import MIMEText
  11
  12 from invirt import database
  13
  14 def 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
  28 def 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
  38 def 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
  85 if __name__ == '__main__':
  86     sys.exit(main(sys.argv))