onthisday/telegram-bot/telegram-onthisday.py

#!/usr/bin/env python3
"""Telegram bot for onthisday.

Build it with: docker build -t telegram-onthisday .
Run it with something like: docker run -ti --rm -e EVENTS_TOKEN=your-telegram-token telegram-onthisday

Copyright 2019 Davide Alberani <da@erlug.linux.it> Apache 2.0 license
"""

import os
import sys
import time
import logging
import subprocess
from telegram.ext import Updater, CommandHandler

logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)

logger = logging.getLogger(__name__)


def getEvents(day=None):
    cmd = ['python3', '/onthisday/onthisday.py']
    if day:
        try:
            time.strptime(day, '%m/%d')
            cmd += ['--date', day]
        except Exception:
            pass
    if len(sys.argv) > 1:
        cmd += sys.argv[1:]
    process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = process.communicate()
    try:
        stdout = stdout.strip()
        stdout = stdout.decode('utf8')
    except:
        return 'uh-oh: something was wrong with the encoding of the events; try again'
    if process.returncode != 0:
        return 'something terrible is happening: exit code: %s, stderr: %s' % (
                process.returncode, stderr.decode('utf8'))
    if not stdout:
        return 'sadness: the events list is empty; try again'
    return stdout


def events(bot, update, args=None):
    day = None
    if args and args[0]:
        day = args[0].strip()
    events = getEvents(day)
    logging.info('%s wants some news; serving:\n%s' % (update.message.from_user.name, events))
    update.message.reply_text(events)


def about(bot, update):
    logging.info('%s required more info' % update.message.from_user.name)
    update.message.reply_text('See https://github.com/alberanid/onthisday\n\n/today to use the current date\n\n/date 07/30 for July 30')


if __name__ == '__main__':
    if 'EVENTS_TOKEN' not in os.environ:
        print("Please specify the Telegram token in the EVENTS_TOKEN environment variable")
    logging.info('start serving very important facts')
    updater = Updater(os.environ['EVENTS_TOKEN'])
    updater.dispatcher.add_handler(CommandHandler('today', events, pass_args=True))
    updater.dispatcher.add_handler(CommandHandler('start', events, pass_args=True))
    updater.dispatcher.add_handler(CommandHandler('date', events, pass_args=True))
    updater.dispatcher.add_handler(CommandHandler('about', about))
    updater.dispatcher.add_handler(CommandHandler('help', about))
    updater.start_polling()
    updater.idle()