Oct 13 13:22:36 vps1707806210 python[13578]: 2024-10-13 13:22:36,278 - bot.api - INFO - Цены для сервиса baj: {'0': {}}
Oct 13 13:22:36 vps1707806210 python[13578]: 2024-10-13 13:22:36,278 - bot.api - ERROR - Страна '0' отсутствует в данных: {'0': {}}
Oct 13 13:22:36 vps1707806210 python[13578]: 2024-10-13 13:22:36,279 - bot.handlers.sms - WARNING - Попытка 1 не удалась, повтор через 2 секунд.
Oct 13 13:22:38 vps1707806210 python[13578]: 2024-10-13 13:22:38,281 - bot.api - INFO - Запрос цен для сервиса: baj, страна: 0, валюта: 643
Oct 13 13:22:38 vps1707806210 python[13578]: 2024-10-13 13:22:38,475 - bot.api - INFO - Цены для сервиса baj: {'0': {}}
Oct 13 13:22:38 vps1707806210 python[13578]: 2024-10-13 13:22:38,475 - bot.api - ERROR - Страна '0' отсутствует в данных: {'0': {}}
Oct 13 13:22:38 vps1707806210 python[13578]: 2024-10-13 13:22:38,476 - bot.handlers.sms - WARNING - Попытка 2 не удалась, повтор через 2 секунд.
Oct 13 13:22:40 vps1707806210 python[13578]: 2024-10-13 13:22:40,479 - bot.api - INFO - Запрос цен для сервиса: baj, страна: 0, валюта: 643
Oct 13 13:22:40 vps1707806210 python[13578]: 2024-10-13 13:22:40,640 - bot.api - INFO - Цены для сервиса baj: {'0': {}}
Oct 13 13:22:40 vps1707806210 python[13578]: 2024-10-13 13:22:40,640 - bot.api - ERROR - Страна '0' отсутствует в данных: {'0': {}}
Oct 13 13:22:40 vps1707806210 python[13578]: 2024-10-13 13:22:40,641 - bot.handlers.sms - WARNING - Попытка 3 не удалась, повтор через 2 секунд.
Oct 13 13:22:42 vps1707806210 python[13578]: 2024-10-13 13:22:42,721 - aiogram.event - INFO - Update id=480218641 is handled. Duration 6776 ms by bot id=7625597085



1. так, смотри, когда идет запрос и отображатеся сообщение "⏳ Пожалуйста, подождите, идет обработка вашего запроса...", нужно под этим сообщением добавить кнопку "Отмена", которая будет вовзаращть пользователя к списку сервисов
2. когда идет запрос и отображатеся сообщение "⏳ Пожалуйста, подождите, идет обработка вашего запроса...", нужно, чтобы в этом сообщении так же отображалось кол-во запросов, к примеру "⏳ Пожалуйста, подождите, идет обработка вашего запроса...1/3" и чтобы сообщение менялось согласно реальным запросам и так же увеличь промежуток между запросами, увеличь их дот 5 секунд


import asyncio
import logging
import time
import json
from pathlib import Path
from typing import Optional, List, Tuple
from aiogram import Dispatcher, types, F
from aiogram.utils.keyboard import InlineKeyboardBuilder
from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup

from ..data import get_user_api_key, get_user_operator, get_user_country
from ..api import SmsHubAPI
from ..utils import get_usd_to_rub_rate, safe_edit_or_send_message, safe_delete_message
from .main_menu import send_main_menu
from .activation import user_activations, poll_for_sms
from ..constants import OPERATOR_DISPLAY_NAMES, COUNTRIES, SERVICE_DISPLAY_NAMES

logger = logging.getLogger(__name__)

# Глобальные настройки повторных попыток
MAX_RETRIES = 3
RETRY_DELAY = 2  # секунды

class SmsOrderState(StatesGroup):
    waiting_for_service = State()
    waiting_for_max_price = State()

SERVICES: List[Tuple[str, str]] = list(SERVICE_DISPLAY_NAMES.items())
POPULARITY_FILE = Path("service_popularity.json")

def load_popularity_data():
    if POPULARITY_FILE.exists():
        with open(POPULARITY_FILE, 'r', encoding='utf-8') as f:
            return json.load(f)
    return {}

def save_popularity_data(popularity_data):
    with open(POPULARITY_FILE, 'w', encoding='utf-8') as f:
        json.dump(popularity_data, f, ensure_ascii=False, indent=4)

