#
# Faiyaz Ahmed <faiyaza@cs.princeton.edu>
#
-# $Id: $
+# $Id: mailer.py,v 1.8 2007/06/29 12:42:22 soltesz Exp $
from emailTxt import *
-import xml, xmlrpclib
import smtplib
+from config import config
+import logging
+import os
+import time
+
+config = config()
+logger = logging.getLogger("monitor")
MTA="localhost"
-FROM="support@planet-lab.org"
+FROM="monitor@planet-lab.org"
+
+def reformat_for_rt(text):
+ lines = text.split("\n")
+ spaced_text = ""
+ for line in lines:
+ spaced_text += " %s\n" %line
+ return spaced_text
+
+
+def emailViaRT(subject, text, to):
+ """Use RT command line tools to send email.
+ return the generated RT ticket ID number.
+ """
+ i_ticket = 0
+
+ if config.mail and config.debug:
+ to = [config.email]
-XMLRPC_SERVER = 'https://www.planet-lab.org/PLCAPI/'
+ os.environ['PATH'] = os.environ['PATH'] + ":/home/soltesz/rpm/opt/rt3/bin/"
+ os.environ['RTSERVER'] = "https://rt.planet-lab.org/"
+ os.environ['RTUSER'] = "monitor"
+ os.environ['RTPASSWD'] = "ssorcmor"
+ os.environ['RTDEBUG'] = "0"
+ # NOTE: AdminCc: (in PLC's RT configuration) gets an email sent.
+ # This is not the case (surprisingly) for Cc:
+ input_text = "Subject: %s\n"
+ input_text += "Requestor: monitor@planet-lab.org\n"
+ input_text += "id: ticket/new\n"
+ input_text += "Queue: Monitor\n"
+ for recipient in to:
+ input_text += "AdminCc: %s\n" % recipient
+ input_text += "Text: %s"
-def siteId(hostname):
- api = xmlrpclib.Server(XMLRPC_SERVER, verbose=False)
- anon = {'AuthMethod': "anonymous"}
- site_id = api.AnonAdmQuerySite (anon, {"node_hostname": hostname})
- if len(site_id) == 1:
- loginbase = api.AnonAdmGetSites (anon, site_id, ["login_base"])
- return loginbase[0]['login_base']
+ spaced_text = reformat_for_rt(text)
+ #if config.debug:
+ # print input_text % (subject, "")
+
+ if config.mail and not config.debug:
+ cmd = "rt create -i -t ticket"
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ f_in.write(input_text % (subject, spaced_text))
+ f_in.flush()
+ f_in.close()
+ value = f_out.read()
+
+ # TODO: rt doesn't write to stderr on error!!!
+ if value == "":
+ raise Exception, f_err.read()
+
+ i_ticket = int(value.split()[2])
+ # clean up the child process.
+ f_in.close(); del f_in
+ f_out.close(); del f_out
+ f_err.close(); del f_err
+ os.wait()
+ elif config.mail and config.debug:
+ email(subject, spaced_text, to)
+ i_ticket = 0
+ else:
+ i_ticket = 0
-def email (subject, text, to, cc = None):
+ return i_ticket
+
+def email(subject, text, to):
"""Create a mime-message that will render HTML in popular
MUAs, text in better ones"""
import MimeWriter
import mimetools
import cStringIO
+ if config.mail and config.debug:
+ to = [config.email]
+
out = cStringIO.StringIO() # output buffer for our message
txtin = cStringIO.StringIO(text)
# message body
#
writer.addheader("Subject", subject)
- writer.addheader("To", to)
- if cc: writer.addheader("CC", cc)
+ if to.__class__ == [].__class__ :
+ writer.addheader("To", to[0])
+ cc = ""
+ for dest in to[1:len(to)]:
+ cc +="%s, " % dest
+ cc = cc.rstrip(", ")
+ writer.addheader("Cc", cc)
+ else:
+ writer.addheader("To", to)
+
+ if config.bcc and not config.debug:
+ writer.addheader("Bcc", config.email)
+
+ writer.addheader("Reply-To", 'monitor@planet-lab.org')
+
writer.addheader("MIME-Version", "1.0")
#
# start the multipart section of the message
msg = out.getvalue()
out.close()
- #server = smtplib.SMTP(MTA)
- #server.sendmail(FROM, (to,cc), msg)
- #server.quit()
+ # three cases:
+ # mail but no-debug
+ # mail and debug, 'to' changed at the beginning'
+ # nomail, but report who I'd send to.
+ if config.mail:
+ try:
+ # This is normal operation
+ server = smtplib.SMTP(MTA)
+ server.sendmail(FROM, to, msg)
+ if config.bcc and not config.debug:
+ server.sendmail(FROM, config.email, msg)
+ server.quit()
+ except Exception, err:
+ print "Mailer error: %s" % err
+ else:
+ #print "Would mail %s" %to
+ logger.debug("Would send mail to %s" % to)
if __name__=="__main__":
import smtplib
import emailTxt
- id = siteId("alice.cs.princeeton.edu")
- if id:
- email('TEST', emailTxt.mailtxt.STANDARD % {'hostname': "ALICE.cs.princeton.edu"}, "tech-" + id + "@sites.planet-lab.org")
- else:
- print "No dice."
+ import plc
+ #email("[spam] bcc test from golf.cs.princeton.edu",
+ # "It gets to both recipients",
+ # "soltesz@cs.utk.edu")
+ #emailViaRT("rt via golf",
+ # "It gets to both recipients",
+ # "soltesz@cs.utk.edu")
+ id = emailViaRT("TEST 7",
+ mailtxt.newbootcd_one[1] % {'hostname_list':"hostname list..."},
+ ["soltesz@cs.utk.edu", "soltesz@romcross.org", "soltesz@cs.princeton.edu"])
+ print "ticketid: %d" % id
+ #id = plc.siteId(["alice.cs.princeton.edu"])
+ #print id
+ #if id:
+ #email('TEST', emailTxt.mailtxt.ssh % {'hostname': "ALICE.cs.princeton.edu"}, "tech-" + id + "@sites.planet-lab.org")
+ #else:
+ # print "No dice."
+ #email("TEST111", "I'd like to see if this works anywhere", ["soltesz@cs.princeton.edu", "soltesz@cs.utk.edu"])
+ #print "mailer does nothing in main()"