diff --git a/handlers/payment/cryprobot_pay.py b/handlers/payment/cryprobot_pay.py index 173dc95d..2e5dd39a 100644 --- a/handlers/payment/cryprobot_pay.py +++ b/handlers/payment/cryprobot_pay.py @@ -50,8 +50,8 @@ async def send_message_with_deletion( return sent_message -@router.callback_query(F.data == "replenish_balance") -async def process_callback_replenish_balance( +@router.callback_query(F.data == "pay_cryptobot") +async def process_callback_pay_cryptobot( callback_query: types.CallbackQuery, state: FSMContext ): tg_id = callback_query.from_user.id diff --git a/handlers/payment/stars_pay.py b/handlers/payment/stars_pay.py index ff464cbe..deb49a52 100644 --- a/handlers/payment/stars_pay.py +++ b/handlers/payment/stars_pay.py @@ -46,8 +46,8 @@ async def send_message_with_deletion( return sent_message -@router.callback_query(F.data == "replenish_balance") -async def process_callback_replenish_balance( +@router.callback_query(F.data == "pay_stars") +async def process_callback_pay_stars( callback_query: types.CallbackQuery, state: FSMContext ): tg_id = callback_query.from_user.id diff --git a/handlers/payment/yookassa_pay.py b/handlers/payment/yookassa_pay.py index b1d1f36c..49d8fba0 100644 --- a/handlers/payment/yookassa_pay.py +++ b/handlers/payment/yookassa_pay.py @@ -55,8 +55,8 @@ async def send_message_with_deletion( return sent_message -@router.callback_query(F.data == "replenish_balance") -async def process_callback_replenish_balance( +@router.callback_query(F.data == "pay_yookassa") +async def process_callback_pay_yookassa( callback_query: types.CallbackQuery, state: FSMContext ): tg_id = callback_query.from_user.id diff --git a/handlers/profile.py b/handlers/profile.py index 8ee36394..0f168a73 100644 --- a/handlers/profile.py +++ b/handlers/profile.py @@ -7,7 +7,7 @@ from aiogram.utils.keyboard import InlineKeyboardBuilder from loguru import logger from bot import bot -from config import CHANNEL_URL, FREEKASSA_ENABLE, PAYMENT_METHOD, YOOKASSA_ENABLE +from config import CHANNEL_URL, CRYPTO_BOT_ENABLE, FREEKASSA_ENABLE, STARS_ENABLE, YOOKASSA_ENABLE from database import get_balance, get_key_count, get_referral_stats from handlers.texts import get_referral_link, invite_message_send, profile_message_send @@ -41,15 +41,32 @@ async def process_callback_view_profile( InlineKeyboardButton(text="➕ Устройство", callback_data="create_key"), InlineKeyboardButton(text="📱 Мои устр-ва", callback_data="view_keys"), ) - if YOOKASSA_ENABLE or FREEKASSA_ENABLE: + if YOOKASSA_ENABLE: builder.row( InlineKeyboardButton( - text="💳 Пополнить баланс", - callback_data=( - "pay_freekassa" - if PAYMENT_METHOD == "freekassa" - else "replenish_balance" - ), + text="💳 Пополнить баланс ЯКассой", + callback_data="pay_yookassa", + ) + ) + if FREEKASSA_ENABLE: + builder.row( + InlineKeyboardButton( + text="💳 Пополнить баланс Freekassa", + callback_data="pay_freekassa", + ) + ) + if CRYPTO_BOT_ENABLE: + builder.row( + InlineKeyboardButton( + text="💳 Пополнить баланс CryptoBot", + callback_data="pay_cryptobot", + ) + ) + if STARS_ENABLE: + builder.row( + InlineKeyboardButton( + text="💳 Пополнить баланс Звездами", + callback_data="pay_stars", ) ) builder.row(