def update_service_popularity(service_code: str):
    popularity_data = load_popularity_data()
    popularity_data[service_code] = popularity_data.get(service_code, 0) + 1
    save_popularity_data(popularity_data)

def get_sorted_services():
    popularity_data = load_popularity_data()
    return sorted(SERVICES, key=lambda service: popularity_data.get(service[0], 0), reverse=True)

def get_service_pages(services: List[Tuple[str, str]], page_size: int = 18) -> List[List[Tuple[str, str]]]:
    return [services[i:i+page_size] for i in range(0, len(services), page_size)]

def build_service_keyboard(page_number: int = 0) -> InlineKeyboardBuilder:
    sorted_services = get_sorted_services()
    services_pages = get_service_pages(sorted_services)
    keyboard = InlineKeyboardBuilder()

    if page_number < len(services_pages):
        current_page = services_pages[page_number]
        for code, name in current_page:
            keyboard.button(text=name, callback_data=f"service_select:{code}")
        if page_number > 0:
            keyboard.button(text="⬅️ Назад", callback_data=f"service_page:{page_number - 1}", width=3)
        if page_number < len(services_pages) - 1:
            keyboard.button(text="➡️ Вперед", callback_data=f"service_page:{page_number + 1}", width=3)
    keyboard.button(text="🔙 Назад в меню", callback_data="cancel_operation", width=3)
    keyboard.adjust(3)
    return keyboard

async def safe_answer_callback_query(callback_query: types.CallbackQuery):
    try:
        await callback_query.answer()
    except Exception as e:
        logger.warning(f"Не удалось ответить на callback_query: {e}")

async def get_sms_handler(callback_query: types.CallbackQuery, state: FSMContext):
    await safe_answer_callback_query(callback_query)
    user_api_key = get_user_api_key(callback_query.from_user.id)
    if not user_api_key:
        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="⚙️ Настройки", callback_data="settings")
        keyboard.button(text="🔙 Назад", callback_data="back_to_main_menu")
        keyboard.adjust(1)
        await safe_edit_or_send_message(
            message=callback_query.message,
            new_text="Пожалуйста, сначала настройте ваш API ключ в разделе 'Настройки'.",
            state=state,
            reply_markup=keyboard.as_markup()
        )
        return

    logger.info(f"Пользователь {callback_query.from_user.id} начал процесс получения номера.")
    keyboard = build_service_keyboard()
    await safe_edit_or_send_message(
        message=callback_query.message,
        new_text=(
            "Выберите сервис, нажав на кнопку ниже.\n"
            "Если вы хотите, вы также можете ввести полное название сервиса или хотя бы часть его названия, чтобы получить результаты поиска."
        ),
        state=state,
        reply_markup=keyboard.as_markup()
    )
    await state.set_state(SmsOrderState.waiting_for_service)

async def cancel_operation_handler(callback_query: types.CallbackQuery, state: FSMContext):
    await safe_answer_callback_query(callback_query)
    logger.info(f"Пользователь {callback_query.from_user.id} отменил операцию.")
    await state.set_state(None)
    await send_main_menu(callback_query.message, state)

async def service_page_handler(callback_query: types.CallbackQuery, state: FSMContext):
    await safe_answer_callback_query(callback_query)
    page_number = int(callback_query.data.split(":")[1])
    keyboard = build_service_keyboard(page_number)
    await safe_edit_or_send_message(
        message=callback_query.message,
        new_text="Выберите сервис или введите код сервиса вручную:",
        state=state,
        reply_markup=keyboard.as_markup()
    )

async def service_selected_handler(callback_query: types.CallbackQuery, state: FSMContext):
    await safe_answer_callback_query(callback_query)
    service_code = callback_query.data.split(":")[1]
    update_service_popularity(service_code)
    await state.update_data(selected_service=service_code)
    await service_handler_logic(
        callback_query.message, state,
        user_id=callback_query.from_user.id,
        service_code=service_code,
        is_service_selection=True
    )

