19 Commits

Author SHA1 Message Date
d3e47a662f Временные изменения 2025-12-05 18:45:25 +03:00
root
51645096d3 Мелкое изменение что бы работали подписк 2025-12-01 16:36:21 +03:00
unknown
00edfdbd19 вова помоги (merge front-misha-ui) 2025-12-01 12:57:44 +03:00
unknown
94df5ee9a5 вова помоги 2025-12-01 12:50:24 +03:00
root
0d9a86ee54 Для криптобота так же пофиксил 2025-11-30 18:13:00 +03:00
root
12c60b884a Ошибка в способе передачи данных, нужен был query 2025-11-30 16:57:08 +03:00
root
6276490ee1 Изменил call_api добавив тип данных и преобразую во float amount 2025-11-30 16:51:47 +03:00
root
3e976d9960 Исправил ошибку с id и изменил call_api что бы он работал с billing 2025-11-30 16:41:40 +03:00
root
77ca27a09c Получение ссылки на оплату юкассы и криптобота 2025-11-30 16:33:04 +03:00
root
76daf3244f Временные изменения что бы тестировать бекенд, в дальнейшем должен быть заменён на то что делает Михаил 2025-11-27 20:57:11 +03:00
unknown
d1417653c3 Фикс keyboards: убрал дубли, добавил account/popup/payment_methods 2025-11-25 22:40:42 +03:00
unknown
cbf113f7bb Реферальная программа: кнопка и экран 2025-11-25 20:20:38 +03:00
unknown
a572c90194 UI: карточки подписок и выбор тарифов 2025-11-25 19:18:22 +03:00
unknown
f042cc2f9e UI: тарифы Basic/Pro/Family и меню оплат 2025-11-25 15:40:33 +03:00
unknown
779f13b98e Обновлён профиль и логика выбора способа оплаты 2025-11-25 07:55:27 +03:00
unknown
c7ded9e499 Обновлён список сумм пополнения 2025-11-25 07:32:55 +03:00
unknown
cda94583ed Сделал две кнопки, как у меня 2025-11-25 00:11:33 +03:00
unknown
79307b9791 Обновлён текст профиля и статус подписки 2025-11-24 23:48:29 +03:00
unknown
c1021ae8f1 Обновил логику бота / фиксы 2025-11-23 16:50:19 +03:00
7 changed files with 1756 additions and 486 deletions

View File

@@ -1,7 +1,16 @@
from .start import router as start_router from .start import router as start_router
from .profile import router as profile_router from .profile import router as profile_router
from .subscriptions import router as subscriptions_router from .subscriptions import router as subscriptions_router
from .support import router as support_router from .support import router as support_router
from .referrals import router as referrals_router
# Экспортируем все маршрутизаторы
routers = [start_router,profile_router,subscriptions_router,support_router] # Список всех роутеров бота
routers = [
start_router,
profile_router,
subscriptions_router,
support_router,
referrals_router,
]
__all__ = ["routers"]

View File

