main.py

import logging

import logzero
from telegram.ext import Updater, CommandHandler, MessageHandler, InlineQueryHandler, \
    ChosenInlineResultHandler, RegexHandler

import config
import handlers
import model
from custom_filters import is_in_database

logger = logzero.setup_logger(__name__, level=logging.INFO)

meme_storage = model.get_storage()


def error_handler(_, update, error):
    logger.warning('Update "%s" caused error "%s"', update, error)


def main():
    updater = Updater(config.TOKEN)

    dp = updater.dispatcher

    dp.add_handler(handlers.conversation)
    dp.add_handler(MessageHandler(is_in_database, handlers.meme))
    dp.add_handler(CommandHandler('name', handlers.name))
    dp.add_handler(CommandHandler('delete', handlers.delete))
    dp.add_handler(CommandHandler('rename', handlers.rename, pass_args=True))
    dp.add_handler(CommandHandler('fix', handlers.fix))
    dp.add_handler(CommandHandler('my', handlers.my))
    dp.add_handler(CommandHandler(['start', 'help'], handlers.help_))
    dp.add_handler(RegexHandler('/(?P<id>\d+)', handlers.get_by_id, pass_groupdict=True))
    dp.add_handler(InlineQueryHandler(handlers.inlinequery))
    dp.add_handler(ChosenInlineResultHandler(handlers.chosen_inline_result))

    dp.add_error_handler(error_handler)

    updater.start_polling()
    updater.idle()


if __name__ == '__main__':
    main()