Source code for cobbler.modules.installation.post_puppet

This module signs newly installed client puppet certificates if the
puppet master server is running on the same machine as the Cobbler

Based on:
import logging
import re
from typing import TYPE_CHECKING, List

from cobbler import utils

    from cobbler.api import CobblerAPI

logger = logging.getLogger()

[docs]def register() -> str: """ The mandatory Cobbler module registration hook. """ # this pure python trigger acts as if it were a legacy shell-trigger, but is much faster. # the return of this method indicates the trigger type return "/var/lib/cobbler/triggers/install/post/*"
[docs]def run(api: "CobblerAPI", args: List[str]) -> int: """ The obligatory Cobbler modules hook. :param api: The api to resolve all information with. :param args: This is an array with two items. The first must be ``system``, if the value is different we do an early and the second is the name of this system or profile. :return: ``0`` or nothing. """ objtype = args[0] name = args[1] if objtype != "system": return 0 settings = api.settings() if not settings.puppet_auto_setup: return 0 if not settings.sign_puppet_certs_automatically: return 0 system = api.find_system(name) if system is None or isinstance(system, list): raise ValueError("Ambigous search match!") blendered_system = utils.blender(api, False, system) hostname = blendered_system["hostname"] if not re.match(r"[\w-]+\..+", hostname): search_domains = blendered_system["name_servers_search"] if search_domains: hostname += "." + search_domains[0] puppetca_path = settings.puppetca_path cmd = [puppetca_path, "cert", "sign", hostname] return_code = 0 try: return_code = utils.subprocess_call(cmd, shell=False) except Exception: logger.warning("failed to execute %s", puppetca_path) if return_code != 0: logger.warning("signing of puppet cert for %s failed", name) return 0