3 Commits

Author SHA1 Message Date
unknown
ead48389b7 Много чего изменил, надеюсь, запушил правильно 2025-12-06 11:49:36 +03:00
unknown
33414dee65 Исправил много чего в фронте 2025-12-06 11:38:34 +03:00
unknown
709e8f09eb Чет исправил, я забыл 2025-12-02 14:28:03 +03:00
5 changed files with 315 additions and 110 deletions

View File

@@ -1,7 +1,12 @@
# Профиль. последнее изменение 24.11.2025
from aiogram import Router, types
from aiogram.types import CallbackQuery
from aiogram import Router, types, F
from aiogram.types import (
CallbackQuery,
LabeledPrice,
Message,
PreCheckoutQuery,
)
import logging
from datetime import datetime
from aiogram.enums.parse_mode import ParseMode
@@ -88,11 +93,12 @@ async def profile_callback_handler(callback: CallbackQuery):
sub_status = "⚫ Нет активных"
if sub_data:
expiry_date = sub_data.get("expiry_date")
if expiry_date:
end_date_str = sub_data.get("end_date") # Исправил
if end_date_str:
try:
is_expired = datetime.fromisoformat(
expiry_date) < datetime.now()
is_expired = (
datetime.fromisoformat(end_date_str) < datetime.now()
)
except ValueError:
is_expired = True
else:
@@ -233,15 +239,52 @@ async def popup_confirm_callback_handler(callback: CallbackQuery):
await callback.answer()
# ===== Telegram Stars =====
@router.callback_query(lambda callback: callback.data.startswith("method_stars_"))
async def method_stars_handler(callback: CallbackQuery):
"""
Заглушка: оплата через Telegram Stars.
Оплата через Telegram Stars.
Формируем invoice прямо из бота, без отдельного биллинга.
"""
amount = callback.data.split("_")[-1]
await callback.message.edit_text(
f"⭐ Оплата через Telegram Stars на {amount} ₽ пока в разработке.\n\n"
"Позже сюда подвяжем реальный платёж.",
try:
amount_str = callback.data.split("_")[-1]
amount_rub = int(float(amount_str))
except Exception:
await callback.message.answer("Некорректная сумма для оплаты.")
await callback.answer()
return
# Внутренний payload, чтобы при успешной оплате понять, что это пополнение баланса
payload = f"stars_topup:{callback.from_user.id}:{amount_rub}"
# 1 ₽ == 1 Star, если нужна другая конвертация — меняй тут
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()
@@ -383,3 +426,87 @@ async def guide_callback_handler(callback: CallbackQuery):
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(
"Оплата прошла, но произошла ошибка при обработке. "
"Если баланс не изменился — напиши, пожалуйста, в поддержку."
)

View File

@@ -3,6 +3,8 @@ from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode
import logging
from .start import call_api # используем уже готовый HTTP-хелпер
router = Router()
logger = logging.getLogger(__name__)
@@ -13,22 +15,27 @@ async def _build_referral_text(bot, user_id: int) -> str:
link = f"https://t.me/{bot_username}?start=ref_{user_id}"
invited_count = ""
try:
data = await call_api("GET", f"/user/{user_id}/referrals")
if isinstance(data, dict):
invited_count = str(data.get("invited_count", 0))
except Exception as e:
logger.exception(
f"Ошибка при получении количества рефералов для {user_id}: {e}"
)
text = (
"👥 <b>Реферальная программа</b>\n\n"
"Зови друзей в Lark VPN и получай бонусы на баланс.\n\n"
f"🔗 Твоя ссылка:\n<code>{link}</code>\n\n"
"👤 Приглашено: \n"
"💰 Начислено бонусов: — ₽\n\n"
"Бонусы падают автоматически, когда приглашённые пополняют баланс."
f"👤 Приглашено: {invited_count}\n\n"
)
return text
@router.message(Command("referrals"))
async def referrals_command(message: types.Message):
"""
Команда /referrals — показывает текст реферальной программы.
"""
logger.info(f"Получена команда /referrals от {message.from_user.id}")
try:
text = await _build_referral_text(message.bot, message.from_user.id)
@@ -40,9 +47,6 @@ async def referrals_command(message: types.Message):
@router.callback_query(lambda callback: callback.data == "referral")
async def referrals_callback(callback: types.CallbackQuery):
"""
Кнопка «Реферальная программа» в главном меню.
"""
try:
text = await _build_referral_text(
callback.message.bot,

View File

@@ -1,113 +1,188 @@
from aiogram import Router, types
from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery
import logging
from instences.config import BASE_URL_FASTAPI
import aiohttp
from keyboard.keyboards import main_keyboard
from aiogram.enums.parse_mode import ParseMode
import logging
import aiohttp
from aiogram.types import Message, CallbackQuery
from instences.config import BASE_URL_FASTAPI
from keyboard.keyboards import main_keyboard
from .referrals import _build_referral_text
router = Router()
logger = logging.getLogger(__name__)
async def call_api(method, endpoint, data=None):
"""
Выполняет HTTP-запрос к FastAPI.
:param method: HTTP метод (GET, POST, и т.д.)
:param endpoint: конечная точка API
:param data: тело запроса (если необходимо)
:return: JSON-ответ или "ERROR" при неуспехе
"""
async def call_api(method: str, endpoint: str, data: dict | None = None):
"""Мини-обёртка для запросов к backend API."""
url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"Инициализация запроса: {method} {url} с данными {data}")
logger.info(f"API {method} {url} data={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 == 404:
logger.debug(f"Код {response.status}, возвращаю ничего")
async with session.request(method, url, json=data) as resp:
logger.info(f"API response {resp.status} {resp.reason}")
if resp.status in {200, 201}:
return await resp.json()
if resp.status in {400, 404}:
try:
return await resp.json()
except Exception:
return None
logger.error(
f"Ошибка в запросе: статус {response.status}, причина {response.reason}"
f"Unexpected status {resp.status}: {await resp.text()}"
)
return "ERROR"
except Exception as e:
logger.exception(f"Исключение при выполнении запроса к {url}: {e}")
logger.exception(f"API exception {url}: {e}")
return "ERROR"
def _welcome_text(username: str | None) -> str:
"""
Текст приветствия в /start и в главном меню.
Имя пока не используем — оставляем сигнатуру на будущее.
"""
"""Текст приветствия в /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"))
async def start_command(message: Message):
"""/start с обработкой реферального параметра.
Логика:
- проверяем, есть ли пользователь в БД по telegram_id;
- если нет — создаём через /user/create;
- если есть корректный ref_ и пользователь новый — вызываем
/user/{referrer_id}/add_referral с invited_id = telegram_id.
"""
Обработчик команды /start.
Визуал и текст — обновлены, логика работы с API не тронута.
"""
user_id = message.from_user.id
username = message.from_user.username
referrer_id = _parse_referrer_id(message)
logger.info(
f"Получена команда /start от пользователя: "
f"{message.from_user.id} ({message.from_user.username})"
f"[start] Команда /start от {user_id} (@{username}), "
f"text={message.text!r}, referrer_id={referrer_id}"
)
try:
user_data = await call_api("GET", f"/user/{message.from_user.id}")
if not user_data:
logger.debug(
"Пользователь не найден в базе, создаем новую запись."
)
await call_api(
# 1. Проверяем, есть ли пользователь в БД
existing = await call_api("GET", f"/user/{user_id}")
user_exists = existing not in (None, "ERROR")
# 2. Если пользователя нет — создаём
if not user_exists:
logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
create_result = await call_api(
"POST",
"/user/create",
{"telegram_id": message.from_user.id},
{"telegram_id": user_id},
)
if create_result == "ERROR":
logger.error(
f"[start] Не удалось создать пользователя {user_id} в БД"
)
logger.debug("Отправка приветственного сообщения пользователю.")
# 3. Обработка рефералки, если параметр есть
if referrer_id is not None:
# 3.1. Самореферал
if referrer_id == user_id:
logger.info(
f"[start] Пользователь {user_id} попытался зайти "
f"по своей реферальной ссылке."
)
await message.answer(
_welcome_text(message.from_user.username),
"Нельзя переходить по своей же реферальной ссылке."
)
# 3.2. Пользователь уже зарегистрирован в боте
elif user_exists:
logger.info(
f"[start] Пользователь {user_id} уже есть в БД, "
f"реферальная ссылка {referrer_id} не сработает."
)
await message.answer(
"Вы уже зарегистрированы в боте, "
"реферальная ссылка не сработает."
)
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
else:
payload = {
"invited_id": user_id,
}
logger.info(
f"[start] Фиксирую реферала в бекенде: "
f"referrer_id={referrer_id}, payload={payload}"
)
result = await call_api(
"POST",
f"/user/{referrer_id}/add_referral",
payload,
)
if result == "ERROR":
logger.error(
f"[start] Ошибка при фиксации реферала через "
f"/user/{referrer_id}/add_referral: referrer={referrer_id}, "
f"invited={user_id}"
)
await message.answer("Вы вошли по реферальной ссылке.")
# 4. В любом случае показываем главное меню
await message.answer(
_welcome_text(username),
reply_markup=main_keyboard(),
)
logger.info("Приветственное сообщение отправлено.")
logger.info(f"[start] Главное меню отправлено пользователю {user_id}.")
except Exception as e:
logger.exception(
f"Ошибка при обработке команды /start для пользователя "
f"{message.from_user.id}: {e}"
f"[start] Ошибка при обработке /start для пользователя {user_id}: {e}"
)
await message.answer("Произошла ошибка. Попробуйте позже.")
@router.message(Command("referrals"))
async def referrals_menu_command(message: Message):
"""
Команда /referrals из бокового меню Telegram.
"""Команда /referrals из бокового меню Telegram.
Показывает текст реферальной программы.
"""
logger.info(
f"Получена команда /referrals от пользователя: "
f"{message.from_user.id} ({message.from_user.username})"
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("Реферальная программа отправлена пользователю.")
logger.info("[start] Реферальная программа отправлена пользователю.")
except Exception as e:
logger.exception(
f"Ошибка при обработке команды /referrals для пользователя "
f"[start] Ошибка при обработке /referrals для пользователя "
f"{message.from_user.id}: {e}"
)
await message.answer("Произошла ошибка. Попробуйте позже.")
@@ -115,25 +190,26 @@ async def referrals_menu_command(message: Message):
@router.callback_query(lambda callback: callback.data == "base")
async def start_callback_handler(callback: CallbackQuery):
"""
Обработчик callback_query с data="base".
Возвращает пользователя в главное меню.
"""
"""Callback с data="base" — возврат в главное меню."""
try:
user_data = await call_api("GET", f"/user/{callback.from_user.id}")
if not user_data:
user_id = callback.from_user.id
username = callback.from_user.username
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": callback.from_user.id},
{"telegram_id": user_id},
)
await callback.message.edit_text(
_welcome_text(callback.from_user.username),
_welcome_text(username),
reply_markup=main_keyboard(),
)
except Exception as e:
logger.exception(f"Ошибка при обработке callback с data='base': {e}")
logger.exception(f"[start] Ошибка при обработке callback 'base': {e}")
await callback.message.answer("Произошла ошибка. Попробуйте позже.")
finally:
await callback.answer()

View File

@@ -129,15 +129,15 @@ async def _show_subscriptions_view(
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")
plan_title = _plan_human_title(sub.get("plan_name"))
end_raw = sub.get("end_date") # Исправил
date_str = ""
days_left = None
status_emoji = "🟢"
if expiry_raw:
if end_raw:
try:
dt = datetime.fromisoformat(expiry_raw)
dt = datetime.fromisoformat(end_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
@@ -272,19 +272,22 @@ async def cb_sub_cfg(callback: types.CallbackQuery):
await callback.answer()
return
if not isinstance(detail, str) or not detail:
uri = None
if isinstance(detail, str):
uri = detail.strip()
elif isinstance(detail, list) and detail:
uri = str(detail[0]).strip()
if not uri:
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,
f"Твой конфиг:\n<code>{uri}</code>",
parse_mode=ParseMode.HTML,
)
await callback.answer()
@@ -310,7 +313,7 @@ async def cb_sub_renew(callback: types.CallbackQuery):
await callback.answer()
return
plan_id = target.get("plan")
plan_id = target.get("plan_name")
if not plan_id:
await callback.message.answer("Не удалось определить тариф для продления.")
await callback.answer()

View File

@@ -121,7 +121,7 @@ def payment_methods_keyboard(amount: int):
)
builder.row(
InlineKeyboardButton(
text="💵 YooKassa",
text="💵 СБП",
callback_data=f"method_ykassa_{amount}",
)
)
@@ -317,10 +317,11 @@ def tarif_Lark_family_keyboard():
def subscriptions_card_keyboard(sub_id: str, index: int, total: int):
"""
Карточка подписки:
навигация, конфиг, продление, новая, назад.
навигация, конфиг, назад в главное меню.
"""
builder = InlineKeyboardBuilder()
# Навигация по подпискам
nav = []
if index > 0:
nav.append(
@@ -339,28 +340,22 @@ def subscriptions_card_keyboard(sub_id: str, index: int, total: int):
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",
callback_data="base", # было "profile" изменил
)
)
return builder.as_markup()