from aiogram import types, Dispatcher from aiogram.filters import Command from databases.postgresql import DatabaseManager from databases.model import User, Subscription, Transaction, Administrators from databases.db_config import get_postgres_session 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 # Инициализируем менеджер базы данных db_manager = DatabaseManager(get_postgres_session) 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): """ Обработчик callback_query с data="profile". """ user = await db_manager.create_user(telegram_id=callback.from_user.id) if user == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return if user: text = f"""Ваш профиль:\nID: {user.username}\nБаланс: {user.balance}""" await callback.message.edit_text( text, reply_markup=account_keyboard() ) else: await callback.message.edit_text("Вы еще не зарегистрированы.") await callback.answer() async def balance_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="balance". """ user = await db_manager.create_user(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"Ваш баланс: {user.balance} ₽", reply_markup=balance_keyboard() ) else: await callback.message.edit_text("Вы еще не зарегистрированы.") await callback.answer() async def popup_callback_handler(callback: types.CallbackQuery): """ Обработчик callback_query с data="popup". """ user = await db_manager.create_user(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): """ Обработчик callback_query с data="tranhist". """ user = await db_manager.create_user(callback.from_user.id) trans = await db_manager.last_transaction(user.id) if trans == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return if not trans: await callback.message.edit_text( "У вас нет транзакций. Пожалуйста, пополните баланс.", reply_markup=tranhist_keyboard() ) await callback.answer() return result = "Ваши транзакции:\n" for count, tran in enumerate(trans, 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): """ Обработчик callback_query с data="subhist". """ user = await db_manager.create_user(callback.from_user.id) subs = await db_manager.last_subscription(user.id) if subs == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return if subs is None: await callback.message.edit_text( f"Ты хули тут забыл, ты ж не покупаешь нихуя", reply_markup=account_keyboard() ) await callback.answer() return result = "" count = 0 for sub in subs: if count > 0: result += f"Последняя подписка истекает: {sub.expiry_date}\n" count += 1 result += f"{count}. Истекла {sub.expiry_date}" count += 1 if subs: await callback.message.edit_text( result, reply_markup=account_keyboard() ) else: await callback.message.edit_text("Вы еще не зарегистрированы.") 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 db_manager.update_balance(callback.from_user.id,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): """ Обработчик подтверждения покупки тарифа. """ tariff_info = callback.data.split(":")[1].split("_") tariff_name = tariff_info[0] tariff_class = tariff_info[1] tariff_amount = int(tariff_info[2]) sub = await db_manager.buy_sub(callback.from_user.id, f"{tariff_name}_{tariff_class}_{tariff_amount}") if sub == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return elif sub == "INSUFFICIENT_FUNDS": await callback.message.answer( "Произошла ошибка, не достаточно средств на балансе." ) await callback.answer() return add_to_server = await db_manager.add_to_server(callback.from_user.id) if add_to_server == "ERROR": await callback.message.answer( "Произошла ошибка, попробуйте позже или свяжитесь с администрацией." ) await callback.answer() return # Текст подтверждения на основе тарифа months_text = f"{tariff_amount} месяцев" if tariff_amount > 1 else f"{tariff_amount} месяц" text = f"Вы успешно оформили тариф {tariff_name} на {months_text}. Спасибо за покупку!" await callback.message.edit_text(text=text) 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:"))