12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273 |
- #!/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()
|