telegram.py

import logging
from pwnagotchi.voice import Voice
import pwnagotchi.plugins as plugins


class Telegram(plugins.Plugin):
    __author__ = 'djerfy@gmail.com'
    __version__ = '1.0.0'
    __license__ = 'GPL3'
    __description__ = 'Periodically sent messages to Telegram about the recent activity of pwnagotchi'

    def on_loaded(self):
        logging.info("telegram plugin loaded.")

    # called when there's available internet
    def on_internet_available(self, agent):
        config = agent.config()
        display = agent.view()
        last_session = agent.last_session

        if last_session.is_new() and last_session.handshakes > 0:

            try:
                import telegram
            except ImportError:
                logging.error("Couldn't import telegram")
                return

            logging.info("Detected new activity and internet, time to send a message!")

            picture = '/root/pwnagotchi.png'
            display.on_manual_mode(last_session)
            display.image().save(picture, 'png')
            display.update(force=True)

            try:
                logging.info("Connecting to Telegram...")

                message = Voice(lang=config['main']['lang']).on_last_session_tweet(last_session)

                bot = telegram.Bot(self.options['bot_token'])
                if self.options['send_picture'] is True:
                    bot.sendPhoto(chat_id=self.options['chat_id'], photo=open(picture, 'rb'))
                    logging.info("telegram: picture sent")
                if self.options['send_message'] is True:
                    bot.sendMessage(chat_id=self.options['chat_id'], text=message, disable_web_page_preview=True)
                    logging.info("telegram: message sent: %s" % message)

                last_session.save_session_id()
                display.set('status', 'Telegram notification sent!')
                display.update(force=True)
            except Exception:
                logging.exception("Error while sending on Telegram")