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

"""
Beacon that fires events on vm state changes

.. code-block:: yaml

## minimal
    # - check for vm changes every 1 second (salt default)
    # - does not send events at startup
    beacons:
      vmadm: []

## standard
    # - check for vm changes every 60 seconds
    # - send create event at startup for all vms
    beacons:
      vmadm:
        - interval: 60
        - startup_create_event: True
"""
import logging

import salt.utils.beacons

__virtualname__ = "vmadm"

VMADM_STATE = {
    "first_run": True,
    "vms": [],
}

log = logging.getLogger(__name__)

def __virtual__():
    """
    Provides vmadm beacon on SmartOS
    """
    if "vmadm.list" in __salt__:
        return True
    else:
        return (
            False,
            "{} beacon can only be loaded on SmartOS compute nodes".format(
                __virtualname__
            ),
        )

def validate(config):
    """
    Validate the beacon configuration
    """
    vcfg_ret = True
    vcfg_msg = "Valid beacon configuration"

if not isinstance(config, list):
        vcfg_ret = False
        vcfg_msg = "Configuration for vmadm beacon must be a list!"

return vcfg_ret, vcfg_msg

def beacon(config):
    """
    Poll vmadm for changes
    """
    ret = []

# NOTE: lookup current images
    current_vms = __salt__["vmadm.list"](
        keyed=True,
        order="uuid,state,alias,hostname,dns_domain",
    )

# NOTE: apply configuration
    if VMADM_STATE["first_run"]:
        log.info("Applying configuration for vmadm beacon")

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

if "startup_create_event" not in config or not config["startup_create_event"]:
            VMADM_STATE["vms"] = current_vms

# NOTE: create events
    for uuid in current_vms:
        event = {}
        if uuid not in VMADM_STATE["vms"]:
            event["tag"] = "created/{}".format(uuid)
            for label in current_vms[uuid]:
                if label == "state":
                    continue
                event[label] = current_vms[uuid][label]

if event:
            ret.append(event)

# NOTE: deleted events
    for uuid in VMADM_STATE["vms"]:
        event = {}
        if uuid not in current_vms:
            event["tag"] = "deleted/{}".format(uuid)
            for label in VMADM_STATE["vms"][uuid]:
                if label == "state":
                    continue
                event[label] = VMADM_STATE["vms"][uuid][label]

if event:
            ret.append(event)

# NOTE: state change events
    for uuid in current_vms:
        event = {}
        if (
            VMADM_STATE["first_run"]
            or uuid not in VMADM_STATE["vms"]
            or current_vms[uuid].get("state", "unknown")
            != VMADM_STATE["vms"][uuid].get("state", "unknown")
        ):
            event["tag"] = "{}/{}".format(
                current_vms[uuid].get("state", "unknown"), uuid
            )
            for label in current_vms[uuid]:
                if label == "state":
                    continue
                event[label] = current_vms[uuid][label]

if event:
            ret.append(event)

# NOTE: update stored state
    VMADM_STATE["vms"] = current_vms

# NOTE: disable first_run
    if VMADM_STATE["first_run"]:
        VMADM_STATE["first_run"] = False

return ret

# vim: tabstop=4 expandtab shiftwidth=4 softtabstop=4