flows.py

"""Диалоговые flow - состояния пользователя"""

# Хранилище состояний (можно заменить на Redis если нужно)
user_states = {}
user_data = {}


# Состояния
class State:
    MENU = "menu"
    WAITING_PHONE = "waiting_phone"
    WAITING_NAME = "waiting_name"
    WAITING_QUESTION = "waiting_question"  # Общие вопросы (визы, погода)
    WAITING_WISHES = "waiting_wishes"
    WAITING_CLARIFICATION = "waiting_clarification"  # Уточнения по заявке
    WAITING_CONFIRM = "waiting_confirm"  # Ожидаем подтверждение


def set_state(user_id: int, state: str, data: dict = None):
    """Установить состояние"""
    user_states[user_id] = state
    if data:
        user_data[user_id] = data


def get_state(user_id: int) -> str:
    """Получить состояние"""
    return user_states.get(user_id, State.MENU)


def get_data(user_id: int) -> dict:
    """Получить данные пользователя"""
    return user_data.get(user_id, {})


def update_data(user_id: int, key: str, value):
    """Обновить данные"""
    if user_id not in user_data:
        user_data[user_id] = {}
    user_data[user_id][key] = value


def clear_state(user_id: int):
    """Очистить состояние"""
    user_states.pop(user_id, None)
    user_data.pop(user_id, None)


# ==================== ТЕКСТЫ СООБЩЕНИЙ ====================

TEXTS = {
    "start": """👋 Добро пожаловать!

Я помогу подобрать идеальный тур для вашего отдыха.

Выберите действие ниже 👇""",

    "select_country": """🌍 Выберите страну, куда хотите поехать:""",

    "send_phone": """📱 Чтобы мы могли с вами связаться, отправьте номер телефона.

Нажмите кнопку "📞 Перезвоните мне" ниже 👇""",

    "send_name": """📝 Как к вам обращаться? Напишите ваше имя.""",

    "send_wishes": """💭 Напишите ваши пожелания к поездке:
- Даты или сезон
- Количество человек
- Бюджет (если есть)
- Что важно (отель, пляж, экскурсии...)""",

    "ask_question": """❓ <b>Справка о туризме</b>

Спрошу про визы, погоду, транспорт, достопримечательности.

<b>Чего я НЕ умею:</b>
• Не подбираю туры
• Не рекомендую страны
• Не составляю маршруты

<b>Что могу:</b>
• "Нужна ли виза в Турцию?"
• "Когда лучшее время для Мальдив?"
• "Как долететь до Египта?"
• "Что посмотреть в Риме?" """,

    "thanks_lead": """✅ Спасибо! Ваша заявка принята.

Мы свяжемся с вами в ближайшее время для уточнения деталей. 📞""",

    "cancel": """❌ Операция отменена.

Главное меню 👇""",

    "error": """😕 Что-то пошло не так. Попробуйте снова.""",
}


def format_country_info(country: dict) -> str:
    """Форматировать информацию о стране"""
    name = country.get("name", "Страна")
    flag = country.get("flag", "🌍")
    description = country.get("description", "")
    price_from = country.get("price_from", "")
    visa = country.get("visa_info", "")

    text = f"{flag} <b>{name}</b>\n\n"

    if description:
        text += f"{description}\n\n"

    if price_from:
        text += f"💰 От {price_from} ₽\n"

    if visa:
        text += f"📋 {visa}\n"

    return text


def format_confirm(data: dict) -> str:
    """Форматировать подтверждение заявки"""
    lines = ["📋 <b>Проверьте данные:</b>\n"]
    lines.append(f"👤 Имя: {data.get('name', '')}")
    lines.append(f"📞 Телефон: {data.get('phone', '')}")

    country = data.get("country")
    if country:
        lines.append(f"🌍 Страна: {country.get('name', '')}")

    wishes = data.get("wishes")
    if wishes:
        lines.append(f"💭 Пожелания: {wishes}")

    return "\n".join(lines)