async def service_handler(message: types.Message, state: FSMContext, service_code: Optional[str] = None, user_id: Optional[int] = None):
    user_input = message.text.strip().lower()
    user_id = user_id or message.from_user.id

    logger.info(f"Пользователь {user_id} ввел сервис: {user_input}")

    service_code = next((code for code, name in SERVICE_DISPLAY_NAMES.items() if name.lower() == user_input or code.lower() == user_input), None)

    if service_code is None:
        matched_services = [(code, name) for code, name in SERVICE_DISPLAY_NAMES.items() if user_input in name.lower()]
        if matched_services:
            keyboard = InlineKeyboardBuilder()
            for code, name in matched_services:
                keyboard.button(text=name, callback_data=f"service_select:{code}")
            keyboard.adjust(3)
            keyboard.button(text="🔙 Назад", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Пожалуйста, выберите один из представленных сервисов:",
                state=state,
                reply_markup=keyboard.as_markup()
            )
        else:
            keyboard = InlineKeyboardBuilder()
            keyboard.button(text="🔙 Назад к сервисам", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Сервис не найден. Попробуйте ввести код или полное название сервиса.",
                state=state,
                reply_markup=keyboard.as_markup()
            )
    else:
        await state.update_data(selected_service=service_code)
        await service_handler_logic(message, state, user_id=user_id, service_code=service_code)

async def service_handler_logic(message: types.Message, state: FSMContext, user_id: int, service_code: str, is_service_selection: bool = False):
    logger.info(f"Пользователь {user_id} выбрал сервис: {service_code}")

    user_api_key = get_user_api_key(user_id)
    user_operator = get_user_operator(user_id) or "any"
    user_country_code = get_user_country(user_id) or "0"
    country_name = COUNTRIES.get(user_country_code, user_country_code)

    if not user_api_key:
        await safe_edit_or_send_message(
            message=message,
            new_text="Пожалуйста, сначала настройте ваш API ключ в разделе 'Настройки'.",
            state=state
        )
        await state.set_state(None)
        await send_main_menu(message, state)
        return

    smshub_api = SmsHubAPI(user_api_key)
    usd_to_rub_rate = await get_usd_to_rub_rate()

    loading_message = await safe_edit_or_send_message(
        message=message,
        new_text="⏳ Пожалуйста, подождите, идет обработка вашего запроса...",
        state=state
    )

    try:
        price_info = None
        for attempt in range(MAX_RETRIES):
            try:
                price_info = await smshub_api.get_price_for_service(service_code, country=user_country_code)
                if price_info:
                    break
                else:
                    logger.warning(f"Попытка {attempt + 1} не удалась, повтор через {RETRY_DELAY} секунд.")
                    await asyncio.sleep(RETRY_DELAY)
            except Exception as e:
                logger.error(f"Ошибка попытки {attempt + 1} получить информацию о ценах: {e}")
                await asyncio.sleep(RETRY_DELAY)
        else:
            keyboard = InlineKeyboardBuilder()
            keyboard.button(text="Назад", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Не удалось получить информацию о ценах для выбранного сервиса. Пожалуйста, попробуйте другой сервис или повторите попытку позже.",
                state=state,
                reply_markup=keyboard.as_markup()
            )
            return

        numbers_status = None
        for attempt in range(MAX_RETRIES):
            try:
                numbers_status = await smshub_api.get_numbers_status(country=user_country_code, operator=user_operator)
                if numbers_status:
                    break
                else:
                    logger.warning(f"Попытка {attempt + 1} не удалась, повтор через {RETRY_DELAY} секунд.")
                    await asyncio.sleep(RETRY_DELAY)
            except Exception as e:
                logger.error(f"Ошибка попытки {attempt + 1} получить статус номеров: {e}")
                await asyncio.sleep(RETRY_DELAY)
        else:
            keyboard = InlineKeyboardBuilder()
            keyboard.button(text="Назад", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Не удалось получить информацию о доступных номерах. Попробуйте снова позже.",
                state=state,
                reply_markup=keyboard.as_markup()
            )
            return

        service_key = f"{service_code}_0"
        available_numbers = int(numbers_status.get(service_key, 0) or 0)
        if available_numbers == 0:
            keyboard = InlineKeyboardBuilder()
            keyboard.button(text="Назад", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Нет доступных номеров у выбранного оператора для данного сервиса. Пожалуйста, выберите другого оператора или попробуйте позже.",
                state=state,
                reply_markup=keyboard.as_markup()
            )
            return

        prices = [float(price_str) for price_str in price_info.keys() if price_str is not None]
        if not prices:
            keyboard = InlineKeyboardBuilder()
            keyboard.button(text="Назад", callback_data="service_page:0")
            await safe_edit_or_send_message(
                message=message,
                new_text="Цены для выбранного сервиса недоступны. Пожалуйста, попробуйте другой сервис или повторите попытку позже.",
                state=state,
                reply_markup=keyboard.as_markup()
            )
            return

        min_price_usd = min(prices)
        max_price_usd = max(prices)
        min_price_rub = min_price_usd * usd_to_rub_rate if usd_to_rub_rate else 0
        max_price_rub = max_price_usd * usd_to_rub_rate if usd_to_rub_rate else 0

        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="❌ Отмена", callback_data="cancel_operation")
        keyboard.adjust(1)

        operator_display = OPERATOR_DISPLAY_NAMES.get(user_operator, user_operator)
        service_display = SERVICE_DISPLAY_NAMES.get(service_code, service_code)

        message_text = (
            f"*Страна:* `{country_name}`\n"
            f"*Доступно номеров:* `{available_numbers}`\n"
            f"*Оператор:* `{operator_display}`\n"
            f"*Сервис:* `{service_display}`\n"
            f"*Текущие цены для сервиса* `{service_display}`:\n"
            f"от `{min_price_usd:.4f}` $ (₽{min_price_rub:.2f}) до `{max_price_usd:.4f}` $ (₽{max_price_rub:.2f}).\n\n"
            f"_Введите максимальную стоимость в $ (например, 10):_"
        )

        sent_message = await safe_edit_or_send_message(
            message=message,
            new_text=message_text,
            state=state,
            reply_markup=keyboard.as_markup(),
            parse_mode="Markdown"
        )
        
        if sent_message:
            await state.update_data(last_message_id=sent_message.message_id)

        await state.update_data(
            service=service_code,
            service_display=service_display,
            country_name=country_name,
            available_numbers=available_numbers,
            operator_display=operator_display,
            min_price=min_price_usd,
            max_price=max_price_usd,
            min_price_rub=min_price_rub,
            max_price_rub=max_price_rub,
            user_id=user_id
        )
        await state.set_state(SmsOrderState.waiting_for_max_price)
    except Exception as e:
        logger.error(f"Ошибка при получении данных для сервиса '{service_code}': {e}")
        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="Назад", callback_data="service_page:0")
        await safe_edit_or_send_message(
            message=message,
            new_text="Произошла ошибка при обработке запроса. Попробуйте снова.",
            state=state,
            reply_markup=keyboard.as_markup()
        )
        await state.set_state(None)
        await send_main_menu(message, state)
    finally:
        if loading_message and hasattr(loading_message, 'message_id'):  # Ensure loading message exists before deleting
            await safe_delete_message(message.bot, message.chat.id, loading_message.message_id)

