def remove_hostgroup(self, group):
if isinstance(group, HostGroup):
self.__groups.remove(group)
elif isinstance(group, str):
group = ConfigBuilder.get_hostgroup(group)
self.__groups.remove(group)
return self
def create(id: str, force_create: bool = False) -> T:
ValueChecker.validate_id(id)
hostgroup = None if force_create else ConfigBuilder.get_hostgroup(id)
if None is hostgroup:
hostgroup = HostGroup(id)
ConfigBuilder.add_hostgroup(id, hostgroup)
return hostgroup
def add_hostgroup(self, group):
if isinstance(group, HostGroup):
if group not in self.__groups:
self.__groups.append(group)
elif isinstance(group, str):
group = ConfigBuilder.get_hostgroup(group)
if None is group:
raise Exception('HostGroup does not exist yet!')
self.add_hostgroup(group)
else:
raise Exception('Can only add HostGroup or id of HostGroup!')
return self
