diff --git a/app/handlers/menu.py b/app/handlers/menu.py index 6d25a2c4..a70cb9b1 100644 --- a/app/handlers/menu.py +++ b/app/handlers/menu.py @@ -41,7 +41,7 @@ from app.services.public_offer_service import PublicOfferService from app.services.faq_service import FaqService from app.utils.timezone import format_local_datetime from app.utils.pricing_utils import format_period_description -from app.handlers.subscription.traffic import handle_add_traffic +from app.handlers.subscription.traffic import handle_add_traffic, add_traffic logger = logging.getLogger(__name__) @@ -1385,6 +1385,11 @@ def register_handlers(dp: Dispatcher): F.data == "buy_traffic" ) + dp.callback_query.register( + add_traffic, + F.data.startswith("add_traffic_") + ) + dp.callback_query.register( handle_activate_button, F.data == "activate_button"