Source code for cobbler.modules.managers.genders

"""
Cobbler Module that manages the cluster configuration tool from CHAOS. For more information please see:
`GitHub - chaos/genders <https://github.com/chaos/genders>`_
"""

import distutils.sysconfig
import logging
import sys
import os
import time
from cobbler.templar import Templar

plib = distutils.sysconfig.get_python_lib()
mod_path = "%s/cobbler" % plib
sys.path.insert(0, mod_path)
template_file = "/etc/cobbler/genders.template"
settings_file = "/etc/genders"

logger = logging.getLogger()


[docs]def register() -> str: """ We should run anytime something inside of Cobbler changes. :return: Always ``/var/lib/cobbler/triggers/change/*`` """ return "/var/lib/cobbler/triggers/change/*"
[docs]def write_genders_file(config, profiles_genders, distros_genders, mgmtcls_genders): """ Genders file is over-written when ``manage_genders`` is set in our settings. :param config: The API instance to template the data with. :param profiles_genders: The profiles which should be included. :param distros_genders: The distros which should be included. :param mgmtcls_genders: The management classes which should be included. :raises OSError: Raised in case the template could not be read. """ try: f2 = open(template_file, "r") except: raise OSError("error reading template: %s" % template_file) template_data = "" template_data = f2.read() f2.close() metadata = { "date": time.asctime(time.gmtime()), "profiles_genders": profiles_genders, "distros_genders": distros_genders, "mgmtcls_genders": mgmtcls_genders, } templar_inst = Templar(config) templar_inst.render(template_data, metadata, settings_file)
[docs]def run(api, args) -> int: """ Mandatory Cobbler trigger hook. :param api: The api to resolve information with. :param args: For this implementation unused. :return: ``0`` or ``1``, depending on the outcome of the operation. """ # do not run if we are not enabled. if not api.settings().manage_genders: return 0 profiles_genders = dict() distros_genders = dict() mgmtcls_genders = dict() # let's populate our dicts # TODO: the lists that are created here are strictly comma separated. # /etc/genders allows for host lists that are in the notation similar to: node00[01-07,08,09,70-71] at some point, # need to come up with code to generate these types of lists. # profiles for prof in api.profiles(): # create the key profiles_genders[prof.name] = "" for system in api.find_system(profile=prof.name, return_list=True): profiles_genders[prof.name] += system.name + "," # remove a trailing comma profiles_genders[prof.name] = profiles_genders[prof.name][:-1] if profiles_genders[prof.name] == "": profiles_genders.pop(prof.name, None) # distros for dist in api.distros(): # create the key distros_genders[dist.name] = "" for system in api.find_system(distro=dist.name, return_list=True): distros_genders[dist.name] += system.name + "," # remove a trailing comma distros_genders[dist.name] = distros_genders[dist.name][:-1] if distros_genders[dist.name] == "": distros_genders.pop(dist.name, None) # mgmtclasses for mgmtcls in api.mgmtclasses(): # create the key mgmtcls_genders[mgmtcls.name] = "" for system in api.find_system(mgmt_classes=mgmtcls.name, return_list=True): mgmtcls_genders[mgmtcls.name] += system.name + "," # remove a trailing comma mgmtcls_genders[mgmtcls.name] = mgmtcls_genders[mgmtcls.name][:-1] if mgmtcls_genders[mgmtcls.name] == "": mgmtcls_genders.pop(mgmtcls.name, None) # The file doesn't exist and for some reason the template engine won't create it, so spit out an error and tell the # user what to do. if not os.path.isfile(settings_file): logger.info("Error: " + settings_file + " does not exist.") logger.info("Please run: touch " + settings_file + " as root and try again.") return 1 write_genders_file(api, profiles_genders, distros_genders, mgmtcls_genders) return 0