@@ -1,170 +1,534 @@
from aiogram import Router, types # Профиль. последнее изменениеы
from aiogram.types import CallbackQuery
import logging from aiogram import Router, types, F
from datetime import datetime from aiogram.types import (
from aiogram.enums.parse_mode import ParseMode CallbackQuery,
import locale LabeledPrice,
from instences.config import BASE_URL_FASTAPI Message,
import aiohttp PreCheckoutQuery,
from keyboard.keyboards import account_keyboard, popup_keyboard, tranhist_keyboard, confirm_popup_keyboard, guide_keyboard, balance_keyboard )
import logging
locale.setlocale(locale.LC_TIME, "ru_RU.UTF-8") from datetime import datetime
from aiogram.enums.parse_mode import ParseMode
router = Router() import locale
logger = logging.getLogger(__name__) from instences.config import BASE_URL_FASTAPI
import aiohttp
async def call_api(method, endpoint, data=None): from keyboard.keyboards import (
""" account_keyboard,
Выполняет HTTP-запрос к FastAPI. popup_keyboard,
""" tranhist_keyboard,
url = f"{BASE_URL_FASTAPI}{endpoint}" confirm_popup_keyboard,
logger.info(f"Инициализация запроса: {method} {url} с данными {data}") guide_keyboard,
balance_keyboard,
try: payment_methods_keyboard,
async with aiohttp.ClientSession() as session: )
async with session.request(method, url, json=data) as response:
logger.info(f"Получен ответ от {url}: статус {response.status}") locale.setlocale(locale.LC_TIME, "ru_RU.UTF-8")
if response.status in {200, 201}: router = Router()
result = await response.json() logger = logging.getLogger(__name__)
logger.debug(f"Ответ JSON: {result}")
return result
if response.status == 404: async def call_api(method, endpoint, data=None, base_url=BASE_URL_FASTAPI):
logger.debug(f"Код {response.status}, возвращаю ничего") """
return None Выполняет HTTP-запрос к FastAPI.
logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}") """
return "ERROR" url = f"{base_url}{endpoint}"
except Exception as e: logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
return "ERROR" try:
async with aiohttp.ClientSession() as session:
@router.callback_query(lambda callback: callback.data == "profile") async with session.request(
async def profile_callback_handler(callback: CallbackQuery): method,
""" url,
Обработчик callback_query для профиля. json=data,
""" headers={"Content-Type": "application/json"}
try: ) as response:
user_data = await call_api("GET", f"/user/{callback.from_user.id}") logger.info(
if not user_data: f"Получен ответ от {url}: статус {response.status}"
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.") )
await callback.answer()
return if response.status in {200, 201}:
result = await response.json()
sub_data = await call_api("GET", f"/subscription/{user_data['id']}/last") logger.debug(f"Ответ JSON: {result}")
if sub_data == "ERROR" or not isinstance(sub_data, dict): return result
sub_data = None if response.status == 404:
logger.debug(f"Код {response.status}, возвращаю ничего")
balance_text = f"Баланс: {user_data['balance']}" return None
logger.error(
if not sub_data: f"Ошибка в запросе: статус {response.status}, причина {response.reason}"
text = f"Профиль {callback.from_user.username}\n{balance_text}\nПополните баланс и приобретите подписку, чтобы получить активный статус (🐣,🦅)" )
else: return "ERROR"
expiry_date = sub_data.get("expiry_date") except Exception as e:
formatted_date = datetime.fromisoformat(expiry_date).strftime("%d %B %Y г.") if expiry_date else None logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
is_expired = datetime.fromisoformat(expiry_date) < datetime.now() if expiry_date else True return "ERROR"
status_icon = "✖️" if is_expired else "☑️"
profile_status = "🦅" if "Pro" in sub_data.get("plan", "") else "🐣"
sub_text = ( @router.callback_query(lambda callback: callback.data == "profile")
f"Ваша подписка действует до {formatted_date} {status_icon}" async def profile_callback_handler(callback: CallbackQuery):
if not is_expired else f"Статус подписки: {status_icon}" """
) Профиль пользователя.
text = f"Профиль {callback.from_user.username} {profile_status}:\n{sub_text}\n{balance_text}" Логика работы с API сохранена, изменён только текст/визуал.
"""
await callback.message.edit_text(text, reply_markup=account_keyboard()) try:
except Exception as e: user_data = await call_api("GET", f"/user/{callback.from_user.id}")
logger.exception(f"Ошибка в обработчике профиля: {e}") if not user_data:
await callback.message.answer("Произошла ошибка. Попробуйте позже.") await callback.message.answer(
finally: "Произошла ошибка, попробуйте позже или свяжитесь с администрацией."
await callback.answer() )
await callback.answer()
@router.callback_query(lambda callback: callback.data == "balance") return
async def balance_callback_handler(callback: CallbackQuery):
""" sub_data = await call_api(
Обработчик callback_query для баланса. "GET", f"/subscription/{user_data['telegram_id']}/last"
""" )
user_data = await call_api("GET", f"/user/{callback.from_user.id}") if sub_data == "ERROR" or not isinstance(sub_data, dict):
if not user_data: sub_data = None
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
await callback.answer() username = callback.from_user.username or "-"
return balance = user_data.get("balance", 0)
await callback.message.edit_text( # Статус подписки: Активна / Нет активных
f"Ваш баланс: {user_data['balance']} ₽. Выберите сумму для пополнения 🐥", sub_status = "⚫ Нет активных"
reply_markup=balance_keyboard()
) if sub_data:
await callback.answer() expiry_date = sub_data.get("end_date")
if expiry_date:
@router.callback_query(lambda callback: callback.data == "popup") try:
async def popup_callback_handler(callback: CallbackQuery): is_expired = datetime.fromisoformat(
""" expiry_date) < datetime.now()
Обработчик callback_query для пополнения. except ValueError:
""" is_expired = True
user = await call_api("GET", f"/user/{callback.from_user.id}") else:
if not user: is_expired = True
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.")
await callback.answer() if not is_expired:
return sub_status = "🟢 Активна"
await callback.message.edit_text("Работает в режиме теста!!!", reply_markup=popup_keyboard()) text = (
await callback.answer() "🥚 <b>Профиль</b>\n\n"
f"Пользователь: @{username}\n"
@router.callback_query(lambda callback: callback.data == "tranhist") f"Баланс: {balance}\n"
async def tranhist_callback_handler(callback: CallbackQuery): f"Статус подписки: {sub_status}\n\n"
""" "Выберите действие:"
Обработчик callback_query для истории транзакций. )
"""
user_data = await call_api("GET", f"/user/{callback.from_user.id}") await callback.message.edit_text(
if not user_data: text,
await callback.message.edit_text("Вы еще не зарегистрированы.") parse_mode=ParseMode.HTML,
await callback.answer() reply_markup=account_keyboard(),
return )
except Exception as e:
try: logger.exception(f"Ошибка в обработчике профиля: {e}")
transactions = await call_api("GET", f"/user/{user_data['id']}/transactions") await callback.message.answer("Произошла ошибка. Попробуйте позже.")
if not transactions: finally:
await callback.message.edit_text("У вас нет транзакций.", reply_markup=tranhist_keyboard()) await callback.answer()
await callback.answer()
return
@router.callback_query(lambda callback: callback.data == "balance")
result = "Ваши транзакции:```\n" async def balance_callback_handler(callback: CallbackQuery):
for count, tran in enumerate(transactions, start=1): """
dt = datetime.fromisoformat(tran['created_at']).strftime("%d.%m.%Y %H:%M:%S") При нажатии «Пополнить баланс» показываем выбор суммы пополнения.
result += f"{count}. Сумма: {tran['amount']}, Дата: {dt}\n" """
if len(result) > 4000: await callback.message.edit_text(
result += "...\nСлишком много транзакций для отображения." "💳 Выберите сумму пополнения:",
break reply_markup=popup_keyboard(),
result += "```" )
await callback.message.edit_text(result,parse_mode=ParseMode.MARKDOWN_V2, reply_markup=tranhist_keyboard()) await callback.answer()
except Exception as e:
logger.error(f"Ошибка обработки транзакций: {e}")
await callback.message.edit_text("Произошла ошибка. Попробуйте позже.") @router.callback_query(lambda callback: callback.data == "popup")
finally: async def popup_callback_handler(callback: CallbackQuery):
await callback.answer() """
Обработчик callback_query для выбора суммы пополнения.
@router.callback_query(lambda callback: callback.data.startswith("popup:")) """
async def popup_confirm_callback_handler(callback: CallbackQuery): user = await call_api("GET", f"/user/{callback.from_user.id}")
""" if not user:
Обработчик подтверждения пополнения баланса. await callback.message.answer(
""" "Произошла ошибка, попробуйте позже или свяжитесь с администрацией."
data = callback.data.split(":") )
popup_info = data[1] await callback.answer()
result = await call_api("POST", f"/user/{callback.from_user.id}/balance/{float(popup_info)}") return
if result == "ERROR":
await callback.message.answer("Произошла ошибка, попробуйте позже или свяжитесь с администрацией.") await callback.message.edit_text(
await callback.answer() "Выбери сумму для пополнения баланса.",
return reply_markup=popup_keyboard(),
)
text = f"Вы пополнили свой баланс на {popup_info} ₽. Спасибо!" await callback.answer()
await callback.message.edit_text(text=text, reply_markup=confirm_popup_keyboard())
await callback.answer()
@router.callback_query(lambda callback: callback.data == "tranhist")
@router.callback_query(lambda callback: callback.data == "guide") async def tranhist_callback_handler(callback: CallbackQuery):
async def guide_callback_handler(callback: CallbackQuery): """
""" Обработчик callback_query для истории транзакций.
Обработчик callback_query для руководства. (Логику и формат Markdown_V2 не трогаем, чтобы не поймать новые баги)
""" """
await callback.message.edit_text( user_data = await call_api("GET", f"/user/{callback.from_user.id}")
"Руководство по использованию продукта:", if not user_data:
reply_markup=guide_keyboard() await callback.message.edit_text("Вы еще не зарегистрированы.")
) await callback.answer()
await callback.answer() return
try:
transactions = await call_api(
"GET", f"/user/{user_data['telegram_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):
dt = datetime.fromisoformat(tran["created_at"]).strftime(
"%d.%m.%Y %H:%M:%S"
)
result += f"{count}. Сумма: {tran['amount']}, Дата: {dt}\n"
if len(result) > 4000:
result += "...\nСлишком много транзакций для отображения."
break
result += "```"
await callback.message.edit_text(
result,
parse_mode=ParseMode.MARKDOWN_V2,
reply_markup=tranhist_keyboard(),
)
except Exception as e:
logger.error(f"Ошибка обработки транзакций: {e}")
await callback.message.edit_text(
"Произошла ошибка. Попробуйте позже."
)
finally:
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("popup:"))
async def popup_confirm_callback_handler(callback: CallbackQuery):
"""
После выбора суммы показываем варианты оплаты.
Разрешены только суммы 200, 300, 600, 1000 ₽.
"""
try:
_, amount_raw = callback.data.split(":", maxsplit=1)
amount = int(float(amount_raw))
except Exception:
await callback.message.answer("Некорректная сумма пополнения.")
await callback.answer()
return
if amount not in {200, 300, 600, 1000}:
await callback.message.answer(
"Эта сумма пополнения недоступна. Выбери вариант от 200 до 1000 ₽."
)
await callback.answer()
return
text = (
f"💰 Сумма пополнения: {amount}\n\n"
"Выбери способ оплаты:"
)
await callback.message.edit_text(
text=text,
reply_markup=payment_methods_keyboard(amount),
)
await callback.answer()
# ===== Telegram Stars =====
@router.callback_query(lambda callback: callback.data.startswith("method_stars_"))
async def method_stars_handler(callback: CallbackQuery):
"""
Оплата через Telegram Stars.
Формируем invoice прямо из бота, без отдельного биллинга.
"""
try:
amount_str = callback.data.split("_")[-1]
amount_rub = int(float(amount_str))
except Exception:
await callback.message.answer("Некорректная сумма для оплаты.")
await callback.answer()
return
payload = f"stars_topup:{callback.from_user.id}:{amount_rub}"
stars_amount = amount_rub
prices = [
LabeledPrice(
label=f"Пополнение баланса на {amount_rub}",
amount=stars_amount,
)
]
try:
await callback.message.answer_invoice(
title="Пополнение баланса Lark VPN",
description=(
f"Пополнение баланса на {amount_rub} ₽ через Telegram Stars.\n\n"
"После успешной оплаты баланс будет зачислен автоматически."
),
payload=payload,
provider_token="", # для Stars провайдер пустой
currency="XTR",
prices=prices,
)
await callback.answer()
except Exception as e:
logger.exception(f"Ошибка при отправке invoice Telegram Stars: {e}")
await callback.message.answer(
"Не удалось создать счёт в Telegram Stars. Попробуй позже или выбери другой способ оплаты."
)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("method_ykassa_"))
async def method_ykassa_handler(callback: CallbackQuery):
"""
Обработчик оплаты через YooKassa.
"""
amount = callback.data.split("_")[-1]
try:
await callback.answer()
except Exception:
pass
endpoint = (
f"/billing/payments/init?"
f"user_id={callback.from_user.id}&amount={float(amount)}&provider=yookassa"
)
logger.info(f"Отправка запроса на инициализацию платежа: {endpoint}")
result = await call_api("POST", endpoint, None, "http://billing:8000")
if result == "ERROR" or not isinstance(result, dict):
await callback.message.edit_text(
"❌ Произошла ошибка при создании платежа. Попробуйте позже."
)
return
if not result.get("success", False):
error_msg = (
result.get("error")
or result.get("detail")
or "Неизвестная ошибка"
)
await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
return
payment_url = result.get("confirmation_url", "#")
payment_id = result.get("payment_id", "")
await callback.message.edit_text(
f"💵 <b>Оплата через YooKassa</b>\n\n"
f"💰 Сумма: <code>{amount}</code> руб\n"
f"📋 ID платежа: <code>{payment_id}</code>\n\n"
f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
parse_mode=ParseMode.HTML,
disable_web_page_preview=True,
reply_markup=types.InlineKeyboardMarkup(
inline_keyboard=[[
types.InlineKeyboardButton(
text="🔄 Проверить оплату",
callback_data=f"check_payment:{payment_id}",
)
]]
),
)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("method_crypto_"))
async def method_crypto_handler(callback: CallbackQuery):
"""
Оплата через CryptoBot.
"""
amount = callback.data.split("_")[-1]
try:
await callback.answer()
except Exception:
pass
endpoint = (
f"/billing/payments/init?"
f"user_id={callback.from_user.id}&amount={float(amount)}&provider=cryptobot"
)
logger.info(f"Отправка запроса на инициализацию платежа: {endpoint}")
result = await call_api("POST", endpoint, None, "http://billing:8000")
if result == "ERROR" or not isinstance(result, dict):
await callback.message.edit_text(
"❌ Произошла ошибка при создании платежа. Попробуйте позже."
)
return
if not result.get("success", False):
error_msg = (
result.get("error")
or result.get("detail")
or "Неизвестная ошибка"
)
await callback.message.edit_text(f"❌ Ошибка: {error_msg}")
return
payment_url = result.get("confirmation_url", "#")
payment_id = result.get("payment_id", "")
await callback.message.edit_text(
f"💵 <b>Оплата через Сryptobot</b>\n\n"
f"💰 Сумма: <code>{amount}</code> руб\n"
f"📋 ID платежа: <code>{payment_id}</code>\n\n"
f"➡️ <a href='{payment_url}'>Перейти к оплате</a>\n\n"
f"<i>После оплаты нажмите кнопку 'Проверить оплату'</i>",
parse_mode=ParseMode.HTML,
disable_web_page_preview=True,
reply_markup=types.InlineKeyboardMarkup(
inline_keyboard=[[
types.InlineKeyboardButton(
text="🔄 Проверить оплату",
callback_data=f"check_payment:{payment_id}",
)
]]
),
)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("method_sbp_"))
async def method_sbp_handler(callback: types.CallbackQuery):
try:
amount = int(callback.data.split("_")[2])
except Exception:
await callback.message.answer("Ошибка суммы платежа.")
await callback.answer()
return
endpoint = (
f"/billing/payments/init?"
f"user_id={callback.from_user.id}&amount={float(amount)}&provider=sbp"
)
result = await call_api("POST", endpoint,None ,"http://billing:8000")
if result == "ERROR" or not isinstance(result, dict):
await callback.message.answer("Ошибка при инициализации платежа.")
await callback.answer()
return
url = result.get("confirmation_url")
if not url:
await callback.message.answer("СБП временно недоступно.")
await callback.answer()
return
await callback.message.answer(
f"🏦 Платёж через СБП:\nПерейди по ссылке для оплаты:\n{url}"
)
await callback.answer()
@router.callback_query(lambda callback: callback.data == "guide")
async def guide_callback_handler(callback: CallbackQuery):
"""
Обработчик callback_query для руководства.
"""
await callback.message.edit_text(
"Выбери платформу, для которой нужно руководство по подключению:",
reply_markup=guide_keyboard(),
)
await callback.answer()
# ===== Служебные хендлеры для платежей Telegram Stars =====
@router.pre_checkout_query()
async def pre_checkout_query_handler(pre_checkout_query: PreCheckoutQuery):
"""
Обязательный шаг для Telegram Payments:
подтверждаем pre_checkout_query, иначе платёж не пройдёт.
"""
try:
await pre_checkout_query.bot.answer_pre_checkout_query(
pre_checkout_query.id,
ok=True,
)
except Exception as e:
logger.exception(f"Ошибка при answer_pre_checkout_query: {e}")
async def _process_stars_topup(message: Message):
"""
Логика зачисления средств после успешной оплаты Stars.
"""
sp = message.successful_payment
if not sp:
return
payload = sp.invoice_payload or ""
parts = payload.split(":")
if len(parts) != 3 or parts[0] != "stars_topup":
logger.info(
"successful_payment не относится к пополнению баланса Stars.")
return
_, telegram_id_str, amount_str = parts
try:
amount_rub = int(amount_str)
except ValueError:
# На всякий случай fallback к total_amount
amount_rub = sp.total_amount
data = {
"telegram_id": telegram_id_str,
"amount": amount_rub,
"currency": sp.currency,
"provider": "telegram_stars",
"telegram_payment_charge_id": sp.telegram_payment_charge_id,
}
logger.info(
f"Обработка успешного платежа Telegram Stars: "
f"user={telegram_id_str}, amount={amount_rub}, currency={sp.currency}"
)
result = await call_api("POST", "/user/deposit", data)
if result == "ERROR" or result is None:
await message.answer(
"⭐ Оплата через Telegram Stars прошла, но не удалось автоматически обновить баланс.\n"
"Если баланс не изменился — напиши, пожалуйста, в поддержку и укажи время платежа."
)
return
await message.answer(
f"⭐ Оплата через Telegram Stars успешно проведена.\n"
f"На твой баланс зачислено {amount_rub} ₽."
)
@router.message(F.successful_payment)
async def successful_payment_handler(message: Message):
"""
Глобальный хендлер успешных платежей.
Сейчас используем только для пополнения баланса через Stars.
"""
try:
await _process_stars_topup(message)
except Exception as e:
logger.exception(f"Ошибка при обработке successful_payment: {e}")
await message.answer(
"Оплата прошла, но произошла ошибка при обработке. "
"Если баланс не изменился — напиши, пожалуйста, в поддержку."
)

