Little fixes, add exception handler

birthday_pool_bot/telegram_bot/ui/dispatcher.py

@@ -8,10 +8,12 @@ from .callback_data import (
     MenuCallbackData,
     NewSubscriptionCallbackData,
     NewSubscriptionConfirmCallbackData,
+    NotifyNewPoolCallbackData,
     PoolActionEnum,
     PoolCallbackData,
     PoolsBackCallbackData,
     PoolsCallbackData,
+    SubscribeToNewPoolAnswerEnum,
     SubscriptionActionEnum,
     SubscriptionCallbackData,
     SubscriptionsCallbackData,
@@ -19,6 +21,7 @@ from .callback_data import (
 from .middlewares import (
     AuthMiddleware,
     DependsMiddleware,
+    ExceptionHandlerMiddleware,
     # TypingMiddleware,
 )
 from .states import (
@@ -36,6 +39,7 @@ def setup_dispatcher(
         dispatcher: aiogram.Dispatcher,
         repositories_container: RepositoriesContainer,
 ):
+    exception_handler_middleware = ExceptionHandlerMiddleware()
     # typing_middleware = TypingMiddleware()
     auth_middleware = AuthMiddleware(users_repository=repositories_container.users)
     users_repository_middleware = DependsMiddleware(
@@ -51,6 +55,7 @@ def setup_dispatcher(
         object_=repositories_container.subscriptions,
     )
     middlewares = [
+        exception_handler_middleware,
         # typing_middleware,
         auth_middleware,
         users_repository_middleware,
@@ -333,6 +338,16 @@ def setup_router(router: aiogram.Router) -> aiogram.Router:
         PoolsCallbackData.filter(),
     )
 
+    # Notification about new pool
+    router.callback_query.register(
+        handlers.subscribe_to_new_pool_callback_handler,
+        NotifyNewPoolCallbackData.filter(aiogram.F.answer == SubscribeToNewPoolAnswerEnum.YES),
+    )
+    router.callback_query.register(
+        handlers.hide_new_pool_notification,
+        NotifyNewPoolCallbackData.filter(aiogram.F.answer == SubscribeToNewPoolAnswerEnum.NO),
+    )
+
     # Fallback
     router.message.register(handlers.fallback_message_handler)