class HsMonitor(object):
def __init__(self):
# setup the log mode
file = 'log-testfornagiospushfromhisparc'
handler = logging.handlers.TimedRotatingFileHandler(
file, when='midnight', backupCount=14)
handler.setFormatter(formatter)
logger.addHandler(handler)
logger.setLevel(level=logging.DEBUG)
# read the configuration file
try:
self.cfg = EConfigParser()
self.cfg.read([CONFIG_INI_PATH1, CONFIG_INI_PATH2])
except:
logger.error("Cannot open the config file!", severity=2)
return
else:
logger.debug("Initilize variables")
# list of all the threads
self.hsThreads = []
# Assume one server (eventwarehouse)
# if the local is also specified it will be added
self.numServers = 1
def createBufferListener(self, interpreter):
# get the information from configuration file
bufferdb = {}
bufferdb['host'] = self.cfg.get('BufferDB', 'Host')
bufferdb['db'] = self.cfg.get('BufferDB', 'DB')
bufferdb['user'] = self.cfg.get('BufferDB', 'Username')
bufferdb['password'] = self.cfg.get('BufferDB', 'Password')
bufferdb['poll_interval'] = self.cfg.get('BufferDB', 'Poll_Interval')
bufferdb['poll_limit'] = self.cfg.get('BufferDB', 'Poll_Limit')
bufferdb['keep_buffer_data'] = self.cfg.get('BufferDB',
'KeepBufferData')
# create an instance of BufferListener class
buffLis = BufferListener.BufferListener(bufferdb, interpreter)
if not buffLis:
logger.error("Cannot connect to the buffer database!")
return None
# TODO better error handling
return buffLis
def createCheckScheduler(self, interpreter):
# get the nagios configuration section from config file
nagiosConf = self.cfg.itemsdict('NagiosPush')
machine = re.search('([a-z0-9]+).zip',
self.cfg.get('Station', 'Certificate'))
if machine is None:
nagiosConf['machine_name'] = 'test32bit'
else:
nagiosConf['machine_name'] = machine.group(1)
checkSched = CheckScheduler(nagiosConf, interpreter)
return checkSched
def createUploader(self, serverID, section_name, numServers):
# TODO create default values if parameter doesn't exist
stationID = self.cfg.get("Station", "StationID")
url = self.cfg.get(section_name, "URL")
passw = self.cfg.get(section_name, "Password")
minbs = self.cfg.ifgetint(section_name, "MinBatchSize", 50)
maxbs = self.cfg.ifgetint(section_name, "MaxBatchSize", 50)
if (minbs > maxbs):
raise Exception("Minimum batch size must be less than maximum")
minwait = self.cfg.ifgetfloat(section_name, "MinWait", 1.0)
maxwait = self.cfg.ifgetfloat(section_name, "MaxWait", 60.0)
up = Uploader(serverID, numServers, stationID, passw, url,
minwait, maxwait, minbs, maxbs)
return up