123
handlers/referrals.py Normal file
View File

@@ -0,0 +1,123 @@
from aiogram import Router, types
from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode
import logging
import aiohttp
from instences.config import BASE_URL_FASTAPI
router = Router()
logger = logging.getLogger(__name__)
async def call_api(method: str, endpoint: str, data=None, base_url: str = BASE_URL_FASTAPI):
"""
Универсальный HTTP-запрос к FastAPI для рефералок.
Ожидаем:
GET /user/{telegram_id}/referrals -> {
"invited_count": int
}
"""
url = f"{base_url}{endpoint}"
logger.info(f"[referrals] Запрос: {method} {url} с данными {data}")
try:
async with aiohttp.ClientSession() as session:
async with session.request(
method,
url,
json=data,
headers={"Content-Type": "application/json"},
) as response:
logger.info(
f"[referrals] Ответ от {url}: статус {response.status}"
)
if response.status in {200, 201}:
result = await response.json()
logger.debug(f"[referrals] Ответ JSON: {result}")
return result
if response.status == 404:
logger.debug("[referrals] 404, возвращаю None")
return None
logger.error(
f"[referrals] Ошибка в запросе: статус {response.status}, "
f"причина {response.reason}"
)
return "ERROR"
except Exception as e:
logger.exception(f"[referrals] Исключение при запросе к {url}: {e}")
return "ERROR"
async def _build_referral_text(bot, user_id: int) -> str:
"""
Текст реферальной программы + количество приглашённых.
"""
me = await bot.get_me()
bot_username = me.username or "LarkVPN_bot"
link = f"https://t.me/{bot_username}?start=ref_{user_id}"
text = (
"👥 <b>Реферальная программа</b>\n\n"
"Зови друзей в Lark VPN и получай бонусы на баланс.\n\n"
f"🔗 Твоя ссылка:\n<code>{link}</code>\n\n"
)
invited_count = 0
stats = await call_api("GET", f"/user/{user_id}/referrals")
if isinstance(stats, dict):
raw = stats.get("invited_count")
try:
invited_count = int(raw)
except (TypeError, ValueError):
invited_count = 0
elif stats == "ERROR":
logger.warning(
f"[referrals] Ошибка при получении статистики для user_id={user_id}"
)
text += f"👤 Приглашено: {invited_count}\n\n"
text += "Бонусы начисляются, когда приглашённые пополняют баланс."
return text
@router.message(Command("referrals"))
async def referrals_command(message: types.Message):
"""
Команда /referrals — показывает текст реферальной программы.
"""
logger.info(f"[referrals] Команда /referrals от {message.from_user.id}")
try:
text = await _build_referral_text(message.bot, message.from_user.id)
await message.answer(text, parse_mode=ParseMode.HTML)
except Exception as e:
logger.exception(f"[referrals] Ошибка в обработчике /referrals: {e}")
await message.answer("Произошла ошибка. Попробуй позже.")
@router.callback_query(lambda callback: callback.data == "referral")
async def referrals_callback(callback: types.CallbackQuery):
"""
Кнопка «Реферальная программа» (если где-то есть).
"""
try:
text = await _build_referral_text(
callback.message.bot,
callback.from_user.id,
)
await callback.message.edit_text(
text,
parse_mode=ParseMode.HTML,
)
except Exception as e:
logger.exception(
f"[referrals] Ошибка в обработчике callback 'referral': {e}")
await callback.message.answer("Произошла ошибка. Попробуй позже.")
finally:
await callback.answer()

