LCMbot/LCMbot.py

# -*- coding: utf-8 -*-
from telegram.ext import (Updater, CommandHandler, MessageHandler, Filters,
                          CallbackQueryHandler)
import handlers as hnd
from tale_handler import TaleHandler
import os
import logging
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                     level=logging.INFO)



#acquire token from environment variable LCMBOT_TOKEN (string type)
lcmbot_token = os.environ.get('LCMBOT_TOKEN')

def main():
    # create LCMbot

    updater = Updater(lcmbot_token, use_context=True)
    dp = updater.dispatcher

    # register handlers
    dp.add_handler(CommandHandler('pinglcm', hnd.ping))
    dp.add_handler(CommandHandler('checkwebsite', hnd.is_web_up))
    dp.add_handler(CommandHandler('vietnam', hnd.vietnam))
    dp.add_handler(CommandHandler('sellyourmother', hnd.sell_your_mother))
    dp.add_handler(CommandHandler('abuse150', hnd.abuse_150))
    dp.add_handler(CommandHandler('rndwisdom', hnd.speak, pass_args=True))

    tale_handler = TaleHandler()
    dp.add_handler(CommandHandler('addatale', tale_handler.prompt_user))
    dp.add_handler(MessageHandler([Filters.text], tale_handler.handle_new_tale))
    dp.add_handler(CallbackQueryHandler(tale_handler.save_tale))
    dp.add_handler(CommandHandler('tellatale', tale_handler.tell_a_tale))

    dp.add_error_handler(hnd.error)
    # dp.add_handler(MessageHandler([], hnd.print_msg_info))

    # start LCMbot
    updater.start_polling()

    # run until the process receives SIGINT, SIGTERM or SIGABRT
    updater.idle()


if __name__ == '__main__':
    main()