###########
# Use this section to configure this program for your environment.
###########

# The Url variable specifies the URL of your Iguana server.
# Replace "localhost" with the location of your server.
# If your server is running on any port other than 6543, replace 6543 with your
# server's port number.
# Replace "admin" and "password" with the user name and password you use to log
# on to your Iguana server as an administrator.
Url = 'http://localhost:6543/status.html?Format=text&UserName=admin&Password=password'

# Specify the address from which email is to be sent, the recipient of the message,
# and the subject header for the email message.
MailFrom = 'pymon@linuxox'
MailTo = 'john.doe@yourcompanynamehere.com'
MailSubject = '[pymon] testing'

# Specify SMTP configuration information here.
SmtpHost = 'mailserver2'
SmtpPort = 25
SmtpEhlo = 'pymon'

# The name of the file that will contain the channel monitoring information.
# If this file does not exist, this script creates it for you.
DataFileName = 'pymon.dat'

# The names of the channels to be monitored.
# If this list is empty, all channels will be monitored.
ChannelNamesToMonitor = [ 'FileFile' ]

###########
# End of configuration information: do not change anything below this point.
###########

import urllib
import smtplib
import sys


LogLines = []

# read the channel status summary data into an array of lines

print "accessing", Url
Stream = urllib.urlopen(Url)
Lines = Stream.readlines()
print "data received"
Lines = map( lambda x: x.strip(), Lines)
ChannelLines = Lines[2:]

# ^these are per-channel status lines

# load the file with previously seen error counts
# if the file doesn't exist, it will be created the first time without triggering errors
OldCounts = {}
try:
   DataLines = map( lambda x: x.strip(), file(DataFileName,'rb').readlines() )
   for DataLine in DataLines:
      (ChannelName, SourceErrorCount, DestErrorCount) = DataLine.split(',')
      OldCounts[ChannelName] = (SourceErrorCount, DestErrorCount)
except:
   OldCounts = {}

def removeQuotes(Value):
   """ helper function to remove double quotes around strings from per-channel status lines """
   if len(Value)>=2 and Value[0]=='"' and Value[-1]=='"':
      return Value[1:-1]
   return Value

NewCounts = {}
# the counts seen during the current run,
# used to overwrite the data file

HaveErrors = False
# ^ whether we've seen (a change in) error counts, and thus should send the email

for ChannelLine in ChannelLines:
   ChannelParts = ChannelLine.split(',')
   ChannelParts = map( lambda x: x.strip(), ChannelParts)
   ChannelParts = map( removeQuotes, ChannelParts)
   if len(ChannelParts) < 9:
      print "WARNING!!! ", len(ChannelParts), " < 9 so ignoring this line"
      continue
   # print ChannelParts
   (ChannelName, ChannelStatus, SourceType, SourceStatus, DestType, DestStatus,
    SuccessCount, SourceErrorCount, DestErrorCount) = ChannelParts[:9]
   # print "success: %s source err: %s dest err: %s" % (SuccessCount, SourceErrorCount, DestErrorCount)

   if len(ChannelNamesToMonitor) and (ChannelName not in ChannelNamesToMonitor):
      continue

   if ChannelName in OldCounts:
      (OldSourceErrorCount, OldDestErrorCount) = OldCounts[ChannelName]
      if SourceErrorCount != OldSourceErrorCount:
         LogLines.append( "channel(%s).source(%s) has %s errors!" % (ChannelName, SourceType, SourceErrorCount) )
         HaveErrors = True
      if DestErrorCount != OldDestErrorCount:
         LogLines.append( "channel(%s).destination(%s) has %s errors!" % (ChannelName, DestType, DestErrorCount) )
         HaveErrors = True

   # always collect currently seen counts, to update the data file
   NewCounts[ChannelName]=(SourceErrorCount,DestErrorCount)

# update (overwrite) the data file
# this is done in any case - otherwise we'd have to detect changes...
DataFile = file(DataFileName,'wb')
for ChannelName in NewCounts:
   DataFile.write( "%s,%s,%s\n" % (ChannelName, NewCounts[ChannelName][0], NewCounts[ChannelName][1]) )
DataFile.close()
# done updating the data file

if not HaveErrors:
   print "no errors"
   sys.exit(0)

# generate an email

LogLines.insert(0, "have errors!")
print '\n'.join(LogLines)

MailDate = smtplib.rfc822.formatdate()

Headers = [
   'Content-Type: text/plain',
   'From: <'+MailFrom+'>',
   'To: <'+MailTo+'>',
   'Subject: '+MailSubject,
   'Date: '+MailDate,
   '',
]

MailMessage = '\r\n'.join(Headers + LogLines) + '\r\n'
print
print MailMessage

# send the generated message over SMTP

# SMTP error codes being checked below
# may potentially come up as different successful values - fix as needed if an assertion fails

Smtp = smtplib.SMTP(SmtpHost, SmtpPort)
EhloRes = Smtp.ehlo(SmtpEhlo)
print "EHLO:", EhloRes
assert EhloRes[0] == 250

MailRes = Smtp.mail(MailFrom)
print 'MAIL', MailFrom, ':', MailRes
assert MailRes[0] == 250

RcptRes = Smtp.rcpt(MailTo)
print 'RCPT', MailTo, ':', RcptRes
assert RcptRes[0] == 250

DataRes = Smtp.data(MailMessage)
print 'DATA: ', DataRes
assert DataRes[0] == 250

print "email sent"