outage-mail: make subject line an option

[invirt/scripts/outage.git] / outage-mail

diff --git a/outage-mail b/outage-mail
index ed75737..84ba26e 100644 (file)
--- a/outage-mail
+++ b/outage-mail
@@ -1,46 +1,74 @@
 #!/usr/bin/python
-import smtplib
-from invirt.database import *
-from email.mime.text import MIMEText
-
-sender = 'Greg Price <price@mit.edu>'
-sendername = sender.split()[0]
-host = 'arklay-mansion'
+"""
 
-message = """\
-One of the four XVM host servers, %s, failed tonight at
-about 23:17 and was rebooted.  We're working to understand the cause.
+VM names on stdin.
+"""
 
-Your VM %%s was running on %s.  We have restarted it.
-
-We realize that this is an inconvenience for you, and we apologize for
-the unexpected downtime.
-
-- %s
-for the XVM team
-""" % (host, host, sendername)
+import smtplib
+import optparse
+import sys
+from email.mime.text import MIMEText
 
-vms = "amd64-test brain-trust cluster-test david-vista debathena-livecd-build ecprice gradbook greg-ns1 greg-ns2 groovy htns iodine jsoltren latex liftm medkaz mhd mksvn niska polarix price-lenny price-test8 qren2 shawn-of-awesome staxowax tdc tdc-secure thesecrete tor transistor watmap".split()
+from invirt import database
 
-def send_mail(vm):
-    contact = Machine.query.filter_by(name=vm).first().contact
+def send_mail(smtp, opts, message, vm):
+    contact = database.Machine.query.filter_by(name=vm).first().contact
     if '@' not in contact:
         contact += '@mit.edu'
     msg = MIMEText(message % vm)
     msg['To'] = contact
     msg['CC'] = 'XVM <xvm@mit.edu>'
     msg['Reply-To'] = 'XVM <xvm@mit.edu>'
-    msg['From'] = sender
-    msg['Subject'] = '[xvm] Unexpected reboot of your VM %s' % vm
-    s.sendmail(sender,
+    msg['From'] = opts.from_addr
+    msg['Subject'] = opts.subject % vm
+    smtp.sendmail(opts.from_addr,
         [contact, 'xvm@mit.edu'],
         msg.as_string())
 
-connect()
-s = smtplib.SMTP()
-s.connect()
+def main(argv):
+    parser = optparse.OptionParser(
+        usage = '%prog {-m message, -f from} [options] <vm-list',
+        description = __doc__.strip())
+    parser.add_option('-m', '--message',
+            type = 'string',
+            dest = 'message',
+            help = 'filename with body of message')
+    parser.add_option('-s', '--subject',
+            type = 'string',
+            dest = 'subject',
+            default = '[xvm] Unexpected reboot of your VM %s',
+            help = 'subject line of message; %s for VM name')
+    parser.add_option('-f', '--from',
+            type = 'string',
+            dest = 'from_addr',
+            help = 'for From: and envelope-from; first word is %(sig)s')
+    parser.add_option('--host',
+            type = 'string',
+            dest = 'host',
+            help = 'host that failed; %(host)s in message')
+    parser.add_option('--time',
+            type = 'string',
+            dest = 'time',
+            help = 'time of failure; %(time)s in message')
+    opts, args = parser.parse_args()
+
+    if len(args) or not opts.message or not opts.from_addr:
+        parser.print_help(sys.stderr)
+        return 2
+
+    opts.sig = opts.from_addr.split()[0]
+    message = file(opts.message).read() % opts.__dict__
+
+    vms = sys.stdin.read().split()
+
+    database.connect()
+    s = smtplib.SMTP()
+    s.connect()
+
+    for vm in vms:
+        send_mail(s, opts, message, vm)
 
-for vm in vms:
-    send_mail(vm)
+    s.close()
 
-s.close()
+if __name__ == '__main__':
+    sys.exit(main(sys.argv))