View File

@@ -1,95 +1,224 @@
from aiogram import Router, types from aiogram import Router, types
from aiogram.filters import Command from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery from aiogram.types import Message, CallbackQuery
from aiogram.enums.parse_mode import ParseMode
import logging import logging
from instences.config import BASE_URL_FASTAPI
import aiohttp import aiohttp
from instences.config import BASE_URL_FASTAPI
from keyboard.keyboards import main_keyboard from keyboard.keyboards import main_keyboard
from .referrals import _build_referral_text
router = Router() router = Router()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
async def call_api(method, endpoint, data=None):
async def call_api(method: str, endpoint: str, data=None):
""" """
Выполняет HTTP-запрос к FastAPI. Выполняет HTTP-запрос к FastAPI.
:param method: HTTP метод (GET, POST, и т.д.) Возвращает:
:param endpoint: конечная точка API - dict при 200/201
:param data: тело запроса (если необходимо) - None при 404
:return: JSON-ответ или "ERROR" при неуспехе - "ERROR" при остальных ошибках
""" """
url = f"{BASE_URL_FASTAPI}{endpoint}" url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"Инициализация запроса: {method} {url} с данными {data}") logger.info(f"[start] Запрос: {method} {url} с данными {data}")
try: try:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.request(method, url, json=data) as response: async with session.request(method, url, json=data) as response:
logger.info(f"Получен ответ от {url}: статус {response.status}") logger.info(
f"[start] Ответ от {url}: статус {response.status}"
)
if response.status in {200, 201}: if response.status in {200, 201}:
result = await response.json() result = await response.json()
logger.debug(f"Ответ JSON: {result}") logger.debug(f"[start] Ответ JSON: {result}")
return result return result
if response.status == 404: if response.status == 404:
logger.debug(f"Код {response.status}, возвращаю ничего") logger.debug("[start] Получен 404, возвращаю None")
return None return None
logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
logger.error(
f"[start] Ошибка в запросе: статус {response.status}, "
f"причина {response.reason}"
)
return "ERROR" return "ERROR"
except Exception as e: except Exception as e:
logger.exception(f"Исключение при выполнении запроса к {url}: {e}") logger.exception(f"[start] Исключение при запросе к {url}: {e}")
return "ERROR" return "ERROR"
def _welcome_text(username: str | None) -> str:
"""
Текст приветствия в /start и в главном меню.
"""
return "🥚 Lark Security\n\nВыберите действие из меню ниже."
def _parse_referrer_id(message: Message) -> int | None:
"""
Достаём ref_<telegram_id> из /start.
Примеры:
/start
/start ref_123456789
"""
text = message.text or ""
parts = text.split(maxsplit=1)
if len(parts) < 2:
return None
arg = parts[1].strip()
if not arg.startswith("ref_"):
return None
raw_id = arg[4:]
if not raw_id.isdigit():
return None
try:
return int(raw_id)
except ValueError:
return None
@router.message(Command("start")) @router.message(Command("start"))
async def start_command(message: Message): async def start_command(message: Message):
""" """
Обработчик команды /start. /start c обработкой реферального параметра.
""" """
logger.info(f"Получена команда /start от пользователя: {message.from_user.id} ({message.from_user.username})") user_id = message.from_user.id
username = message.from_user.username
referrer_id = _parse_referrer_id(message)
logger.info(
f"[start] Команда /start от {user_id} (@{username}), "
f"text={message.text!r}, referrer_id={referrer_id}"
)
try: try:
user_data = await call_api("GET", f"/user/{message.from_user.id}") # 1. Проверяем, есть ли пользователь в БД
if not user_data: existing = await call_api("GET", f"/user/{user_id}")
logger.debug("Пользователь не найден в базе, создаем новую запись.") user_exists = existing not in (None, "ERROR")
await call_api("POST", "/user/create", {"telegram_id": message.from_user.id})
logger.debug("Отправка приветственного сообщения пользователю.") # 2. Если пользователя нет — создаём
if not user_exists:
logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
create_result = await call_api(
"POST",
"/user/create",
{"telegram_id": user_id},
)
if create_result == "ERROR":
logger.error(
f"[start] Не удалось создать пользователя {user_id} в БД"
)
# 3. Обработка рефералки, если параметр есть
if referrer_id is not None:
# 3.1. Самореферал
if referrer_id == user_id:
logger.info(
f"[start] Пользователь {user_id} попытался зайти "
f"по своей реферальной ссылке."
)
await message.answer(
"Нельзя переходить по своей же реферальной ссылке."
)
# 3.2. Пользователь уже зарегистрирован в боте
elif user_exists:
logger.info(
f"[start] Пользователь {user_id} уже есть в БД, "
f"реферальная ссылка {referrer_id} не сработает."
)
await message.answer(
"Вы уже зарегистрированы в боте, "
"реферальная ссылка не сработает."
)
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
else:
payload = {
"referrer_id": referrer_id,
"telegram_id": user_id,
}
logger.info(
f"[start] Фиксирую реферала в бекенде: {payload}"
)
result = await call_api(
"POST",
"/user/referrals/track",
payload,
)
if result == "ERROR":
logger.error(
f"[start] Ошибка при фиксации реферала: {payload}"
)
await message.answer(
"Вы вошли по реферальной ссылке."
)
# 4. В любом случае показываем главное меню
await message.answer( await message.answer(
f"""Приветствуем в рядах птенец {message.from_user.username}🐣 _welcome_text(username),
\nСкорее пополни баланс, приобрети подписку и получай доступ к TikTok, YouTube, Discord, Instagram на всех устройствах без ограничений ❕❕❕ reply_markup=main_keyboard(),
\nОзнакомься с руководством по пользованию и выбери подходящий тариф 🦅
\nСледи за акциями, спец-предложениями, а также розыгрышами по ссылке ниже
\n👇👇👇
\nhttps://t.me/+0z5xqn3F1m02OTJi
\nС любовью ваши пернатые разработчики 🤍🤍🤍""",
reply_markup=main_keyboard()
) )
logger.info("Приветственное сообщение отправлено.") logger.info(f"[start] Главное меню отправлено пользователю {user_id}.")
except Exception as e: except Exception as e:
logger.exception(f"Ошибка при обработке команды /start для пользователя {message.from_user.id}: {e}") logger.exception(
f"[start] Ошибка при обработке /start для пользователя {user_id}: {e}"
)
await message.answer("Произошла ошибка. Попробуйте позже.") await message.answer("Произошла ошибка. Попробуйте позже.")
@router.message(Command("referrals"))
async def referrals_menu_command(message: Message):
"""
Команда /referrals из бокового меню Telegram.
Показывает текст реферальной программы.
"""
logger.info(
f"[start] Команда /referrals от {message.from_user.id} "
f"(@{message.from_user.username})"
)
try:
text = await _build_referral_text(message.bot, message.from_user.id)
await message.answer(text, parse_mode=ParseMode.HTML)
logger.info("[start] Реферальная программа отправлена пользователю.")
except Exception as e:
logger.exception(
f"[start] Ошибка при обработке /referrals для пользователя "
f"{message.from_user.id}: {e}"
)
await message.answer("Произошла ошибка. Попробуйте позже.")
@router.callback_query(lambda callback: callback.data == "base") @router.callback_query(lambda callback: callback.data == "base")
async def start_callback_handler(callback: CallbackQuery): async def start_callback_handler(callback: CallbackQuery):
""" """
Обработчик callback_query с data="base". Callback с data="base" — возврат в главное меню.
""" """
try: try:
user_data = await call_api("GET", f"/user/{callback.from_user.id}") user_id = callback.from_user.id
if not user_data: username = callback.from_user.username
await call_api("POST", "/user/create", {"telegram_id": callback.from_user.id}) logger.info(f"[start] callback 'base' от {user_id} (@{username})")
user_data = await call_api("GET", f"/user/{user_id}")
if user_data in (None, "ERROR"):
await call_api(
"POST",
"/user/create",
{"telegram_id": user_id},
)
await callback.message.edit_text( await callback.message.edit_text(
f"""Приветствуем в рядах птенец {callback.from_user.username}🐣 _welcome_text(username),
\nСкорее пополни баланс, приобрети подписку и получай доступ к TikTok, YouTube, Discord, Instagram на всех устройствах без ограничений ❕❕❕ reply_markup=main_keyboard(),
\nОзнакомься с руководством по пользованию и выбери подходящий тариф 🦅
\nСледи за акциями, спец-предложениями, а также розыгрышами по ссылке ниже
\n👇👇👇
\nhttps://t.me/+0z5xqn3F1m02OTJi
\nС любовью ваши пернатые разработчики 🤍🤍🤍""",
reply_markup=main_keyboard()
) )
except Exception as e: except Exception as e:
logger.exception(f"Ошибка при обработке callback с data='base': {e}") logger.exception(f"[start] Ошибка при обработке callback 'base': {e}")
await callback.message.answer("Произошла ошибка. Попробуйте позже.") await callback.message.answer("Произошла ошибка. Попробуйте позже.")
finally: finally:
await callback.answer() await callback.answer()

