Исправил много чего в фронте

This commit is contained in:
unknown
2025-12-06 11:38:34 +03:00
parent 709e8f09eb
commit 33414dee65
5 changed files with 152 additions and 220 deletions

View File

@@ -93,11 +93,12 @@ async def profile_callback_handler(callback: CallbackQuery):
sub_status = "⚫ Нет активных" sub_status = "⚫ Нет активных"
if sub_data: if sub_data:
expiry_date = sub_data.get("expiry_date") end_date_str = sub_data.get("end_date") # Исправил
if expiry_date: if end_date_str:
try: try:
is_expired = datetime.fromisoformat( is_expired = (
expiry_date) < datetime.now() datetime.fromisoformat(end_date_str) < datetime.now()
)
except ValueError: except ValueError:
is_expired = True is_expired = True
else: else:

View File

@@ -2,126 +2,40 @@ from aiogram import Router, types
from aiogram.filters import Command from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode from aiogram.enums.parse_mode import ParseMode
import logging import logging
import aiohttp
from instences.config import BASE_URL_FASTAPI from .start import call_api # используем уже готовый HTTP-хелпер
router = Router() router = Router()
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
async def call_api(method, endpoint, data=None, base_url=BASE_URL_FASTAPI):
"""
Универсальный HTTP-запрос к FastAPI для рефералок.
Ожидаем от бекенда:
GET /user/{telegram_id}/referrals -> {
"invited_count": int,
"bonus_total": float
}
"""
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(
f"[referrals] Код {response.status}, возвращаю None"
)
return None
logger.error(
f"[referrals] Ошибка в запросе: статус {response.status}, причина {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: async def _build_referral_text(bot, user_id: int) -> str:
"""
Текст реферальной программы + статистика, если есть.
"""
me = await bot.get_me() me = await bot.get_me()
bot_username = me.username or "LarkVPN_bot" bot_username = me.username or "LarkVPN_bot"
link = f"https://t.me/{bot_username}?start=ref_{user_id}" 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 = ( text = (
"👥 <b>Реферальная программа</b>\n\n" "👥 <b>Реферальная программа</b>\n\n"
"Зови друзей в Lark VPN и получай бонусы на баланс.\n\n" "Зови друзей в Lark VPN и получай бонусы на баланс.\n\n"
f"🔗 Твоя ссылка:\n<code>{link}</code>\n\n" f"🔗 Твоя ссылка:\n<code>{link}</code>\n\n"
f"👤 Приглашено: {invited_count}\n\n"
) )
invited_line = ""
bonus_line = ""
stats = await call_api("GET", f"/user/{user_id}/referrals")
if isinstance(stats, dict):
invited = stats.get("invited_count")
bonus = stats.get("bonus_total")
if invited is not None:
try:
invited_int = int(invited)
except (TypeError, ValueError):
invited_int = None
if invited_int is not None:
invited_line = f"👤 Приглашено: {invited_int}\n"
if bonus is not None:
try:
bonus_val = float(bonus)
except (TypeError, ValueError):
bonus_val = None
# Строку с бонусами показываем только если реально что-то начислено
if bonus_val is not None and bonus_val > 0:
bonus_line = f"💰 Начислено бонусов: {bonus_val:.2f}\n"
else:
logger.warning(
f"[referrals] Не удалось получить статистику для user_id={user_id}: {stats}"
)
if invited_line:
text += invited_line
if bonus_line:
text += bonus_line
if invited_line or bonus_line:
text += "\n"
text += (
"Бонусы падают автоматически, когда приглашённые пополняют баланс."
)
return text return text
@router.message(Command("referrals")) @router.message(Command("referrals"))
async def referrals_command(message: types.Message): async def referrals_command(message: types.Message):
"""
Команда /referrals — показывает текст реферальной программы.
"""
logger.info(f"Получена команда /referrals от {message.from_user.id}") logger.info(f"Получена команда /referrals от {message.from_user.id}")
try: try:
text = await _build_referral_text(message.bot, message.from_user.id) text = await _build_referral_text(message.bot, message.from_user.id)
@@ -133,9 +47,6 @@ async def referrals_command(message: types.Message):
@router.callback_query(lambda callback: callback.data == "referral") @router.callback_query(lambda callback: callback.data == "referral")
async def referrals_callback(callback: types.CallbackQuery): async def referrals_callback(callback: types.CallbackQuery):
"""
Кнопка «Реферальная программа» (если где-то есть).
"""
try: try:
text = await _build_referral_text( text = await _build_referral_text(
callback.message.bot, callback.message.bot,

View File

@@ -1,62 +1,52 @@
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
import logging
from instences.config import BASE_URL_FASTAPI
import aiohttp
from keyboard.keyboards import main_keyboard
from aiogram.enums.parse_mode import ParseMode 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 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: dict | None = None):
""" """Мини-обёртка для запросов к backend API."""
Выполняет HTTP-запрос к FastAPI.
:param method: HTTP метод (GET, POST, и т.д.)
:param endpoint: конечная точка API
:param data: тело запроса (если необходимо)
:return: JSON-ответ или "ERROR" при неуспехе
"""
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( logger.info(f"API response {resp.status} {resp.reason}")
f"Получен ответ от {url}: статус {response.status}" if resp.status in {200, 201}:
) return await resp.json()
if resp.status in {400, 404}:
if response.status in {200, 201}: try:
result = await response.json() return await resp.json()
logger.debug(f"Ответ JSON: {result}") except Exception:
return result return None
if response.status == 404:
logger.debug(f"Код {response.status}, возвращаю ничего")
return None
logger.error( logger.error(
f"Ошибка в запросе: статус {response.status}, причина {response.reason}" f"Unexpected status {resp.status}: {await resp.text()}"
) )
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 _welcome_text(username: str | None) -> str: def _welcome_text(username: str | None) -> str:
""" """Текст приветствия в /start и в главном меню."""
Текст приветствия в /start и в главном меню.
Имя пока не используем — оставляем сигнатуру на будущее.
"""
return "🥚 Lark Security\n\nВыберите действие из меню ниже." return "🥚 Lark Security\n\nВыберите действие из меню ниже."
def _parse_referrer_id(message: Message) -> int | None: def _parse_referrer_id(message: Message) -> int | None:
""" """Достаём ref_<telegram_id> из /start.
Достаём ref_<telegram_id> из /start.
Примеры: Примеры:
/start /start
/start ref_123456789 /start ref_123456789
@@ -82,86 +72,117 @@ def _parse_referrer_id(message: Message) -> int | None:
@router.message(Command("start")) @router.message(Command("start"))
async def start_command(message: Message): async def start_command(message: Message):
""" """/start с обработкой реферального параметра.
Обработчик команды /start.
Добавлена обработка реферального payload'а ref_<telegram_id>.
"""
logger.info(
f"Получена команда /start от пользователя: "
f"{message.from_user.id} ({message.from_user.username}) | text='{message.text}'"
)
Логика:
- проверяем, есть ли пользователь в БД по telegram_id;
- если нет — создаём через /user/create;
- если есть корректный ref_ и пользователь новый — вызываем
/user/{referrer_id}/add_referral с invited_id = telegram_id.
"""
user_id = message.from_user.id user_id = message.from_user.id
username = message.from_user.username
referrer_id = _parse_referrer_id(message) 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:
# 1. Проверяем/создаём пользователя в базе # 1. Проверяем, есть ли пользователь в БД
user_data = await call_api("GET", f"/user/{user_id}") existing = await call_api("GET", f"/user/{user_id}")
if not user_data: user_exists = existing not in (None, "ERROR")
logger.debug(
"Пользователь не найден в базе, создаем новую запись." # 2. Если пользователя нет — создаём
) if not user_exists:
await call_api( logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
create_result = await call_api(
"POST", "POST",
"/user/create", "/user/create",
{"telegram_id": user_id}, {"telegram_id": user_id},
) )
if create_result == "ERROR":
# 2. Если есть реферер и он не сам пользователь — отправляем инфу в бекенд
if referrer_id and referrer_id != user_id:
payload = {
"referrer_id": referrer_id,
"telegram_id": user_id,
}
logger.info(
f"Отправка данных о реферале в бекенд: {payload}"
)
result = await call_api(
"POST",
"/user/referrals/track",
payload,
)
if result == "ERROR":
logger.error( logger.error(
f"Не удалось зафиксировать реферала в бекенде: {payload}" f"[start] Не удалось создать пользователя {user_id} в БД"
) )
elif referrer_id == user_id:
logger.info(
"Обнаружена попытка самореферала, запрос в бекенд не отправляем."
)
# 3. Приветственное сообщение и главное меню # 3. Обработка рефералки, если параметр есть
logger.debug("Отправка приветственного сообщения пользователю.") 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 = {
"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( await message.answer(
_welcome_text(message.from_user.username), _welcome_text(username),
reply_markup=main_keyboard(), reply_markup=main_keyboard(),
) )
logger.info("Приветственное сообщение отправлено.") logger.info(f"[start] Главное меню отправлено пользователю {user_id}.")
except Exception as e: except Exception as e:
logger.exception( logger.exception(
f"Ошибка при обработке команды /start для пользователя " f"[start] Ошибка при обработке /start для пользователя {user_id}: {e}"
f"{user_id}: {e}"
) )
await message.answer("Произошла ошибка. Попробуйте позже.") await message.answer("Произошла ошибка. Попробуйте позже.")
@router.message(Command("referrals")) @router.message(Command("referrals"))
async def referrals_menu_command(message: Message): async def referrals_menu_command(message: Message):
""" """Команда /referrals из бокового меню Telegram.
Команда /referrals из бокового меню Telegram.
Показывает текст реферальной программы. Показывает текст реферальной программы.
""" """
logger.info( logger.info(
f"Получена команда /referrals от пользователя: " f"[start] Команда /referrals от {message.from_user.id} "
f"{message.from_user.id} ({message.from_user.username})" f"(@{message.from_user.username})"
) )
try: try:
text = await _build_referral_text(message.bot, message.from_user.id) text = await _build_referral_text(message.bot, message.from_user.id)
await message.answer(text, parse_mode=ParseMode.HTML) await message.answer(text, parse_mode=ParseMode.HTML)
logger.info("Реферальная программа отправлена пользователю.") logger.info("[start] Реферальная программа отправлена пользователю.")
except Exception as e: except Exception as e:
logger.exception( logger.exception(
f"Ошибка при обработке команды /referrals для пользователя " f"[start] Ошибка при обработке /referrals для пользователя "
f"{message.from_user.id}: {e}" f"{message.from_user.id}: {e}"
) )
await message.answer("Произошла ошибка. Попробуйте позже.") await message.answer("Произошла ошибка. Попробуйте позже.")
@@ -169,25 +190,26 @@ async def referrals_menu_command(message: Message):
@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 с data="base" — возврат в главное меню."""
Обработчик callback_query с 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
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( await call_api(
"POST", "POST",
"/user/create", "/user/create",
{"telegram_id": callback.from_user.id}, {"telegram_id": user_id},
) )
await callback.message.edit_text( await callback.message.edit_text(
_welcome_text(callback.from_user.username), _welcome_text(username),
reply_markup=main_keyboard(), 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

@@ -129,15 +129,15 @@ async def _show_subscriptions_view(
index = max(0, min(index, len(subs) - 1)) index = max(0, min(index, len(subs) - 1))
sub = subs[index] sub = subs[index]
plan_title = _plan_human_title(sub.get("plan")) plan_title = _plan_human_title(sub.get("plan_name"))
expiry_raw = sub.get("expiry_date") end_raw = sub.get("end_date") # Исправил
date_str = "" date_str = ""
days_left = None days_left = None
status_emoji = "🟢" status_emoji = "🟢"
if expiry_raw: if end_raw:
try: try:
dt = datetime.fromisoformat(expiry_raw) dt = datetime.fromisoformat(end_raw)
date_str = dt.strftime("%d.%m.%Y") date_str = dt.strftime("%d.%m.%Y")
now = datetime.now(dt.tzinfo) if dt.tzinfo else datetime.now() now = datetime.now(dt.tzinfo) if dt.tzinfo else datetime.now()
days_left = (dt.date() - now.date()).days days_left = (dt.date() - now.date()).days
@@ -272,19 +272,22 @@ async def cb_sub_cfg(callback: types.CallbackQuery):
await callback.answer() await callback.answer()
return 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.message.answer(
"Что-то пошло не так при выдаче конфига." "Что-то пошло не так при выдаче конфига."
) )
await callback.answer() await callback.answer()
return return
escaped = escape_markdown_v2(detail)
text = f"Твой конфиг:\n```{escaped}```"
await callback.message.answer( await callback.message.answer(
text, f"Твой конфиг:\n<code>{uri}</code>",
parse_mode=ParseMode.MARKDOWN_V2, parse_mode=ParseMode.HTML,
) )
await callback.answer() await callback.answer()
@@ -310,7 +313,7 @@ async def cb_sub_renew(callback: types.CallbackQuery):
await callback.answer() await callback.answer()
return return
plan_id = target.get("plan") plan_id = target.get("plan_name")
if not plan_id: if not plan_id:
await callback.message.answer("Не удалось определить тариф для продления.") await callback.message.answer("Не удалось определить тариф для продления.")
await callback.answer() await callback.answer()

View File

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