from aiogram import types, Dispatcher from aiogram.filters import Command import aiohttp from instences.config import BASE_URL_FASTAPI from keyboard.keyboards import subhist_keyboard,confirm_popup_keyboard,tarif_confirm_keyboard, popup_keyboard, main_keyboard,faq_keyboard, account_keyboard, buy_keyboard,balance_keyboard,guide_keyboard,tarif_Lark_keyboard,tarif_Lark_pro_keyboard,tranhist_keyboard # Инициализируем менеджер базы данных async def call_api(method, endpoint, data=None): async with aiohttp.ClientSession() as session: url = f"{BASE_URL_FASTAPI}{endpoint}" async with session.request(method, url, json=data) as response: if response.status in {200, 201}: return await response.json() return "ERROR" async def popup_command(message: types.Message): """ Обработчик команды для отправки popup-сообщения. """ await message.answer("HAHAHHAHAHAHAHHAHA", reply_markup=popup_keyboard()) async def subhist_command(message: types.Message): """ Обработчик команды для отправки истории подписок. """ await message.answer("subhist", reply_markup=subhist_keyboard()) async def start_command(message: types.Message): """ Обработчик команды /start. """ await message.answer(f"""Приветствуем в рядах птенец {message.from_user.username}🐣 \nСкорее пополни баланс, приобрети подписку и получай доступ к TikTok, YouTube, Discord, Instagram на всех устройствах без ограничений ❕❕❕ \nОзнакомься с руководством по пользованию и выбери подходящий тариф 🦅 \nСледи за акциями, спец-предложениями, а также розыгрышами по ссылке ниже \n👇👇👇 \nhttps://t.me/+0z5xqn3F1m02OTJi \nС любовью ваши пернатые разработчики 🤍🤍🤍""", reply_markup=main_keyboard()) async def start_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="base". """ await callback.message.edit_text( f"""Приветствуем в рядах птенец {callback.from_user.username}🐣 \nСкорее пополни баланс, приобрети подписку и получай доступ к TikTok, YouTube, Discord, Instagram на всех устройствах без ограничений ❕❕❕ \nОзнакомься с руководством по пользованию и выбери подходящий тариф 🦅 \nСледи за акциями, спец-предложениями, а также розыгрышами по ссылке ниже \n👇👇👇 \nhttps://t.me/+0z5xqn3F1m02OTJi \nС любовью ваши пернатые разработчики 🤍🤍🤍""", reply_markup=main_keyboard() ) async def profile_callback_handler(callback: types.CallbackQuery): user_data = await call_api("POST", "/user/create", {"telegram_id": callback.from_user.id}) if not user_data: await callback.message.answer("Произошла ошибка, попробуйте позже.") await callback.answer() return text = f"Ваш профиль:\nID: {user_data['username']}\nБаланс: {user_data['balance']} ₽" await callback.message.edit_text(text, reply_markup=account_keyboard()) await callback.answer() async def balance_callback_handler(callback: types.CallbackQuery): user_data = await call_api("GET", f"/user/{callback.from_user.id}") if not user_data: await callback.message.answer("Вы еще не зарегистрированы.") await callback.answer() return await callback.message.edit_text( f"Ваш баланс: {user_data['balance']} ₽. Выберите сумму для пополнения 🐥", reply_markup=balance_keyboard() ) await callback.answer() async def popup_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="popup". """ user = await call_api("POST", "/user/create", {"telegram_id": callback.from_user.id}) if user == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return if user: await callback.message.edit_text( f"Работает в режиме теста!!!", reply_markup=popup_keyboard() ) else: await callback.message.edit_text("Вы еще не зарегистрированы.") await callback.answer() async def tranhist_callback_handler(callback: types.CallbackQuery): user_data = await call_api("GET", f"/user/{callback.from_user.id}") if not user_data: await callback.message.edit_text("Вы еще не зарегистрированы.") await callback.answer() return transactions = await call_api("GET", f"/user/{user_data['id']}/transactions") if not transactions: await callback.message.edit_text("У вас нет транзакций.", reply_markup=tranhist_keyboard()) await callback.answer() return result = "Ваши транзакции:\n" for count, tran in enumerate(transactions, start=1): result += f"{count}. Сумма: {tran['amount']}, Дата: {tran['created_at']}\n" await callback.message.edit_text(result, reply_markup=tranhist_keyboard()) await callback.answer() async def subhist_callback_handler(callback: types.CallbackQuery): user_data = await call_api("GET", f"/user/{callback.from_user.id}") if not user_data: await callback.message.edit_text("Вы еще не зарегистрированы.") await callback.answer() return subscriptions = await call_api("GET", f"/subscription/{user_data['id']}/last") if not subscriptions: await callback.message.edit_text("У вас нет активных подписок.", reply_markup=account_keyboard()) await callback.answer() return result = "Ваши подписки:\n" for count, sub in enumerate(subscriptions, start=1): result += f"{count}. Тариф: {sub['plan']}, Истекает: {sub['expiry_date']}\n" await callback.message.edit_text(result, reply_markup=account_keyboard()) await callback.answer() async def buy_subscription_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="buy_subscription". """ await callback.message.edit_text( f"Ознакомься с условиями в вкладке \"О тарифах\" и выбери подходящий 🦅", reply_markup=buy_keyboard() ) async def guide_callback_handler(callback:types.CallbackQuery): """ Обработчик callback_query с data="guide". """ await callback.message.edit_text( f"Руководство по использованию продкута что мы высрали;)", reply_markup=guide_keyboard() ) async def subs_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="subs". """ await callback.message.edit_text( f"Подписки птенчик", reply_markup=tarif_Lark_keyboard() ) async def subs_pro_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="subs_pro". """ await callback.message.edit_text( f"Подписки птенчик ПРО", reply_markup=tarif_Lark_pro_keyboard() ) # async def about_tarifs_callback_handler(callback: types.CallbackQuery): # """ # Обработчик callback_query с data="about_tarifs". # """ # await callback.message.edit_text( # f"Бла бла бла, хуйня на хуйне", # reply_markup=about_tarifs_keyboard() # ) async def faq_callback_handler(callback:types.CallbackQuery): """ Обработчик callback_query с data="faq". """ await callback.message.edit_text( f"FAQ YOU", reply_markup=faq_keyboard() ) 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) async def popup_confirm_callback_handler(callback: types.CallbackQuery): """ Обработчик подтверждения пополнения баланса. """ data = callback.data.split(":") popup_info = data[1] result = await call_api("POST", f"/user/{callback.from_user.id}/balance", {"telegram_id": callback.from_user.id, "amount": popup_info}) if result == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return text = f"Вы пополнили свой баланс на {popup_info}. P.S. Мы завтра закрываемся" await callback.message.edit_text(text=text, reply_markup=confirm_popup_keyboard()) async def confirm_callback_handler(callback: types.CallbackQuery): 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}) if result and result.get("message"): await callback.message.edit_text(f"Подписка успешно оформлена!") else: await callback.message.edit_text("Произошла ошибка при оформлении подписки.") await callback.answer() def register_handlers(dp: Dispatcher): """ Регистрация хэндлеров в диспетчере. """ dp.callback_query.register(popup_callback_handler, lambda c: c.data == "popup") dp.callback_query.register(start_callback_handler, lambda c: c.data == "base") dp.callback_query.register(subs_callback_handler, lambda c: c.data == "subs") dp.callback_query.register(subs_pro_callback_handler, lambda c: c.data == "subs_pro") dp.callback_query.register(faq_callback_handler, lambda c: c.data == "faq") dp.callback_query.register(profile_callback_handler, lambda c: c.data == "profile") dp.callback_query.register(tranhist_callback_handler, lambda c: c.data == "tranhist") dp.callback_query.register(buy_subscription_callback_handler, lambda c: c.data == "buy_subscription") dp.message.register(start_command, Command("start")) dp.callback_query.register(balance_callback_handler, lambda c: c.data == "balance") dp.callback_query.register(guide_callback_handler, lambda c: c.data == "guide") # dp.callback_query.register(about_tarifs_callback_handler, lambda c: c.data == "about_tarifs") dp.callback_query.register(lark_tariff_callback_handler, lambda c: c.data.startswith("Lark:")) dp.callback_query.register(confirm_callback_handler, lambda c: c.data.startswith("confirm:")) dp.callback_query.register(popup_confirm_callback_handler, lambda c: c.data.startswith("popup:"))