repeater.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

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

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

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

logger = logging.getLogger(__name__)


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():
    updater = Updater(token, use_context=True)
    dp = updater.dispatcher

    # log all errors
    dp.add_error_handler(error)

    repeat_filters = Filters.text & ~Filters.command
    # add repeat handler
    dp.add_handler(MessageHandler(repeat_filters, repeat))
    # add repeat clear handler
    dp.add_handler(MessageHandler(Filters.all & ~repeat_filters, repeat))

    # 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()