fixed callback bug

OneMagicKey · Apr 3, 2024 · 2754038 · 2754038
1 parent b55c20d
commit 2754038
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/bot.py b/src/bot.py
@@ -265,6 +265,7 @@ async def process_image(message: types.Message):
 
 
 @dp.callback_query(aiogram.F.func(lambda call: call.data.startswith("yolo")))
+@auth
 async def callback_model(call: types.CallbackQuery):
     """
     Callback function to handle model keyboard buttons.
@@ -283,6 +284,7 @@ async def callback_model(call: types.CallbackQuery):
 
 
 @dp.callback_query(aiogram.F.func(lambda call: call.data.startswith("language")))
+@auth
 async def callback_language(call: types.CallbackQuery):
     """
     Callback function to handle language keyboard buttons.
@@ -301,6 +303,7 @@ async def callback_language(call: types.CallbackQuery):
 
 
 @dp.callback_query(aiogram.F.func(lambda call: call.data.startswith("color")))
+@auth
 async def callback_color_scheme(call: types.CallbackQuery):
     """
     Callback function to handle color_scheme keyboard buttons.
@@ -319,6 +322,7 @@ async def callback_color_scheme(call: types.CallbackQuery):
 
 
 @dp.callback_query(aiogram.F.func(lambda call: call.data.startswith("retina")))
+@auth
 async def callback_retina_masks(call: types.CallbackQuery):
     """
     Callback function to handle retina_masks keyboard buttons.