Много переделал
This commit is contained in:
138
handlers/subscriptions.py
Normal file
138
handlers/subscriptions.py
Normal file
@@ -0,0 +1,138 @@
|
||||
from aiogram import Router, types
|
||||
import logging
|
||||
from instences.config import BASE_URL_FASTAPI
|
||||
import aiohttp
|
||||
from aiogram.enums.parse_mode import ParseMode
|
||||
from aiogram.filters import Command
|
||||
from keyboard.keyboards import tarif_Lark_pro_keyboard, tarif_Lark_keyboard, tarif_confirm_keyboard,buy_keyboard
|
||||
|
||||
router = Router()
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
async def call_api(method, endpoint, data=None):
|
||||
"""
|
||||
Выполняет HTTP-запрос к FastAPI.
|
||||
"""
|
||||
url = f"{BASE_URL_FASTAPI}{endpoint}"
|
||||
logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
|
||||
|
||||
try:
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.request(method, url, json=data) as response:
|
||||
logger.info(f"Получен ответ от {url}: статус {response.status}")
|
||||
|
||||
if response.status in {200, 201}:
|
||||
result = await response.json()
|
||||
logger.debug(f"Ответ JSON: {result}")
|
||||
return result
|
||||
if response.status in {404,400}:
|
||||
logger.debug(f"Код {response.status}, возвращаю ничего")
|
||||
return await response.json()
|
||||
logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
|
||||
return "ERROR"
|
||||
except Exception as e:
|
||||
logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
|
||||
return "ERROR"
|
||||
|
||||
@router.message(Command("subscriptions"))
|
||||
async def supp(message: types.Message):
|
||||
"""
|
||||
Меню сапп системы
|
||||
"""
|
||||
text = ""
|
||||
result = await call_api("GET", f"/uri?telegram_id={message.from_user.id}")
|
||||
uri = result.get('detail',"Error")
|
||||
if uri == "SUB_ERROR":
|
||||
text = f"Вы ещё не приобрели подписки!!"
|
||||
elif "vless" in uri:
|
||||
text = f"Ваша подписка:```{uri}```"
|
||||
else:
|
||||
text = "Произошла ошибка при получении URI"
|
||||
await message.answer(
|
||||
text,
|
||||
parse_mode=ParseMode.MARKDOWN_V2
|
||||
)
|
||||
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "buy_subscription")
|
||||
async def buy_subscription_callback_handler(callback: types.CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query с data="buy_subscription".
|
||||
"""
|
||||
await callback.message.edit_text(
|
||||
f"Ознакомься с условиями в вкладке \"О тарифах\" и выбери подходящий 🦅",
|
||||
reply_markup=buy_keyboard()
|
||||
)
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "subs")
|
||||
async def subs_callback_handler(callback: types.CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query с data="subs".
|
||||
"""
|
||||
await callback.message.edit_text(
|
||||
"Подписки птенчик",
|
||||
reply_markup=tarif_Lark_keyboard()
|
||||
)
|
||||
|
||||
@router.callback_query(lambda callback: callback.data == "subs_pro")
|
||||
async def subs_pro_callback_handler(callback: types.CallbackQuery):
|
||||
"""
|
||||
Обработчик callback_query с data="subs_pro".
|
||||
"""
|
||||
await callback.message.edit_text(
|
||||
"Подписки птенчик ПРО",
|
||||
reply_markup=tarif_Lark_pro_keyboard()
|
||||
)
|
||||
|
||||
@router.callback_query(lambda callback: callback.data.startswith("Lark:"))
|
||||
async def lark_tariff_callback_handler(callback: types.CallbackQuery):
|
||||
"""
|
||||
Обработчик для выбора тарифа Lark.
|
||||
"""
|
||||
data = callback.data.split(":")
|
||||
tariff_name = data[0]
|
||||
tariff_class = data[1]
|
||||
tariff_time = int(data[2])
|
||||
|
||||
# Определение окончания для месяцев
|
||||
if tariff_time == 1:
|
||||
months = f"{tariff_time} месяц"
|
||||
elif 2 <= tariff_time <= 4:
|
||||
months = f"{tariff_time} месяца"
|
||||
else:
|
||||
months = f"{tariff_time} месяцев"
|
||||
|
||||
text = f"Тариф {tariff_name} на {months}. Продолжите покупку..."
|
||||
|
||||
# Рендеринг клавиатуры
|
||||
keyboard = tarif_confirm_keyboard(tariff_name, tariff_time, tariff_class)
|
||||
await callback.message.edit_text(text=text, reply_markup=keyboard)
|
||||
|
||||
@router.callback_query(lambda callback: callback.data.startswith("confirm:"))
|
||||
async def confirm_callback_handler(callback: types.CallbackQuery):
|
||||
"""
|
||||
Обработчик подтверждения подписки.
|
||||
"""
|
||||
try:
|
||||
data = callback.data.split(":")[1]
|
||||
tariff_info = data.split("_")
|
||||
plan_id = f"{tariff_info[0]}_{tariff_info[1]}_{tariff_info[2]}"
|
||||
result = await call_api("POST", "/subscription/buy", {"telegram_id": callback.from_user.id, "plan_id": plan_id})
|
||||
detail = result.get("detail", {})
|
||||
|
||||
if detail == "ERROR":
|
||||
await callback.message.edit_text("Произошла ошибка при оформлении подписки.")
|
||||
elif detail == "INSUFFICIENT_FUNDS":
|
||||
await callback.message.edit_text("Денег на вашем балансе не достаточно.")
|
||||
elif detail == "TARIFF_NOT_FOUND":
|
||||
await callback.message.edit_text("Ваш тариф не найден.")
|
||||
elif detail == "ACTIVE_SUBSCRIPTION_EXISTS":
|
||||
await callback.message.edit_text("Вы уже имеете активную подписку.")
|
||||
else:
|
||||
uri = result.get("message", {})
|
||||
await callback.message.edit_text(f"Подписка успешно оформлена!\nВаш конфиг для подключения: ```{uri}```")
|
||||
except Exception as e:
|
||||
logger.exception(f"Ошибка при обработке подтверждения подписки: {e}")
|
||||
await callback.message.edit_text("Произошла ошибка при оформлении подписки.")
|
||||
finally:
|
||||
await callback.answer()
|
||||
Reference in New Issue
Block a user