View File

@@ -1,168 +1,512 @@
from aiogram import Router, types 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.enums.parse_mode import ParseMode
from aiogram.filters import Command from aiogram.filters import Command
from keyboard.keyboards import tarif_Lark_pro_keyboard, tarif_Lark_keyboard, tarif_confirm_keyboard,buy_keyboard import logging
from datetime import datetime
import aiohttp
from instences.config import BASE_URL_FASTAPI
from keyboard.keyboards import (
buy_keyboard,
tarif_Lark_keyboard,
tarif_Lark_pro_keyboard,
tarif_Lark_family_keyboard,
tarif_confirm_keyboard,
subscriptions_card_keyboard,
)
router = Router() router = Router()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
async def call_api(method, endpoint, data=None):
async def call_api(method: str, endpoint: str, data: dict | None = None):
""" """
Выполняет HTTP-запрос к FastAPI. Унифицированный вызов backend API.
""" """
url = f"{BASE_URL_FASTAPI}{endpoint}" url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"Инициализация запроса: {method} {url} с данными {data}") logger.info(f"API {method} {url} data={data}")
try: try:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.request(method, url, json=data) as response: async with session.request(method, url, json=data) as resp:
logger.info(f"Получен ответ от {url}: статус {response.status}") logger.info(f"API response {resp.status} {resp.reason}")
if resp.status in {200, 201}:
if response.status in {200, 201}: return await resp.json()
result = await response.json() if resp.status in {400, 404}:
logger.debug(f"Ответ JSON: {result}") # Возвращаем JSON с detail, чтобы разобрать ошибку
return result try:
if response.status in {404,400}: return await resp.json()
result = await response.json() except Exception:
logger.debug(f"Код {response.status}, возвращаю {result}") return None
return result logger.error(f"Unexpected status {resp.status}: {await resp.text()}")
logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
return "ERROR" return "ERROR"
except Exception as e: except Exception as e:
logger.exception(f"Исключение при выполнении запроса к {url}: {e}") logger.exception(f"API exception {url}: {e}")
return "ERROR" return "ERROR"
def escape_markdown_v2(text: str) -> str: def escape_markdown_v2(text: str) -> str:
""" """
Экранирует специальные символы для Markdown_V2. Экранирует спецсимволы для Markdown_V2.
""" """
special_chars = r"_*[]()~`>#+-=|{}.!" special_chars = r"_*[]()~`>#+-=|{}.!\\"
for char in special_chars: for ch in special_chars:
text = text.replace(char, f"\\{char}") text = text.replace(ch, f"\\{ch}")
return text return text
def _plan_human_title(plan: str) -> str:
"""
Красивое имя тарифа по plan-строке из БД.
"""
if not plan:
return ""
if plan.startswith("Lark_Standart"):
return "Lark Basic"
if plan.startswith("Lark_Pro"):
return "Lark Pro"
if plan.startswith("Lark_Family"):
return "Lark Family"
return plan
async def _fetch_user_and_subs(telegram_id: int):
"""
Берём пользователя и список его подписок с backend.
"""
user = await call_api("GET", f"/user/{telegram_id}")
if not user or not isinstance(user, dict):
return None, []
user_id = user["telegram_id"]
subs = await call_api("GET", f"/subscriptions/{user_id}")
if subs == "ERROR" or not isinstance(subs, list):
subs = []
return user, subs
async def _show_subscriptions_view(
msg: types.Message,
telegram_id: int,
index: int = 0,
edit: bool = False,
):
"""
Рендер карточки подписки: навигация + конфиг + продление.
"""
user, subs = await _fetch_user_and_subs(telegram_id)
if not user:
text = "Не удалось получить данные пользователя. Попробуй чуть позже."
if edit:
await msg.edit_text(text)
else:
await msg.answer(text)
return
if not subs:
text = (
"У тебя пока нет активных подписок.\n\n"
"Можешь подключить тариф через меню подписок."
)
if edit:
await msg.edit_text(
text,
parse_mode=ParseMode.HTML,
reply_markup=buy_keyboard(),
)
else:
await msg.answer(
text,
parse_mode=ParseMode.HTML,
reply_markup=buy_keyboard(),
)
return
# Нормализуем индекс
index = max(0, min(index, len(subs) - 1))
sub = subs[index]
plan_title = _plan_human_title(sub.get("plan"))
expiry_raw = sub.get("expiry_date")
date_str = ""
days_left = None
status_emoji = "🟢"
if expiry_raw:
try:
dt = datetime.fromisoformat(expiry_raw)
date_str = dt.strftime("%d.%m.%Y")
now = datetime.now(dt.tzinfo) if dt.tzinfo else datetime.now()
days_left = (dt.date() - now.date()).days
if days_left < 0:
status_emoji = ""
else:
status_emoji = "🟢"
except ValueError:
status_emoji = ""
lines = [
f"{status_emoji} <b>Подписка</b>",
f"Тариф: {plan_title}",
]
if days_left is None:
lines.append(f"Действует до {date_str}")
else:
if days_left >= 0:
lines.append(f"Действует до {date_str} (ещё {days_left} дн.)")
else:
lines.append(f"Истекла {date_str}")
lines.append("")
lines.append(f"{index + 1} из {len(subs)}")
text = "\n".join(lines)
kb = subscriptions_card_keyboard(
sub_id=sub["id"],
index=index,
total=len(subs),
)
if edit:
await msg.edit_text(text, parse_mode=ParseMode.HTML, reply_markup=kb)
else:
await msg.answer(text, parse_mode=ParseMode.HTML, reply_markup=kb)
# ===== Мои подписки / карточка =====
@router.message(Command("subscriptions")) @router.message(Command("subscriptions"))
async def supp(message: types.Message): async def cmd_subscriptions(message: types.Message):
""" """
Меню системы подписок /subscriptions — открыть карточку подписок.
""" """
text = "" await _show_subscriptions_view(
uri = None # Инициализация переменной msg=message,
try: telegram_id=message.from_user.id,
# Вызов API для получения URI index=0,
result = await call_api("GET", f"/uri?telegram_id={message.from_user.id}") edit=False,
uri = result.get('detail', "Error") # Получаем URI из ответа или "Error", если ключ отсутствует
# Проверка результата
if uri == "Error":
text = escape_markdown_v2("Произошла ошибка при получении URI")
elif uri == "SUB_ERROR":
text = escape_markdown_v2("Вы ещё не приобрели подписки!!")
elif "vless" in uri:
escaped_uri = escape_markdown_v2(uri) # Экранирование URI
text = f"Ваша подписка: ```{escaped_uri}```"
else:
text = escape_markdown_v2("Произошла ошибка при обработке URI")
except Exception as e:
# Логирование ошибок
logger.error(f"Ошибка при вызове API для подписки: {e}")
text = escape_markdown_v2("Произошла неожиданная ошибка при получении подписки.")
# Ответ пользователю
await message.answer(
text,
parse_mode=ParseMode.MARKDOWN_V2
) )
@router.callback_query(lambda c: c.data == "go_subscriptions")
async def cb_go_subscriptions(callback: types.CallbackQuery):
"""
Кнопка из других экранов — перейти в 'Мои подписки'.
"""
await _show_subscriptions_view(
msg=callback.message,
telegram_id=callback.from_user.id,
index=0,
edit=True,
)
await callback.answer()
@router.callback_query(lambda c: c.data.startswith("sub_prev:"))
async def cb_sub_prev(callback: types.CallbackQuery):
"""
Навигация: предыдущая подписка.
"""
try:
index = int(callback.data.split(":")[1])
except (IndexError, ValueError):
index = 0
await _show_subscriptions_view(
msg=callback.message,
telegram_id=callback.from_user.id,
index=index,
edit=True,
)
await callback.answer()
@router.callback_query(lambda c: c.data.startswith("sub_next:"))
async def cb_sub_next(callback: types.CallbackQuery):
"""
Навигация: следующая подписка.
"""
try:
index = int(callback.data.split(":")[1])
except (IndexError, ValueError):
index = 0
await _show_subscriptions_view(
msg=callback.message,
telegram_id=callback.from_user.id,
index=index,
edit=True,
)
await callback.answer()
@router.callback_query(lambda c: c.data.startswith("sub_cfg:"))
async def cb_sub_cfg(callback: types.CallbackQuery):
"""
Кнопка 'Конфиг' в карточке подписки.
Тянем URI через /uri?telegram_id=...
"""
result = await call_api(
"GET",
f"/uri?telegram_id={callback.from_user.id}",
)
if result == "ERROR" or not isinstance(result, dict):
await callback.message.answer(
"Не удалось получить конфиг. Попробуй позже."
)
await callback.answer()
return
detail = result.get("detail")
if detail == "SUB_ERROR":
await callback.message.answer(
"У тебя нет активной подписки для генерации конфига."
)
await callback.answer()
return
if not isinstance(detail, str) or not detail:
await callback.message.answer(
"Что-то пошло не так при выдаче конфига."
)
await callback.answer()
return
escaped = escape_markdown_v2(detail)
text = f"Твой конфиг:\n```{escaped}```"
await callback.message.answer(
text,
parse_mode=ParseMode.MARKDOWN_V2,
)
await callback.answer()
@router.callback_query(lambda c: c.data.startswith("sub_renew:"))
async def cb_sub_renew(callback: types.CallbackQuery):
"""
Кнопка 'Продлить' в карточке подписки.
По факту — повторный /subscription/buy с тем же планом.
"""
sub_id = callback.data.split(":", maxsplit=1)[
1] if ":" in callback.data else None
user, subs = await _fetch_user_and_subs(callback.from_user.id)
if not user or not subs:
await callback.message.answer("Подписка не найдена.")
await callback.answer()
return
target = next((s for s in subs if s.get("id") == sub_id), None)
if not target:
await callback.message.answer("Подписка не найдена.")
await callback.answer()
return
plan_id = target.get("plan")
if not plan_id:
await callback.message.answer("Не удалось определить тариф для продления.")
await callback.answer()
return
result = await call_api(
"POST",
"/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
)
# Ошибки backend (detail)
if result == "ERROR" or not isinstance(result, dict):
await callback.message.answer("Ошибка при продлении. Попробуй позже.")
await callback.answer()
return
detail = result.get("detail")
if detail == "INSUFFICIENT_FUNDS":
await callback.message.answer("Недостаточно средств на балансе.")
await callback.answer()
return
if detail == "TARIFF_NOT_FOUND":
await callback.message.answer("Тариф не найден.")
await callback.answer()
return
if detail == "ACTIVE_SUBSCRIPTION_EXISTS":
await callback.message.answer("У тебя уже есть активная подписка этого типа.")
await callback.answer()
return
# Успех — backend вернёт message с URI
uri = result.get("message")
if isinstance(uri, str) and uri:
escaped_uri = escape_markdown_v2(uri)
await callback.message.answer(
"Подписка успешно продлена.\n"
f"Твой конфиг:\n```{escaped_uri}```",
parse_mode=ParseMode.MARKDOWN_V2,
)
else:
await callback.message.answer("Подписка успешно продлена.")
# Перерисовать карточку (обновлённый список)
await _show_subscriptions_view(
msg=callback.message,
telegram_id=callback.from_user.id,
index=0,
edit=True,
)
await callback.answer()
# ===== Меню покупки подписки (Basic / Pro / Family) =====
@router.callback_query(lambda callback: callback.data == "buy_subscription") @router.callback_query(lambda callback: callback.data == "buy_subscription")
async def buy_subscription_callback_handler(callback: types.CallbackQuery): async def buy_subscription_callback_handler(callback: types.CallbackQuery):
""" """
Обработчик callback_query с data="buy_subscription". Меню выбора тарифа: Basic / Pro / Family + показ баланса.
""" """
await callback.message.edit_text( user_data = await call_api("GET", f"/user/{callback.from_user.id}")
f"Ознакомься с условиями в вкладке \"О тарифах\" и выбери подходящий 🦅", balance = user_data.get("balance", 0) if user_data else "?"
reply_markup=buy_keyboard()
text = (
" <b>Подключить новую подписку</b>\n\n"
"🐣 Lark Basic - 200 ₽ / мес\n"
"🦅 Lark Pro - 400 ₽ / мес\n"
"👨‍👩‍👧 Lark Family - 700 ₽ / мес\n\n"
f"💰 Баланс: {balance}\n\n"
"Выбери тариф:"
) )
await callback.message.edit_text(
text,
parse_mode=ParseMode.HTML,
reply_markup=buy_keyboard(),
)
await callback.answer()
@router.callback_query(lambda callback: callback.data == "subs") @router.callback_query(lambda callback: callback.data == "subs")
async def subs_callback_handler(callback: types.CallbackQuery): async def subs_callback_handler(callback: types.CallbackQuery):
""" """
Обработчик callback_query с data="subs". Кнопка 'Lark Basic' — раскрываем варианты сроков.
""" """
await callback.message.edit_text( await callback.message.edit_text(
"Подписки птенчик", "Тариф Lark Basic — выбери срок:",
reply_markup=tarif_Lark_keyboard() reply_markup=tarif_Lark_keyboard(),
) )
await callback.answer()
@router.callback_query(lambda callback: callback.data == "subs_pro") @router.callback_query(lambda callback: callback.data == "subs_pro")
async def subs_pro_callback_handler(callback: types.CallbackQuery): async def subs_pro_callback_handler(callback: types.CallbackQuery):
""" """
Обработчик callback_query с data="subs_pro". Кнопка 'Lark Pro'.
""" """
await callback.message.edit_text( await callback.message.edit_text(
"Подписки птенчик ПРО", "Тариф Lark Pro — выбери срок:",
reply_markup=tarif_Lark_pro_keyboard() reply_markup=tarif_Lark_pro_keyboard(),
) )
await callback.answer()
@router.callback_query(lambda callback: callback.data == "subs_family")
async def subs_family_callback_handler(callback: types.CallbackQuery):
"""
Кнопка 'Lark Family'.
"""
await callback.message.edit_text(
"Тариф Lark Family — выбери срок:",
reply_markup=tarif_Lark_family_keyboard(),
)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("Lark:")) @router.callback_query(lambda callback: callback.data.startswith("Lark:"))
async def lark_tariff_callback_handler(callback: types.CallbackQuery): async def lark_tariff_callback_handler(callback: types.CallbackQuery):
""" """
Обработчик для выбора тарифа Lark. Обработчик выбора конкретного тарифа Lark.
callback_data: Lark:<Class>:<months>
""" """
data = callback.data.split(":") try:
tariff_name = data[0] _, tariff_class, months_str = callback.data.split(":")
tariff_class = data[1] months = int(months_str)
tariff_time = int(data[2]) except Exception:
await callback.message.answer("Некорректные данные тарифа.")
await callback.answer()
return
# Определение окончания для месяцев if months == 1:
if tariff_time == 1: months_label = "1 месяц"
months = f"{tariff_time} месяц" elif 2 <= months <= 4:
elif 2 <= tariff_time <= 4: months_label = f"{months} месяца"
months = f"{tariff_time} месяца"
else: else:
months = f"{tariff_time} месяцев" months_label = f"{months} месяцев"
text = f"Тариф {tariff_name} на {months}. Продолжите покупку..." text = f"Тариф Lark {tariff_class} на {months_label}. Продолжить покупку?"
# Рендеринг клавиатуры keyboard = tarif_confirm_keyboard("Lark", months, tariff_class)
keyboard = tarif_confirm_keyboard(tariff_name, tariff_time, tariff_class)
await callback.message.edit_text(text=text, reply_markup=keyboard) await callback.message.edit_text(text=text, reply_markup=keyboard)
await callback.answer()
@router.callback_query(lambda callback: callback.data.startswith("confirm:")) @router.callback_query(lambda callback: callback.data.startswith("confirm:"))
async def confirm_callback_handler(callback: types.CallbackQuery): async def confirm_callback_handler(callback: types.CallbackQuery):
""" """
Обработчик подтверждения подписки. Подтверждение покупки тарифа.
""" """
try: try:
data = callback.data.split(":")[1] data = callback.data.split(":", maxsplit=1)[1]
tariff_info = data.split("_") name, classif, amount_str = data.split("_")
plan_id = f"{tariff_info[0]}_{tariff_info[1]}_{tariff_info[2]}" plan_id = f"{name}_{classif}_{amount_str}"
result = await call_api("POST", "/subscription/buy", {"telegram_id": callback.from_user.id, "plan_id": plan_id}) except Exception:
detail = result.get("detail", {}) await callback.message.edit_text("Некорректные данные тарифа.")
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", {})
escaped_text = escape_markdown_v2(f"Подписка успешно оформлена!")
answer_text = f"Ваш конфиг для подключения: ```{uri}```"
await callback.message.edit_text(escaped_text)
await callback.message.answer(answer_text, parse_mode=ParseMode.MARKDOWN_V2)
except Exception as e:
logger.exception(f"Ошибка при обработке подтверждения подписки: {e}")
await callback.message.edit_text("Произошла ошибка при оформлении подписки.")
finally:
await callback.answer() await callback.answer()
return
result = await call_api(
"POST",
"/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
)
if result == "ERROR" or not isinstance(result, dict):
await callback.message.edit_text("Ошибка при оформлении подписки.")
await callback.answer()
return
detail = result.get("detail")
if detail == "INSUFFICIENT_FUNDS":
await callback.message.edit_text(
"Недостаточно средств. Попробуй пополнить баланс."
)
await callback.answer()
return
if detail == "TARIFF_NOT_FOUND":
await callback.message.edit_text("Тариф не найден.")
await callback.answer()
return
if detail == "ACTIVE_SUBSCRIPTION_EXISTS":
await callback.message.edit_text("У тебя уже есть активная подписка.")
await callback.answer()
return
uri = result.get("message", "")
if uri:
escaped_uri = escape_markdown_v2(uri)
answer_text = f"Подписка успешно оформлена!\n\nТвой конфиг:\n```{escaped_uri}```"
await callback.message.edit_text("Подписка успешно оформлена.")
await callback.message.answer(
answer_text,
parse_mode=ParseMode.MARKDOWN_V2,
)
else:
await callback.message.edit_text("Подписка успешно оформлена.")
await callback.answer()

