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 to emit Telegram messages

Requires the python-telegram-bot library

"""
import logging

import salt.utils.beacons

try:
    import telegram

logging.getLogger("telegram").setLevel(logging.CRITICAL)
    HAS_TELEGRAM = True
except ImportError:
    HAS_TELEGRAM = False

log = logging.getLogger(__name__)

__virtualname__ = "telegram_bot_msg"

def __virtual__():
    if HAS_TELEGRAM:
        return __virtualname__
    else:
        return False

def validate(config):
    """
    Validate the beacon configuration
    """
    if not isinstance(config, list):
        return False, "Configuration for telegram_bot_msg beacon must be a list."

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

if not all(
        config.get(required_config) for required_config in ["token", "accept_from"]
    ):
        return (
            False,
            "Not all required configuration for telegram_bot_msg are set.",
        )

if not isinstance(config.get("accept_from"), list):
        return (
            False,
            "Configuration for telegram_bot_msg, "
            "accept_from must be a list of usernames.",
        )

return True, "Valid beacon configuration."

def beacon(config):
    """
    Emit a dict with a key "msgs" whose value is a list of messages
    sent to the configured bot by one of the allowed usernames.

.. code-block:: yaml

beacons:
          telegram_bot_msg:
            - token: "<bot access token>"
            - accept_from:
              - "<valid username>"
            - interval: 10

"""

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

log.debug("telegram_bot_msg beacon starting")
    ret = []
    output = {}
    output["msgs"] = []

bot = telegram.Bot(config["token"])
    updates = bot.get_updates(limit=100, timeout=0)

log.debug("Num updates: %d", len(updates))
    if not updates:
        log.debug("Telegram Bot beacon has no new messages")
        return ret

latest_update_id = 0
    for update in updates:
        message = update.message

if update.update_id > latest_update_id:
            latest_update_id = update.update_id

if message.chat.username in config["accept_from"]:
            output["msgs"].append(message.to_dict())

# mark in the server that previous messages are processed
    bot.get_updates(offset=latest_update_id + 1)

log.debug("Emitting %d messages.", len(output["msgs"]))
    if output["msgs"]:
        ret.append(output)
    return ret