Gecko [ goons4good.org ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2022-10-11 05:09
ssh	[ DIR ]	drwxr-xr-x	2022-10-11 05:09
__init__.py	78.19 KB	-rw-r--r--	2022-05-16 09:16
mixins.py	20.7 KB	-rw-r--r--	2022-05-16 09:16
netapi.py	2.23 KB	-rw-r--r--	2022-05-16 09:16

Rename

"""
The main entry point for salt-api
"""

import logging
import signal

import salt.loader
import salt.utils.process

log = logging.getLogger(__name__)

class RunNetapi(salt.utils.process.SignalHandlingProcess):
    """
    Runner class that's pickable for netapi modules
    """

def __init__(self, opts, fname, **kwargs):
        super().__init__(**kwargs)
        self.opts = opts
        self.fname = fname

def run(self):
        netapi = salt.loader.netapi(self.opts)
        netapi_func = netapi[self.fname]
        netapi_func()

class NetapiClient:
    """
    Start each netapi module that is configured to run
    """

def __init__(self, opts):
        self.opts = opts
        self.process_manager = salt.utils.process.ProcessManager(
            name="NetAPIProcessManager"
        )
        self.netapi = salt.loader.netapi(self.opts)

def run(self):
        """
        Load and start all available api modules
        """
        if not self.netapi:
            log.error("Did not find any netapi configurations, nothing to start")

kwargs = {}
        if salt.utils.platform.is_windows():
            kwargs["log_queue"] = salt.log.setup.get_multiprocessing_logging_queue()
            kwargs[
                "log_queue_level"
            ] = salt.log.setup.get_multiprocessing_logging_level()

for fun in self.netapi:
            if fun.endswith(".start"):
                log.info("Starting %s netapi module", fun)
                self.process_manager.add_process(
                    RunNetapi, args=(self.opts, fun), kwargs=kwargs, name="RunNetapi"
                )

# Install the SIGINT/SIGTERM handlers if not done so far
        if signal.getsignal(signal.SIGINT) is signal.SIG_DFL:
            # No custom signal handling was added, install our own
            signal.signal(signal.SIGINT, self._handle_signals)

if signal.getsignal(signal.SIGTERM) is signal.SIG_DFL:
            # No custom signal handling was added, install our own
            signal.signal(signal.SIGTERM, self._handle_signals)

self.process_manager.run()

def _handle_signals(self, signum, sigframe):
        # escalate the signals to the process manager
        self.process_manager._handle_signals(signum, sigframe)