todobot.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# This program is dedicated to the public domain under the CC0 license.

"""
Simple Bot to reply to Telegram messages.
First, a few handler functions are defined. Then, those functions are passed to
the Dispatcher and registered at their respective places.
Then, the bot is started and runs until we press Ctrl-C on the command line.
Usage:
Basic Echobot example, repeats messages.
Press Ctrl-C on the command line or send a signal to the process to stop the
bot.
"""

import logging
import os
from telegram import Bot, Update, Message
from telegram.ext.callbackcontext import CallbackContext
from telegram.ext import (Updater, CommandHandler,
                          PicklePersistence)

import modules.todocore
import modules.dart
import modules.fun

token = os.getenv('TOKEN')
bot = Bot(token=token)

# Enable logging
logging.basicConfig(filename='bot.log', format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
                    level=logging.INFO)

logger = logging.getLogger(__name__)
pickle = PicklePersistence(filename='telegram_data.pickle')


# Define a few command handlers. These usually take the two arguments update and
# context. Error handlers also receive the raised TelegramError object in error.


def start(update: Update, context: CallbackContext):
    """Send a message when the command /start is issued."""
    assert isinstance(update.message, Message)
    update.message.reply_text('I love Qv2ray!')


def error(update: Update, context: CallbackContext):
    """Log Errors caused by Updates."""
    logger.warning('Update "%s" caused error "%s"', update, context.error)
    print(context.error)


def main():
    """Start the bot."""
    todo_engine = modules.todocore.TodoEngine(bot)
    dart_engine = modules.dart.Darter(bot)
    fun_engine = modules.fun.HaveSomeFun(bot)
    # Create the Updater and pass it your bot's token.
    # Make sure to set use_context=True to use the new context based callbacks
    # Post version 12 this will no longer be necessary
    updater = Updater(token, persistence=pickle, use_context=True)
    # Get the dispatcher to register handlers
    dp = updater.dispatcher
    dp.add_handler(CommandHandler("start", start))

    for command in todo_engine.getCommands():
        dp.add_handler(command)

    for command in dart_engine.getCommands():
        dp.add_handler(command)

    for command in fun_engine.getCommands():
        dp.add_handler(command)

    # log all errors
    dp.add_error_handler(error)

    # Start the Bot
    updater.start_polling()

    # Run the bot until you press Ctrl-C or the process receives SIGINT,
    # SIGTERM or SIGABRT. This should be used most of the time, since
    # start_polling() is non-blocking and will stop the bot gracefully.
    updater.idle()


if __name__ == '__main__':
    main()