View File

@@ -1,159 +1,458 @@
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder
from aiogram.types import InlineKeyboardButton, KeyboardButton from aiogram.types import InlineKeyboardButton, KeyboardButton
def main_keyboard(): def main_keyboard():
""" """
База Главное меню
""" """
builder = InlineKeyboardBuilder() builder = InlineKeyboardBuilder()
builder.row(InlineKeyboardButton(text="Профиль", callback_data="profile")) builder.row(
builder.row(InlineKeyboardButton(text="❔FAQ❔", callback_data="faq")) InlineKeyboardButton(
builder.row(InlineKeyboardButton(text="О нас", url="https://www.youtube.com/watch?v=Zirn-CKck-c")) text="📜 Профиль",
return builder.as_markup() callback_data="profile",
)
def account_keyboard(): )
""" # ------
Аккаунт builder.row(
""" InlineKeyboardButton(
builder = InlineKeyboardBuilder() text="❔ FAQ ❔",
builder.row(InlineKeyboardButton(text="Пополнение баланса", callback_data="popup")) callback_data="faq",
builder.row(InlineKeyboardButton(text="Приобрести подписку", callback_data="buy_subscription")) )
builder.row(InlineKeyboardButton(text="Руководство по подключению", callback_data="guide")) )
builder.row(InlineKeyboardButton(text="📑История транзакций", callback_data="tranhist")) builder.row(
builder.row(InlineKeyboardButton(text="Назад", callback_data="base")) InlineKeyboardButton(
return builder.as_markup() text=" О нас",
url="https://www.youtube.com/watch?v=Zirn-CKck-c",
def ticket_list_keyboard(tickets): )
builder = InlineKeyboardBuilder() )
for ticket in tickets: return builder.as_markup()
builder.row(InlineKeyboardButton(text=f"Тикет: {ticket['subject']}",callback_data=f"ticket_{ticket['id']}"))
builder.row(InlineKeyboardButton(text="Назад", callback_data="main_sup"))
return builder.as_markup() def account_keyboard():
"""
Клавиатура профиля:
def sup_keyboard(): пополнить баланс, история транзакций, назад в главное меню.
builder = InlineKeyboardBuilder() """
builder.row(InlineKeyboardButton(text="Создать запрос", callback_data="make_ticket")) builder = InlineKeyboardBuilder()
builder.row(InlineKeyboardButton(text="Мои запросы", callback_data="my_tickets")) builder.row(
return builder.as_markup() InlineKeyboardButton(
text="🪙 Пополнить баланс",
def ticket_keyboard(): callback_data="balance",
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="Отмена", callback_data="cancel")) )
return builder.as_markup() builder.row(
InlineKeyboardButton(
def buy_keyboard(): text="🧾 История транзакций",
""" callback_data="tranhist",
Приобрести подписку )
""" )
builder = InlineKeyboardBuilder() builder.row(
builder.row(InlineKeyboardButton(text="Тариф Lark", callback_data="subs")) InlineKeyboardButton(
builder.row(InlineKeyboardButton(text="Тариф Lark Pro", callback_data="subs_pro")) text="🔙 Назад",
builder.row(InlineKeyboardButton(text="О тарифах", url="https://t.me/proxylark/19")) callback_data="base",
builder.row(InlineKeyboardButton(text="Назад", callback_data="profile")) )
return builder.as_markup() )
return builder.as_markup()
def subhist_keyboard():
"""
Подписки def balance_keyboard():
""" """
builder = InlineKeyboardBuilder() Экран баланса
builder.button(text="Назад", callback_data="profile") """
return builder.as_markup() builder = InlineKeyboardBuilder()
builder.row(
def popup_keyboard(): InlineKeyboardButton(
""" text="🪙 Пополнить баланс",
Пополнение callback_data="popup",
""" )
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="200₽", callback_data="popup:200"),InlineKeyboardButton(text="500₽", callback_data="popup:500")) builder.row(
builder.row(InlineKeyboardButton(text="1000₽", callback_data="popup:1000"),InlineKeyboardButton(text="2000₽", callback_data="popup:2000")) InlineKeyboardButton(
builder.row(InlineKeyboardButton(text="3000₽", callback_data="popup:3000"),InlineKeyboardButton(text="5000₽", callback_data="popup:5000")) text="🧾 История транзакций",
builder.row(InlineKeyboardButton(text="Назад", callback_data="profile")) callback_data="tranhist",
return builder.as_markup() )
)
def balance_keyboard(): builder.row(
""" InlineKeyboardButton(
Баланс text="🔙 Назад",
""" callback_data="profile",
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="Пополнение", callback_data="popup")) )
builder.row(InlineKeyboardButton(text="История транзакций", callback_data="tranhist")) return builder.as_markup()
builder.row(InlineKeyboardButton(text="Назад", callback_data="profile"))
return builder.as_markup()
def popup_keyboard():
def tarif_Lark_keyboard(): """
""" Суммы пополнения: 200, 300, 600, 1000 ₽.
Тариф Lark """
""" builder = InlineKeyboardBuilder()
builder = InlineKeyboardBuilder() builder.row(
builder.row(InlineKeyboardButton(text="Тариф Lark 1 Месяц", callback_data="Lark:Standart:1")) InlineKeyboardButton(text="200 ₽", callback_data="popup:200"),
builder.row(InlineKeyboardButton(text="Тариф Lark 6 Месяц", callback_data="Lark:Standart:6")) )
builder.row(InlineKeyboardButton(text="Тариф Lark 12 Месяц", callback_data="Lark:Standart:12")) builder.row(
builder.row(InlineKeyboardButton(text="Назад", callback_data="buy_subscription")) InlineKeyboardButton(text="300 ₽", callback_data="popup:300"),
return builder.as_markup() )
builder.row(
def tarif_Lark_pro_keyboard(): InlineKeyboardButton(text="600 ₽", callback_data="popup:600"),
""" )
Тариф Lark Pro builder.row(
""" InlineKeyboardButton(text="1000 ₽", callback_data="popup:1000"),
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="Тариф Lark Pro 1 Месяц", callback_data="Lark:Pro:1")) builder.row(
builder.row(InlineKeyboardButton(text="Тариф Lark Pro 6 Месяц", callback_data="Lark:Pro:6")) InlineKeyboardButton(
builder.row(InlineKeyboardButton(text="Тариф Lark Pro 12 Месяц", callback_data="Lark:Pro:12")) text="🔙 Назад",
builder.row(InlineKeyboardButton(text="Назад", callback_data="buy_subscription")) callback_data="profile", # назад в профиль
return builder.as_markup() )
)
def guide_keyboard(): return builder.as_markup()
"""
Руководство по подключению
""" def payment_methods_keyboard(amount: int):
builder = InlineKeyboardBuilder() """
builder.row(InlineKeyboardButton(text="IOS,Android", callback_data="mob")) Способы оплаты для выбранной суммы.
builder.row(InlineKeyboardButton(text="Windows,MacOS", callback_data="pc")) """
builder.row(InlineKeyboardButton(text="Назад", callback_data="profile")) builder = InlineKeyboardBuilder()
return builder.as_markup() builder.row(
InlineKeyboardButton(
text="⭐ Telegram Stars",
# def about_tarifs_keyboard(): callback_data=f"method_stars_{amount}",
# """ )
# О тарифах )
# """ builder.row(
# builder = InlineKeyboardBuilder() InlineKeyboardButton(
# builder.row(InlineKeyboardButton(text="Назад", callback_data="buy_subscription")) text="💵 YooKassa",
# return builder.as_markup() callback_data=f"method_ykassa_{amount}",
)
)
def faq_keyboard(): builder.row(
""" InlineKeyboardButton(
FAQ text="🪙 CryptoBot",
""" callback_data=f"method_crypto_{amount}",
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="Назад", callback_data="base")) )
return builder.as_markup() builder.row(
InlineKeyboardButton(
def tranhist_keyboard(): text="🏦 СБП (Система быстрых платежей)",
""" callback_data=f"method_sbp_{amount}",
История транзакций )
""" )
builder = InlineKeyboardBuilder() builder.row(
builder.row(InlineKeyboardButton(text="Назад",callback_data="profile")) InlineKeyboardButton(
return builder.as_markup() text="🔙 Назад",
callback_data="popup",
def tarif_confirm_keyboard(name,amount,classif): )
""" )
Подтверждение покупки тарифа return builder.as_markup()
"""
builder = InlineKeyboardBuilder()
builder.row(InlineKeyboardButton(text="Подтвердить", callback_data=f"confirm:{name}_{classif}_{amount}")) def ticket_list_keyboard(tickets):
builder.row(InlineKeyboardButton(text="Отменить",callback_data="buy_subscription")) builder = InlineKeyboardBuilder()
return builder.as_markup() for ticket in tickets:
builder.row(
def confirm_popup_keyboard(): InlineKeyboardButton(
""" text=f"Тикет: {ticket['subject']}",
Подтверждение пополнения callback_data=f"ticket_{ticket['id']}",
""" )
builder = InlineKeyboardBuilder() )
builder.row(InlineKeyboardButton(text="Теперь иди нахуй", callback_data="profile")) builder.row(
return builder.as_markup() InlineKeyboardButton(
text="🔙 Назад",
callback_data="main_sup",
)
)
return builder.as_markup()
def sup_keyboard():
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="📝 Создать запрос",
callback_data="make_ticket",
)
)
builder.row(
InlineKeyboardButton(
text="📂 Мои запросы",
callback_data="my_tickets",
)
)
return builder.as_markup()
def ticket_keyboard():
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🔙 Отмена",
callback_data="cancel",
)
)
return builder.as_markup()
def buy_keyboard():
"""
Меню выбора тарифа.
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🐣 Lark Basic",
callback_data="subs",
)
)
builder.row(
InlineKeyboardButton(
text="🦅 Lark Pro",
callback_data="subs_pro",
)
)
builder.row(
InlineKeyboardButton(
text="👨‍👩‍👧 Lark Family",
callback_data="subs_family",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="profile",
)
)
return builder.as_markup()
def tarif_Lark_keyboard():
"""
Тариф Lark Basic (Standart)
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🐣 Lark 1 месяц",
callback_data="Lark:Standart:1",
)
)
builder.row(
InlineKeyboardButton(
text="🐣 Lark 6 месяцев",
callback_data="Lark:Standart:6",
)
)
builder.row(
InlineKeyboardButton(
text="🐣 Lark 12 месяцев",
callback_data="Lark:Standart:12",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="buy_subscription",
)
)
return builder.as_markup()
def tarif_Lark_pro_keyboard():
"""
Тариф Lark Pro
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🦅 Lark Pro 1 месяц",
callback_data="Lark:Pro:1",
)
)
builder.row(
InlineKeyboardButton(
text="🦅 Lark Pro 6 месяцев",
callback_data="Lark:Pro:6",
)
)
builder.row(
InlineKeyboardButton(
text="🦅 Lark Pro 12 месяцев",
callback_data="Lark:Pro:12",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="buy_subscription",
)
)
return builder.as_markup()
def tarif_Lark_family_keyboard():
"""
Тариф Lark Family.
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="👨‍👩‍👧 Lark Family 1 месяц",
callback_data="Lark:Family:1",
)
)
builder.row(
InlineKeyboardButton(
text="👨‍👩‍👧 Lark Family 6 месяцев",
callback_data="Lark:Family:6",
)
)
builder.row(
InlineKeyboardButton(
text="👨‍👩‍👧 Lark Family 12 месяцев",
callback_data="Lark:Family:12",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="buy_subscription",
)
)
return builder.as_markup()
def subscriptions_card_keyboard(sub_id: str, index: int, total: int):
"""
Карточка подписки:
навигация, конфиг, продление, новая, назад.
"""
builder = InlineKeyboardBuilder()
nav = []
if index > 0:
nav.append(
InlineKeyboardButton(
text="⬅️",
callback_data=f"sub_prev:{index-1}",
)
)
if index < total - 1:
nav.append(
InlineKeyboardButton(
text="➡️",
callback_data=f"sub_next:{index+1}",
)
)
if nav:
builder.row(*nav)
builder.row(
InlineKeyboardButton(
text="🔑 Конфиг",
callback_data=f"sub_cfg:{sub_id}",
),
InlineKeyboardButton(
text="🔁 Продлить",
callback_data=f"sub_renew:{sub_id}",
),
)
builder.row(
InlineKeyboardButton(
text=" Новая",
callback_data="buy_subscription",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="profile",
)
)
return builder.as_markup()
def guide_keyboard():
"""
Руководство по подключению
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="📱 iOS / Android",
callback_data="mob",
)
)
builder.row(
InlineKeyboardButton(
text="💻 Windows / macOS",
callback_data="pc",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="profile",
)
)
return builder.as_markup()
def faq_keyboard():
"""
FAQ
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="base",
)
)
return builder.as_markup()
def tranhist_keyboard():
"""
История транзакций
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="profile",
)
)
return builder.as_markup()
def tarif_confirm_keyboard(name: str, amount: int, classif: str):
"""
Подтверждение покупки тарифа
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="✅ Подтвердить",
callback_data=f"confirm:{name}_{classif}_{amount}",
)
)
builder.row(
InlineKeyboardButton(
text="🔙 Отменить",
callback_data="buy_subscription",
)
)
return builder.as_markup()
def confirm_popup_keyboard():
"""
Подтверждение пополнения.
"""
builder = InlineKeyboardBuilder()
builder.row(
InlineKeyboardButton(
text="✅ Готово, вернуться в профиль",
callback_data="profile",
)
)
return builder.as_markup()

