From 076eb8976006bb06447bf3cdbcffb9eb6ed694f0 Mon Sep 17 00:00:00 2001
From: Egor <egor.andrianovv@gmail.com>
Date: Mon, 22 Dec 2025 15:53:56 +0300
Subject: [PATCH] Update menu.py

---
 app/handlers/menu.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

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"