Gecko [ goons4good.org ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2022-10-11 05:09
__init__.py	18.16 KB	-rw-r--r--	2022-05-16 09:16
adb.py	5.2 KB	-rw-r--r--	2022-05-16 09:16
aix_account.py	1.33 KB	-rw-r--r--	2022-05-16 09:16
avahi_announce.py	8.53 KB	-rw-r--r--	2022-05-16 09:16
bonjour_announce.py	8.21 KB	-rw-r--r--	2022-05-16 09:16
btmp.py	8.33 KB	-rw-r--r--	2022-05-16 09:16
cert_info.py	5.82 KB	-rw-r--r--	2022-05-16 09:16
diskusage.py	3.35 KB	-rw-r--r--	2022-05-16 09:16
glxinfo.py	1.84 KB	-rw-r--r--	2022-05-16 09:16
haproxy.py	3.02 KB	-rw-r--r--	2022-05-16 09:16
inotify.py	11.88 KB	-rw-r--r--	2022-05-16 09:16
journald.py	2.61 KB	-rw-r--r--	2022-05-16 09:16
junos_rre_keys.py	723 B	-rw-r--r--	2022-05-16 09:16
load.py	5.95 KB	-rw-r--r--	2022-05-16 09:16
log_beacon.py	3.67 KB	-rw-r--r--	2022-05-16 09:16
memusage.py	1.57 KB	-rw-r--r--	2022-05-16 09:16
napalm_beacon.py	11.61 KB	-rw-r--r--	2022-05-16 09:16
network_info.py	4.66 KB	-rw-r--r--	2022-05-16 09:16
network_settings.py	6.53 KB	-rw-r--r--	2022-05-16 09:16
pkg.py	2.57 KB	-rw-r--r--	2022-05-16 09:16
proxy_example.py	1.48 KB	-rw-r--r--	2022-05-16 09:16
ps.py	2.23 KB	-rw-r--r--	2022-05-16 09:16
salt_monitor.py	4.06 KB	-rw-r--r--	2022-05-16 09:16
salt_proxy.py	1.81 KB	-rw-r--r--	2022-05-16 09:16
sensehat.py	2.82 KB	-rw-r--r--	2022-05-16 09:16
service.py	6.15 KB	-rw-r--r--	2022-05-16 09:16
sh.py	3.1 KB	-rw-r--r--	2022-05-16 09:16
smartos_imgadm.py	2.6 KB	-rw-r--r--	2022-05-16 09:16
smartos_vmadm.py	3.34 KB	-rw-r--r--	2022-05-16 09:16
status.py	4.11 KB	-rw-r--r--	2022-05-16 09:16
swapusage.py	1.57 KB	-rw-r--r--	2022-05-16 09:16
telegram_bot_msg.py	2.45 KB	-rw-r--r--	2022-05-16 09:16
twilio_txt_msg.py	2.65 KB	-rw-r--r--	2022-05-16 09:16
watchdog.py	4.79 KB	-rw-r--r--	2022-05-16 09:16
wtmp.py	10.17 KB	-rw-r--r--	2022-05-16 09:16

Rename

"""
Send events covering process status
"""
import logging

import salt.utils.beacons

try:
    import salt.utils.psutil_compat as psutil

HAS_PSUTIL = True
except ImportError:
    HAS_PSUTIL = False

log = logging.getLogger(__name__)  # pylint: disable=invalid-name

__virtualname__ = "ps"

def __virtual__():
    if not HAS_PSUTIL:
        return (False, "cannot load ps beacon: psutil not available")
    return __virtualname__

def validate(config):
    """
    Validate the beacon configuration
    """
    # Configuration for ps beacon should be a list of dicts
    if not isinstance(config, list):
        return False, "Configuration for ps beacon must be a list."
    else:
        config = salt.utils.beacons.list_to_dict(config)

if "processes" not in config:
            return False, "Configuration for ps beacon requires processes."
        else:
            if not isinstance(config["processes"], dict):
                return False, "Processes for ps beacon must be a dictionary."

return True, "Valid beacon configuration"

def beacon(config):
    """
    Scan for processes and fire events

Example Config

.. code-block:: yaml

beacons:
          ps:
            - processes:
                salt-master: running
                mysql: stopped

The config above sets up beacons to check that
    processes are running or stopped.
    """
    ret = []
    procs = []
    for proc in psutil.process_iter():
        try:
            _name = proc.name()
        except psutil.NoSuchProcess:
            # The process is now gone
            continue
        if _name not in procs:
            procs.append(_name)

config = salt.utils.beacons.list_to_dict(config)

for process in config.get("processes", {}):
        ret_dict = {}
        if config["processes"][process] == "running":
            if process in procs:
                ret_dict[process] = "Running"
                ret.append(ret_dict)
        elif config["processes"][process] == "stopped":
            if process not in procs:
                ret_dict[process] = "Stopped"
                ret.append(ret_dict)
        else:
            if process not in procs:
                ret_dict[process] = False
                ret.append(ret_dict)
    return ret