Changeset 2252 for trunk/scripts/outage/outage-mail

Timestamp:

Mar 16, 2009, 8:26:21 PM (16 years ago)

Author:

price

Message:

make outage-mail fully reusable

File:

• trunk/scripts/outage/outage-mail (modified) (2 diffs)

trunk/scripts/outage/outage-mail

@@ -1 +1 @@
 #!/usr/bin/python
+"""
+
+VM names on stdin.
+"""
+
 import smtplib
-from invirt.database import *
+import optparse
+import sys
 from email.mime.text import MIMEText
 
-sender = 'Greg Price <price@mit.edu>'
-sendername = sender.split()[0]
-host = 'arklay-mansion'
+from invirt import database
 
-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.
-
-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)
-
-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()
-
-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'
@@ -31 +20 @@
     msg['CC'] = 'XVM <xvm@mit.edu>'
     msg['Reply-To'] = 'XVM <xvm@mit.edu>'
-    msg['From'] = sender
+    msg['From'] = opts.from_addr
     msg['Subject'] = '[xvm] Unexpected reboot of your VM %s' % vm
-    s.sendmail(sender,
+    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('-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()
 
-for vm in vms:
-    send_mail(vm)
+    if len(args) or not opts.message or not opts.from_addr:
+        parser.print_help(sys.stderr)
+        return 2
 
-s.close()
+    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)
+
+    s.close()
+
+if __name__ == '__main__':
+    sys.exit(main(sys.argv))