139 lines
4.6 KiB
Python
139 lines
4.6 KiB
Python
|
|
class HostGroupException(Exception):
|
|
def __init__(self, msg):
|
|
self.msg = msg
|
|
|
|
def __str__(self):
|
|
return repr(self.msg)
|
|
|
|
class HostGroupMissingArgumentException(HostGroupException):
|
|
def __init__(self, missingArgument, hostgroupName):
|
|
super(HostGroupMissingArgumentException, self).__init__("no " + missingArgument + " defined for Hostgroup " + hostgroupName)
|
|
|
|
class HostGroup:
|
|
def __init__(self, name, **kwargs):
|
|
self.name = name
|
|
tmp = "members"
|
|
self.members = []
|
|
if not tmp in kwargs:
|
|
raise HostGroupMissingArgumentException(tmp, name)
|
|
self.add_members(kwargs[tmp])
|
|
|
|
tmp = "hosts"
|
|
self.hosts = []
|
|
if not tmp in kwargs:
|
|
raise HostGroupMissingArgumentException(tmp, name)
|
|
self.add_hosts(kwargs[tmp])
|
|
|
|
tmp = "services"
|
|
if not tmp in kwargs:
|
|
raise HostGroupMissingArgumentException(tmp, name)
|
|
self.add_services(kwargs[tmp])
|
|
|
|
self.parents = []
|
|
tmp = "parents"
|
|
if tmp in kwargs:
|
|
self.add_parent(kwargs[tmp])
|
|
|
|
tmp = "processors"
|
|
self.processors = []
|
|
if tmp in kwargs:
|
|
self.add_processor(kwargs[tmp])
|
|
|
|
def _to_config_dict(self, configDict):
|
|
me = {}
|
|
me["members"] = [member.nameid for member in self.get_members()]
|
|
me["hosts"] = self.hosts
|
|
me["parents"] = [hg.get_name() for hg in self.get_parents()]
|
|
#TODO implement delegation
|
|
#me["services"] = [service._to_config_dict(configDict) for service in self.get_services()]
|
|
#me["processors"] = [processor._to_config_dict(configDict) for processor in self.get_processors()]
|
|
configDict["hostgroups"][self.get_name()] = me
|
|
|
|
|
|
|
|
def __add_internal(self,l,item):
|
|
if isinstance(item, list):
|
|
l.extend(item)
|
|
else:
|
|
l.append(item)
|
|
|
|
def add_members(self, member):
|
|
self.__add_internal(self.get_members(), member)
|
|
|
|
def add_hosts(self, host):
|
|
self.__add_internal(self.get_hosts(), host)
|
|
|
|
def add_processors(self, processor):
|
|
self.__add_internal(self.get_processors(), processor)
|
|
|
|
def add_parents(self, parent):
|
|
self.__add_internal(self.get_parents(), parent)
|
|
|
|
def get_parents(self):
|
|
return self.parents
|
|
|
|
def get_processors(self):
|
|
return self.processors
|
|
|
|
def get_services(self):
|
|
return self.services
|
|
|
|
def get_hosts(self):
|
|
return self.hosts
|
|
|
|
def get_name(self):
|
|
return self.name
|
|
|
|
def get_members(self):
|
|
return self.members
|
|
|
|
|
|
def __str__(self):
|
|
ret = "HostGroup: " + self.name + " threshold: " + str(self.threshold) + " parent: " + self.parent + "\n"
|
|
ret += "members: {\n"
|
|
for itm in self.members:
|
|
ret += str(itm) + "\n"
|
|
ret += "}\n"
|
|
ret += "hosts: {\n"
|
|
for itm in self.hosts:
|
|
ret += str(itm) + "\n"
|
|
ret += "}\n"
|
|
ret += "services: {\n"
|
|
for itm in self.services:
|
|
ret += str(itm) + "\n"
|
|
ret += "}\n"
|
|
return ret
|
|
|
|
|
|
class HostGroupService:
|
|
def __init__(self, services, periods):
|
|
self.services = services
|
|
self.periods = periods
|
|
|
|
def __str__(self):
|
|
return "HostgroupService { " + str([str(s) for s in self.services]) + ", " + str([p.name for p in self.periods]) + "}"
|
|
|
|
|
|
def parseHostGroupList(hostgroups, hosts, members, periods, services, log):
|
|
parsedHostGroups = []
|
|
for hgname, hgValues in hostgroups.items():
|
|
hostGroup = HostGroup(hgname)
|
|
hostGroup.members = [m for m in members if m.nameid in hgValues['members']]
|
|
hostGroup.hosts = [h for h in hosts if h.name in hgValues['hosts']]
|
|
hostGroup.threshold = hgValues['threshold']
|
|
if 'parent' in hgValues:
|
|
hostGroup.parent = hgValues['parent']
|
|
hostGroup.services = []
|
|
for serviceName, servicePeriods in hgValues['services'].items():
|
|
services = []
|
|
for host in hosts:
|
|
service = host.getHostServiceByName(serviceName)
|
|
if service is not None:
|
|
services.append(service)
|
|
else:
|
|
log.w("could not find HostService(" + str(serviceName) + ") for host " + host.name)
|
|
hostGroupPeriods = [p for p in periods if p.name in servicePeriods]
|
|
hostGroup.services.append(HostGroupService(services, hostGroupPeriods))
|
|
parsedHostGroups.append(hostGroup)
|
|
return parsedHostGroups
|