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
6 changed files with 1091 additions and 1184 deletions

View File

@@ -1,4 +1,4 @@
# Профиль. последнее изменениеы
# Профиль. последнее изменение 24.11.2025
from aiogram import Router, types, F
from aiogram.types import (
@@ -79,6 +79,7 @@ async def profile_callback_handler(callback: CallbackQuery):
await callback.answer()
return
# Последняя подписка пользователя
sub_data = await call_api(
"GET", f"/subscription/{user_data['telegram_id']}/last"
)
@@ -92,11 +93,12 @@ async def profile_callback_handler(callback: CallbackQuery):
sub_status = "⚫ Нет активных"
if sub_data:
expiry_date = sub_data.get("end_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:
@@ -253,8 +255,10 @@ async def method_stars_handler(callback: CallbackQuery):
await callback.answer()
return
# Внутренний payload, чтобы при успешной оплате понять, что это пополнение баланса
payload = f"stars_topup:{callback.from_user.id}:{amount_rub}"
# 1 ₽ == 1 Star, если нужна другая конвертация — меняй тут
stars_amount = amount_rub
prices = [
@@ -292,11 +296,14 @@ async def method_ykassa_handler(callback: CallbackQuery):
"""
amount = callback.data.split("_")[-1]
# Сразу отвечаем на callback, чтобы Telegram не считал запрос "старым"
try:
await callback.answer()
except Exception:
# Если по какой-то причине уже отвечали — просто игнорируем
pass
# Формируем URL с query parameters вместо JSON body
endpoint = (
f"/billing/payments/init?"
f"user_id={callback.from_user.id}&amount={float(amount)}&provider=yookassa"
@@ -304,14 +311,17 @@ async def method_ykassa_handler(callback: CallbackQuery):
logger.info(f"Отправка запроса на инициализацию платежа: {endpoint}")
# Отправляем POST запрос с пустым телом (параметры в URL)
result = await call_api("POST", endpoint, None, "http://billing:8000")
# Биллинг вообще не ответил/упал
if result == "ERROR" or not isinstance(result, dict):
await callback.message.edit_text(
"❌ Произошла ошибка при создании платежа. Попробуйте позже."
)
return
# Биллинг вернул ошибку (success = False)
if not result.get("success", False):
error_msg = (
result.get("error")
@@ -352,6 +362,7 @@ async def method_crypto_handler(callback: CallbackQuery):
"""
amount = callback.data.split("_")[-1]
# Сразу отвечаем на callback, чтобы избежать таймаута
try:
await callback.answer()
except Exception:
@@ -405,39 +416,6 @@ async def method_crypto_handler(callback: CallbackQuery):
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):
"""

View File

@@ -2,110 +2,51 @@ 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
from .start import call_api # используем уже готовый HTTP-хелпер
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}"
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"
f"👤 Приглашено: {invited_count}\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}")
logger.info(f"Получена команда /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}")
logger.exception(f"Ошибка в обработчике /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,
@@ -116,8 +57,7 @@ async def referrals_callback(callback: types.CallbackQuery):
parse_mode=ParseMode.HTML,
)
except Exception as e:
logger.exception(
f"[referrals] Ошибка в обработчике callback 'referral': {e}")
logger.exception(f"Ошибка в обработчике callback 'referral': {e}")
await callback.message.answer("Произошла ошибка. Попробуй позже.")
finally:
await callback.answer()

View File

@@ -1,10 +1,11 @@
from aiogram import Router, types
from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery
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
@@ -13,52 +14,39 @@ router = Router()
logger = logging.getLogger(__name__)
async def call_api(method: str, endpoint: str, data=None):
"""
Выполняет HTTP-запрос к FastAPI.
Возвращает:
- dict при 200/201
- None при 404
- "ERROR" при остальных ошибках
"""
async def call_api(method: str, endpoint: str, data: dict | None = None):
"""Мини-обёртка для запросов к backend API."""
url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"[start] Запрос: {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"[start] Ответ от {url}: статус {response.status}"
)
if response.status in {200, 201}:
result = await response.json()
logger.debug(f"[start] Ответ JSON: {result}")
return result
if response.status == 404:
logger.debug("[start] Получен 404, возвращаю None")
return None
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"[start] Ошибка в запросе: статус {response.status}, "
f"причина {response.reason}"
f"Unexpected status {resp.status}: {await resp.text()}"
)
return "ERROR"
except Exception as e:
logger.exception(f"[start] Исключение при запросе к {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.
"""Достаём ref_<telegram_id> из /start.
Примеры:
/start
/start ref_123456789
@@ -84,8 +72,13 @@ def _parse_referrer_id(message: Message) -> int | None:
@router.message(Command("start"))
async def start_command(message: Message):
"""
/start c обработкой реферального параметра.
"""/start с обработкой реферального параметра.
Логика:
- проверяем, есть ли пользователь в БД по telegram_id;
- если нет — создаём через /user/create;
- если есть корректный ref_ и пользователь новый — вызываем
/user/{referrer_id}/add_referral с invited_id = telegram_id.
"""
user_id = message.from_user.id
username = message.from_user.username
@@ -140,24 +133,24 @@ async def start_command(message: Message):
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
else:
payload = {
"referrer_id": referrer_id,
"telegram_id": user_id,
"invited_id": user_id,
}
logger.info(
f"[start] Фиксирую реферала в бекенде: {payload}"
f"[start] Фиксирую реферала в бекенде: "
f"referrer_id={referrer_id}, payload={payload}"
)
result = await call_api(
"POST",
"/user/referrals/track",
f"/user/{referrer_id}/add_referral",
payload,
)
if result == "ERROR":
logger.error(
f"[start] Ошибка при фиксации реферала: {payload}"
f"[start] Ошибка при фиксации реферала через "
f"/user/{referrer_id}/add_referral: referrer={referrer_id}, "
f"invited={user_id}"
)
await message.answer(
"Вы вошли по реферальной ссылке."
)
await message.answer("Вы вошли по реферальной ссылке.")
# 4. В любом случае показываем главное меню
await message.answer(
@@ -175,8 +168,8 @@ async def start_command(message: Message):
@router.message(Command("referrals"))
async def referrals_menu_command(message: Message):
"""
Команда /referrals из бокового меню Telegram.
"""Команда /referrals из бокового меню Telegram.
Показывает текст реферальной программы.
"""
logger.info(
@@ -197,9 +190,7 @@ async def referrals_menu_command(message: Message):
@router.callback_query(lambda callback: callback.data == "base")
async def start_callback_handler(callback: CallbackQuery):
"""
Callback с data="base" — возврат в главное меню.
"""
"""Callback с data="base" — возврат в главное меню."""
try:
user_id = callback.from_user.id
username = callback.from_user.username

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()
@@ -319,7 +322,7 @@ async def cb_sub_renew(callback: types.CallbackQuery):
result = await call_api(
"POST",
"/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
{"telegram_id": str(callback.from_user.id), "plan_id": plan_id},
)
# Ошибки backend (detail)
@@ -472,7 +475,7 @@ async def confirm_callback_handler(callback: types.CallbackQuery):
result = await call_api(
"POST",
"/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
{"telegram_id": str(callback.from_user.id), "plan_id": plan_id},
)
if result == "ERROR" or not isinstance(result, dict):

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()