Telegram Alert

telegram_alerts.py

#!/usr/bin/env python
'''
Script to send alerts/messages to telegram.
e.g. ssh login alerts etc.
'''

import sys
import yaml
import telepot

# config file with telegram TOKEN and chat id. 
with open('/home/user/.my_config.yaml', 'r') as myfile:
    uconfig = yaml.load(myfile)

# pass message as argument to python script
message = sys.argv[1]
TOKEN = uconfig['TOKEN']
chat_id = uconfig['my_chat_id']

if __name__ == "__main__":
    # initialize the bot
    bot = telepot.Bot(TOKEN)

    # send the alert 
    bot.sendMessage(chat_id, message)