+def reformat_for_rt(text):
+ lines = text.split("\n")
+ spaced_text = ""
+ for line in lines:
+ spaced_text += " %s\n" %line
+ return spaced_text
+
+
+def _setupRTenvironment():
+ os.environ['PATH'] = os.environ['PATH'] + ":" + monitorconfig.RT_WEB_TOOLS_PATH
+ os.environ['RTSERVER'] = monitorconfig.RT_WEB_SERVER
+ os.environ['RTUSER'] = monitorconfig.RT_WEB_USER
+ os.environ['RTPASSWD'] = monitorconfig.RT_WEB_PASSWORD
+ os.environ['RTDEBUG'] = monitorconfig.RT_WEB_DEBUG
+ return
+
+def setTicketStatus(ticket_id, status):
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ return {}
+
+ cmd = "rt edit ticket/%s set status=%s" % (ticket_id, status)
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ value = f_out.read()
+ l_values = value.split('\n')
+ return "".join(l_values).strip()
+
+def getTicketStatus(ticket_id):
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ return {}
+
+ cmd = "rt show -t ticket -f id,subject,status,queue,created %s" % (ticket_id)
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ value = f_out.read()
+ l_values = value.split('\n')
+ r_values = {}
+ for line in l_values:
+ if len(line) == 0: continue
+ vals = line.split(':')
+ key = vals[0]
+ r_values[key] = ":".join(vals[1:])
+ r_values[key] = r_values[key].strip()
+
+ r_values['Created'] = calendar.timegm(time.strptime(r_values['Created']))
+ return r_values
+
+def setAdminCCViaRT(ticket_id, to):
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ raise Exception("ERROR: ticket_id must be set to some integer value")
+
+ # This will raise an exception if it is not a valid id.
+ i_ticket_id = int(ticket_id)
+
+ # create a comma-separated list
+ s_to = ",".join(to)
+ cmd = "rt edit ticket/%s set admincc='%s'" % (ticket_id, s_to)
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ value = f_out.read()
+ l_values = value.split()
+ f_in.close() ; f_out.close() ; f_err.close()
+ if len(l_values) > 3 and "updated" in l_values[3]:
+ # Success
+ pass
+ else:
+ print "VALUE:", value
+ print "ERROR: RT failed to update AdminCC for ticket %s" % ticket_id
+
+ return
+
+def setSubjectViaRT(ticket_id, subject):
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ raise Exception("ERROR: ticket_id must be set to some integer value")
+
+ # This will raise an exception if it is not a valid id.
+ i_ticket_id = int(ticket_id)
+
+ cmd = "rt edit ticket/%s set subject='%s'" % (ticket_id, subject)
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ value = f_out.read()
+ l_values = value.split()
+ f_in.close() ; f_out.close() ; f_err.close()
+ if len(l_values) > 3 and "updated" in l_values[3]:
+ # Success
+ pass
+ else:
+ print "VALUE:", value
+ print "ERROR: RT failed to update subject for ticket %s" % ticket_id
+
+ return
+
+
+def addCommentViaRT(ticket_id, comment):
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ raise Exception("ERROR: ticket_id must be set to some integer value")
+
+ # This will raise an exception if it is not a valid id.
+ i_ticket_id = int(ticket_id)
+
+ cmd = "rt comment -m '%s' ticket/%s" % (comment, i_ticket_id)
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ value = f_out.read()
+ l_values = value.split()
+ f_in.close() ; f_out.close() ; f_err.close()
+ if len(l_values) > 1 and "recorded" in l_values[1]:
+ # Success
+ pass
+ else:
+ # Error
+ f_in.close() ; f_out.close() ; f_err.close()
+ print "ERROR: RT failed to add comment to id %s" % ticket_id
+
+ return
+
+def closeTicketViaRT(ticket_id, comment):
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+ if ticket_id == None or ticket_id == "":
+ raise Exception("ERROR: ticket_id must be set to some integer value")
+
+ # This will raise an exception if it is not a valid id.
+ i_ticket_id = int(ticket_id)
+
+ # Append comment to RT ticket
+ addCommentViaRT(ticket_id, comment)
+
+ if not config.debug:
+ cmd = "rt edit ticket/%s set status=resolved" % i_ticket_id
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ f_in.close()
+ value = f_out.read()
+ f_out.close()
+ f_err.close()
+ l_values = value.split()
+ if len(l_values) >= 4 and "updated" in l_values[3]:
+ # Success!!
+ pass
+ else:
+ print "VALUE: ", value
+ # Failed!!
+ print "FAILED to resolve Ticket %s" % ticket_id
+ print "FAILED to resolve Ticket %s" % i_ticket_id
+
+ return
+
+def emailViaRT(subject, text, to, ticket_id=None):
+ if ticket_id == None or ticket_id == "":
+ print "No TICKET"
+ return emailViaRT_NoTicket(subject, text, to)
+
+
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+
+ if config.mail and not config.debug:
+ setSubjectViaRT(ticket_id, subject)
+ setAdminCCViaRT(ticket_id, to)
+
+ cmd = "rt correspond -m - %s" % ticket_id
+ (f_in, f_out, f_err) = os.popen3(cmd)
+ f_in.write(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()
+
+ del f_in
+ f_out.close(); del f_out
+ f_err.close(); del f_err
+ os.wait()
+
+ return ticket_id
+
+
+def emailViaRT_NoTicket(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]
+
+ # Set ENV Variables/PATH
+ _setupRTenvironment()
+
+ # 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"
+
+ # Add a space for each new line to get RT to accept the file.
+ spaced_text = reformat_for_rt(text)
+
+ 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()
+
+ print "MAILER: ticket value == %s" % value.split()[2]
+ 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
+
+ return i_ticket
+