В данной статье хотел бы рассказать об отправке уведомлений в Telegram чат при использовании SaltStack. Начиная с версии 2015.5.0, SaltStack предоставляет интеграцию со Slack из коробки, однако Telegram также является популярным мессенджером и активно используется среди российских пользователей. Поэтому надеюсь, что статья окажется полезна ее читателям.

Введение

Вообще говоря, SaltStack представляет собой конструктор и, как и многие другие инструменты, предоставляет возможности для кастомизации и расширения. В частности, собственные исполняемые модули.

О том, как можно создать Telegram бота на Python подробно рассказывается здесь. В качестве дополнения будет описано, как, используя небольшой Python-модуль, такой подход можно привязать к SaltStack.

Процесс настройки

Приступать к написанию модуля следует после того, как:

1. Создан бот
2. Бот дабавлен в чат
3. Получен chat_id

Примечание. Получить id чата можно с помощью следующего скрипта на python:

import requests

URL = 'https://api.telegram.org/bot'
TOKEN = <токен вашего бота>

try:
    request = requests.post('{url}{token}/getUpdates'.format(url=URL, token=TOKEN))
    print request.json()['result'][0]['message']['chat']['id']
except Exception,e:
    print str(e)

Итак, переходим к главному. Как описано в документации, SaltStack модуль должен располагаться в директории _modules/ и выглядеть следующим образом:

import requests

URL='https://api.telegram.org/bot'

def notify(message, token, chat_id):
        message_data = {
                'chat_id': chat_id,
                'text': message
        }

        try:
                request = requests.post('{url}{token}/sendMessage'.format(url=URL, token=token), data=message_data)
        except Exception,e:
                return False, str(e)

        if not request.status_code == 200:
                return False, "Return status is unsuccessful"
        # для наглядности вторым значением возвращается строка со служебной информацией.
        return True, "Message was successfully sent"

Далее необходимо выполнить команду синхронизации модулей, чтобы они появились на миньонах:
salt '*' saltutil.sync_modules

Если все завершилось успешно, результат будет примерно следующим:

И, напоследок, создаем файл состояния (в данном примере — send_telegram.sls)

send message about minion id:
  module.run:
    # telegram - имя python-модуля, notify - метод в этом модуле
    - name: telegram.notify
    - kwargs:
      message: command executed on minion with id {{ grains['id'] }}
      token: <токен вашего бота>
      chat_id: <id чата в Telegram>

Проверяем работоспособность созданного модуля:
salt '*' state.apply send_telegram

На стороне мастера:

В чате:

Источники

1. Телеграм бот на Python с использованием только requests
2. Writing execution modules
3. How do I use salt states?