10
main.py
View File

@@ -11,7 +11,8 @@ import logging
BOT_TOKEN = os.getenv("TOKEN") BOT_TOKEN = os.getenv("TOKEN")
if not BOT_TOKEN: if not BOT_TOKEN:
raise ValueError("Не задан токен бота. Убедитесь, что переменная окружения 'TOKEN' установлена.") raise ValueError(
"Не задан токен бота. Убедитесь, что переменная окружения 'TOKEN' установлена.")
bot = Bot(token=BOT_TOKEN) bot = Bot(token=BOT_TOKEN)
dp = Dispatcher() dp = Dispatcher()
@@ -23,9 +24,10 @@ dp.message.middleware(AntiSpamMiddleware(rate_limit=1))
async def set_commands(): async def set_commands():
"""Устанавливает команды для бота.""" """Устанавливает команды для бота."""
commands = [ commands = [
BotCommand(command="/start", description="🥚Главное меню"), BotCommand(command="start", description="Запуск бота"),
BotCommand(command="/subscriptions", description="🦴Мои подписки"), BotCommand(command="subscriptions", description="Мои подписки"),
BotCommand(command="/support", description="❕Поддержка"), BotCommand(command="support", description="Техподдержка"),
BotCommand(command="referrals", description="Реферальная программа"),
] ]
await bot.set_my_commands(commands) await bot.set_my_commands(commands)