Archived
refactor: balance handling and upload processing
This commit is contained in:
+24
-5
@@ -1,5 +1,5 @@
|
||||
from aiogram import Router
|
||||
from aiogram.types import CallbackQuery
|
||||
from aiogram.types import CallbackQuery, Message
|
||||
|
||||
from bot.api.asf import get_all_bots
|
||||
from bot.logging_utils import get_logger
|
||||
@@ -27,6 +27,28 @@ async def owned_callback_bot(callback: CallbackQuery, prefix: str):
|
||||
return await get_owned_bot(callback.from_user.id, account_id)
|
||||
|
||||
|
||||
async def send_bots_menu(message: Message, user_id: int) -> None:
|
||||
"""Отправляет отдельное сообщение со списком ботов пользователя."""
|
||||
response = get_all_bots()
|
||||
if response.status_code != 200 or not response.json().get("Success"):
|
||||
await message.answer("Не удалось загрузить список ботов")
|
||||
return
|
||||
|
||||
asf_bots = response.json().get("Result", {})
|
||||
if user_id == ADMIN_ID:
|
||||
await ensure_admin_compatibility(list(asf_bots))
|
||||
|
||||
accounts = await list_user_bot_accounts(user_id)
|
||||
rows = [
|
||||
(account.id, account.bot_name, asf_bots.get(account.bot_name, {}))
|
||||
for account in accounts
|
||||
]
|
||||
await message.answer(
|
||||
f"<b>🤖 Ваших ботов:</b> {len(rows)}\n\n{get_farm_summary({name: bot for _, name, bot in rows})}",
|
||||
reply_markup=bots_keyboard(rows),
|
||||
)
|
||||
|
||||
|
||||
@router.callback_query(lambda c: c.data == "bot_loading")
|
||||
async def bot_loading_handler(callback: CallbackQuery) -> None:
|
||||
await callback.answer(
|
||||
@@ -37,10 +59,8 @@ async def bot_loading_handler(callback: CallbackQuery) -> None:
|
||||
@router.callback_query(lambda c: c.data == "bots")
|
||||
async def bots_handler(callback: CallbackQuery) -> None:
|
||||
await stop_twofa_task(callback.message.message_id)
|
||||
|
||||
response = get_all_bots()
|
||||
await callback.answer()
|
||||
|
||||
response = get_all_bots()
|
||||
if response.status_code != 200 or not response.json().get("Success"):
|
||||
await callback.message.edit_text("Не удалось загрузить список ботов")
|
||||
return
|
||||
@@ -54,7 +74,6 @@ async def bots_handler(callback: CallbackQuery) -> None:
|
||||
(account.id, account.bot_name, asf_bots.get(account.bot_name, {}))
|
||||
for account in accounts
|
||||
]
|
||||
|
||||
await callback.message.edit_text(
|
||||
f"<b>🤖 Ваших ботов:</b> {len(rows)}\n\n{get_farm_summary({name: bot for _, name, bot in rows})}",
|
||||
reply_markup=bots_keyboard(rows),
|
||||
|
||||
Reference in New Issue
Block a user