async def max_price_handler(message: types.Message, state: FSMContext):
    max_price_input = message.text.strip()
    data = await state.get_data()
    user_id = data.get('user_id', message.from_user.id)
    logger.info(f"Пользователь {user_id} указал максимальную стоимость: {max_price_input}")

    try:
        max_price = float(max_price_input)
    except ValueError:
        user_data = await state.get_data()
        min_price = user_data.get('min_price')
        max_price_val = user_data.get('max_price')
        min_price_rub = user_data.get('min_price_rub')
        max_price_rub = user_data.get('max_price_rub')

        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="❌ Отмена", callback_data="cancel_operation")
        keyboard.adjust(1)
        await safe_edit_or_send_message(
            message=message,
            new_text=(
                f"Пожалуйста, введите корректное число для максимальной стоимости. "
                f"От `{min_price:.4f}` $ (₽{min_price_rub:.2f}) до `{max_price_val:.4f}` $ (₽{max_price_rub:.2f})."
            ),
            state=state,
            reply_markup=keyboard.as_markup(),
            parse_mode="Markdown"
        )
        return

    user_data = await state.get_data()
    service = user_data['service']
    service_display = user_data['service_display']
    min_price = user_data['min_price']
    country_name = user_data['country_name']
    operator_display = user_data['operator_display']
    user_api_key = get_user_api_key(user_id)
    smshub_api = SmsHubAPI(user_api_key)
    user_country_code = get_user_country(user_id) or "0"

    if max_price < min_price:
        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="❌ Отмена", callback_data="cancel_operation")
        keyboard.adjust(1)
        await safe_edit_or_send_message(
            message=message,
            new_text=f"Максимальная стоимость должна быть не менее `{min_price:.4f}` $.",
            state=state,
            reply_markup=keyboard.as_markup(),
            parse_mode="Markdown"
        )
        return

    user_operator = get_user_operator(user_id) or "any"
    loading_message = await safe_edit_or_send_message(
        message=message,
        new_text="⏳ Пожалуйста, подождите, идет получение номера...",
        state=state
    )

    result = None
    for attempt in range(MAX_RETRIES):
        try:
            result = await smshub_api.get_number(service, country=user_country_code, max_price=max_price, operator=user_operator)
            if result and result.startswith("ACCESS_NUMBER"):
                break
            elif not result:
                logger.warning(f"Попытка {attempt + 1} не удалась, повтор через {RETRY_DELAY} секунд.")
                await asyncio.sleep(RETRY_DELAY)
        except Exception as e:
            logger.error(f"Ошибка попытки {attempt + 1} получить номер: {e}")
            await asyncio.sleep(RETRY_DELAY)
    else:
        logger.error(f"Не удалось получить номер после {MAX_RETRIES} попыток.")
        suggestion = "Попробуйте увеличить максимальную цену или повторите попытку позже."
        await safe_edit_or_send_message(
            message=message,
            new_text=f"Не удалось получить номер для оператора {operator_display} по данной цене.\n{suggestion}",
            state=state
        )
        await send_main_menu(message, state)

    if loading_message and hasattr(loading_message, 'message_id'):  # Ensure loading message exists before deleting
        await safe_delete_message(message.bot, message.chat.id, loading_message.message_id)

    if result and result.startswith("ACCESS_NUMBER"):
        _, activation_id, number = result.split(":")
        logger.info(f"Пользователь {user_id} получил номер: {number}, активация ID: {activation_id}")

        if user_id not in user_activations:
            user_activations[user_id] = {}
        user_activations[user_id][activation_id] = {
            'number': number,
            'service': service,
            'service_display': service_display,
            'country_name': country_name,
            'status': 'Ожидание SMS',
            'task': None,
            'last_update': time.time()
        }

        keyboard = InlineKeyboardBuilder()
        keyboard.button(text="📋 Управлять активацией", callback_data=f"manage_activation:{activation_id}")
        keyboard.button(text="🔙 Главное меню", callback_data="back_to_main_menu")
        keyboard.adjust(1)

        last_message_id = user_data.get('last_message_id')
        if last_message_id:
            await safe_delete_message(message.bot, message.chat.id, last_message_id)

        message_text = (
            f"*Получен номер:* `{number}`\n"
            f"*Страна:* `{country_name}`\n"
            f"*Сервис:* `{service_display}`\n"
            f"Ожидание SMS...\n"
            f"Вы можете управлять активацией или вернуться в главное меню."
        )
        sent_message = await safe_edit_or_send_message(
            message=message,
            new_text=message_text,
            state=state,
            reply_markup=keyboard.as_markup(),
            parse_mode="Markdown"
        )
        if sent_message:
            await state.update_data(last_message_id=sent_message.message_id)

        await state.set_state(None)

        task = asyncio.create_task(poll_for_sms(
            bot=message.bot,
            user_id=user_id,
            chat_id=message.chat.id,
            activation_id=activation_id,
            smshub_api=smshub_api
        ))
        user_activations[user_id][activation_id]['task'] = task
    else:
        logger.error(f"Не удалось получить номер. Ответ: {result}")
        suggestion = "Попробуйте увеличить максимальную цену или повторите попытку позже."
        await safe_edit_or_send_message(
            message=message,
            new_text=f"Не удалось получить номер для оператора {operator_display} по данной цене.\n{suggestion}",
            state=state
        )
        await send_main_menu(message, state)

def register_sms_handlers(dp: Dispatcher):
    dp.callback_query.register(get_sms_handler, F.data == "get_sms")
    dp.callback_query.register(cancel_operation_handler, F.data == "cancel_operation")
    dp.callback_query.register(service_page_handler, F.data.startswith("service_page:"))
    dp.callback_query.register(service_selected_handler, F.data.startswith("service_select:"))
    dp.message.register(service_handler, SmsOrderState.waiting_for_service)
    dp.message.register(max_price_handler, SmsOrderState.waiting_for_max_price)