class Main:
def __init__(self):
logger.info("The app has started")
databases = Databases()
logger.info("DB initialized")
self.telegram_updater = Updater(token=config.token)
self.user_manager = UserManager(databases, self.telegram_updater.bot)
self.telegram_updater.dispatcher.add_handler(
MessageHandler(filters=[], callback=self.message_handler))
self.telegram_updater.dispatcher.add_handler(
CallbackQueryHandler(self.callback_handler))
self.asyncio_handler = AsyncioFriendlyHandler(
self.initialize_asyncio_loop)
self.telegram_updater.dispatcher.add_handler(self.asyncio_handler)
self.telegram_updater.dispatcher.add_error_handler(self.telegram_error)
logger.debug('Adding asyncio update to queue')
self.telegram_updater.update_queue.put(AsyncioUpdate())
logger.info("Staring polling")
self.telegram_updater.start_polling(
timeout=config.timeout_for_pooling,
bootstrap_retries=config.retries_on_error)
logger.info("Idle")
self.telegram_updater.idle()
def initialize_asyncio_loop(self):
logger.debug('Initializing asyncio loop')
asyncio.set_event_loop(asyncio.new_event_loop())
self.telegram_updater.dispatcher.remove_handler(self.asyncio_handler)
def message_handler(self, bot, update):
self.user_manager.get_user(update.message.chat_id).process_message(
bot, update)
def callback_handler(self, bot, update):
self.user_manager.get_user(
update.callback_query.message.chat_id).process_callback(
bot, update)
def telegram_error(self, bot, update, error):
try:
raise error
except TelegramError as details:
logger.warning('There was a Telegram Error {}'.format(details))
raise
def test_get_user(self):
user_manager = UserManager()
self.assertEqual(11, user_manager.get_user(11).id)