2 Commits

Author SHA1 Message Date
unknown
1db24ace53 fixes 2025-12-06 19:41:13 +03:00
d3e47a662f Временные изменения 2025-12-05 18:45:25 +03:00
7 changed files with 1255 additions and 1168 deletions

View File

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

View File

@@ -2,51 +2,110 @@ 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 .start import call_api from instences.config import BASE_URL_FASTAPI
router = Router() router = Router()
logger = logging.getLogger(__name__) 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: 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_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 return text
@router.message(Command("referrals")) @router.message(Command("referrals"))
async def referrals_command(message: types.Message): async def referrals_command(message: types.Message):
logger.info(f"Получена команда /referrals от {message.from_user.id}") """
Команда /referrals — показывает текст реферальной программы.
"""
logger.info(f"[referrals] Команда /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)
await message.answer(text, parse_mode=ParseMode.HTML) await message.answer(text, parse_mode=ParseMode.HTML)
except Exception as e: except Exception as e:
logger.exception(f"Ошибка в обработчике /referrals: {e}") logger.exception(f"[referrals] Ошибка в обработчике /referrals: {e}")
await message.answer("Произошла ошибка. Попробуй позже.") await message.answer("Произошла ошибка. Попробуй позже.")
@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,
@@ -57,7 +116,8 @@ async def referrals_callback(callback: types.CallbackQuery):
parse_mode=ParseMode.HTML, parse_mode=ParseMode.HTML,
) )
except Exception as e: except Exception as e:
logger.exception(f"Ошибка в обработчике callback 'referral': {e}") logger.exception(
f"[referrals] Ошибка в обработчике callback 'referral': {e}")
await callback.message.answer("Произошла ошибка. Попробуй позже.") await callback.message.answer("Произошла ошибка. Попробуй позже.")
finally: finally:
await callback.answer() await callback.answer()

View File

@@ -1,39 +1,51 @@
from aiogram import Router, types from aiogram import Router, types
from aiogram.types import Message, CallbackQuery
from aiogram.filters import Command from aiogram.filters import Command
from aiogram.types import Message, CallbackQuery
from aiogram.enums.parse_mode import ParseMode from aiogram.enums.parse_mode import ParseMode
import logging import logging
import aiohttp import aiohttp
from instences.config import BASE_URL_FASTAPI 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: str, endpoint: str, data: dict | None = None): async def call_api(method: str, endpoint: str, data=None):
""" """
Мини-обёртка для запросов к backend API. Выполняет HTTP-запрос к FastAPI.
Возвращает:
- dict при 200/201
- None при 404
- "ERROR" при остальных ошибках
""" """
url = f"{BASE_URL_FASTAPI}{endpoint}" url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"API {method} {url} data={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 resp: async with session.request(method, url, json=data) as response:
logger.info(f"API response {resp.status} {resp.reason}") logger.info(
if resp.status in {200, 201}: f"[start] Ответ от {url}: статус {response.status}"
return await resp.json() )
if resp.status in {400, 404}:
try: if response.status in {200, 201}:
return await resp.json() result = await response.json()
except Exception: logger.debug(f"[start] Ответ JSON: {result}")
return result
if response.status == 404:
logger.debug("[start] Получен 404, возвращаю None")
return None return None
logger.error(f"Unexpected status {resp.status}: {await resp.text()}")
logger.error(
f"[start] Ошибка в запросе: статус {response.status}, "
f"причина {response.reason}"
)
return "ERROR" return "ERROR"
except Exception as e: except Exception as e:
logger.exception(f"API exception {url}: {e}") logger.exception(f"[start] Исключение при запросе к {url}: {e}")
return "ERROR" return "ERROR"
@@ -47,7 +59,6 @@ def _welcome_text(username: str | None) -> str:
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
@@ -75,12 +86,6 @@ def _parse_referrer_id(message: Message) -> int | None:
async def start_command(message: Message): async def start_command(message: Message):
""" """
/start c обработкой реферального параметра. /start c обработкой реферального параметра.
Логика:
- проверяем, есть ли пользователь в БД по 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 username = message.from_user.username
@@ -92,9 +97,11 @@ async def start_command(message: Message):
) )
try: try:
# 1. Проверяем, есть ли пользователь в БД
existing = await call_api("GET", f"/user/{user_id}") existing = await call_api("GET", f"/user/{user_id}")
user_exists = isinstance( user_exists = existing not in (None, "ERROR")
existing, dict) and existing.get("id") is not None
# 2. Если пользователя нет — создаём
if not user_exists: if not user_exists:
logger.debug(f"[start] Пользователь {user_id} не найден, создаю.") logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
create_result = await call_api( create_result = await call_api(
@@ -107,7 +114,9 @@ async def start_command(message: Message):
f"[start] Не удалось создать пользователя {user_id} в БД" f"[start] Не удалось создать пользователя {user_id} в БД"
) )
# 3. Обработка рефералки, если параметр есть
if referrer_id is not None: if referrer_id is not None:
# 3.1. Самореферал
if referrer_id == user_id: if referrer_id == user_id:
logger.info( logger.info(
f"[start] Пользователь {user_id} попытался зайти " f"[start] Пользователь {user_id} попытался зайти "
@@ -117,6 +126,7 @@ async def start_command(message: Message):
"Нельзя переходить по своей же реферальной ссылке." "Нельзя переходить по своей же реферальной ссылке."
) )
# 3.2. Пользователь уже зарегистрирован в боте
elif user_exists: elif user_exists:
logger.info( logger.info(
f"[start] Пользователь {user_id} уже есть в БД, " f"[start] Пользователь {user_id} уже есть в БД, "
@@ -127,27 +137,29 @@ async def start_command(message: Message):
"реферальная ссылка не сработает." "реферальная ссылка не сработает."
) )
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
else: else:
payload = { payload = {
"invited_id": user_id, "referrer_id": referrer_id,
"telegram_id": user_id,
} }
logger.info( logger.info(
f"[start] Фиксирую реферала в бекенде: " f"[start] Фиксирую реферала в бекенде: {payload}"
f"referrer_id={referrer_id}, payload={payload}"
) )
result = await call_api( result = await call_api(
"POST", "POST",
f"/user/{referrer_id}/add_referral", "/user/referrals/track",
payload, payload,
) )
if result == "ERROR": if result == "ERROR":
logger.error( logger.error(
f"[start] Ошибка при фиксации реферала через " f"[start] Ошибка при фиксации реферала: {payload}"
f"/user/{referrer_id}/add_referral: referrer={referrer_id}, " )
f"invited={user_id}" await message.answer(
"Вы вошли по реферальной ссылке."
) )
await message.answer("Вы вошли по реферальной ссылке.")
# 4. В любом случае показываем главное меню
await message.answer( await message.answer(
_welcome_text(username), _welcome_text(username),
reply_markup=main_keyboard(), reply_markup=main_keyboard(),
@@ -161,6 +173,28 @@ async def start_command(message: Message):
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):
""" """

View File

@@ -71,15 +71,19 @@ def _plan_human_title(plan: str) -> str:
async def _fetch_user_and_subs(telegram_id: int): async def _fetch_user_and_subs(telegram_id: int):
"""
Берём пользователя и список его подписок с backend.
"""
user = await call_api("GET", f"/user/{telegram_id}") user = await call_api("GET", f"/user/{telegram_id}")
if not isinstance(user, dict): if not user or not isinstance(user, dict):
return None, []
user_id = user.get("id") or user.get("telegram_id") or user.get("user_id")
if not user_id:
return None, [] return None, []
user_id = user["telegram_id"]
subs = await call_api("GET", f"/subscriptions/{user_id}") subs = await call_api("GET", f"/subscriptions/{user_id}")
if subs == "ERROR" or not isinstance(subs, list): if subs == "ERROR" or not isinstance(subs, list):
subs = [] subs = []
return user, subs return user, subs
@@ -125,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_name")) plan_title = _plan_human_title(sub.get("plan"))
end_raw = sub.get("end_date") # Исправил expiry_raw = sub.get("expiry_date")
date_str = "" date_str = ""
days_left = None days_left = None
status_emoji = "🟢" status_emoji = "🟢"
if end_raw: if expiry_raw:
try: try:
dt = datetime.fromisoformat(end_raw) dt = datetime.fromisoformat(expiry_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
@@ -268,28 +272,29 @@ async def cb_sub_cfg(callback: types.CallbackQuery):
await callback.answer() await callback.answer()
return return
uri = None if not isinstance(detail, str) or not detail:
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(
f"Твой конфиг:\n<code>{uri}</code>", text,
parse_mode=ParseMode.HTML, parse_mode=ParseMode.MARKDOWN_V2,
) )
await callback.answer() await callback.answer()
@router.callback_query(lambda c: c.data.startswith("sub_renew:")) @router.callback_query(lambda c: c.data.startswith("sub_renew:"))
async def cb_sub_renew(callback: types.CallbackQuery): async def cb_sub_renew(callback: types.CallbackQuery):
"""
Кнопка 'Продлить' в карточке подписки.
По факту — повторный /subscription/buy с тем же планом.
"""
sub_id = callback.data.split(":", maxsplit=1)[ sub_id = callback.data.split(":", maxsplit=1)[
1] if ":" in callback.data else None 1] if ":" in callback.data else None
user, subs = await _fetch_user_and_subs(callback.from_user.id) user, subs = await _fetch_user_and_subs(callback.from_user.id)
@@ -305,8 +310,8 @@ async def cb_sub_renew(callback: types.CallbackQuery):
await callback.answer() await callback.answer()
return return
plan_name = target.get("plan_name") plan_id = target.get("plan")
if not plan_name: if not plan_id:
await callback.message.answer("Не удалось определить тариф для продления.") await callback.message.answer("Не удалось определить тариф для продления.")
await callback.answer() await callback.answer()
return return
@@ -314,9 +319,10 @@ async def cb_sub_renew(callback: types.CallbackQuery):
result = await call_api( result = await call_api(
"POST", "POST",
"/subscription/buy", "/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_name}, {"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
) )
# Ошибки backend (detail)
if result == "ERROR" or not isinstance(result, dict): if result == "ERROR" or not isinstance(result, dict):
await callback.message.answer("Ошибка при продлении. Попробуй позже.") await callback.message.answer("Ошибка при продлении. Попробуй позже.")
await callback.answer() await callback.answer()
@@ -336,16 +342,19 @@ async def cb_sub_renew(callback: types.CallbackQuery):
await callback.answer() await callback.answer()
return return
# Успех — backend вернёт message с URI
uri = result.get("message") uri = result.get("message")
if isinstance(uri, str) and uri: if isinstance(uri, str) and uri:
escaped_uri = escape_markdown_v2(uri)
await callback.message.answer( await callback.message.answer(
"Подписка успешно продлена.\n" "Подписка успешно продлена.\n"
f"Твой конфиг:\n<code>{uri}</code>", f"Твой конфиг:\n```{escaped_uri}```",
parse_mode=ParseMode.HTML, parse_mode=ParseMode.MARKDOWN_V2,
) )
else: else:
await callback.message.answer("Подписка успешно продлена.") await callback.message.answer("Подписка успешно продлена.")
# Перерисовать карточку (обновлённый список)
await _show_subscriptions_view( await _show_subscriptions_view(
msg=callback.message, msg=callback.message,
telegram_id=callback.from_user.id, telegram_id=callback.from_user.id,
@@ -354,8 +363,8 @@ async def cb_sub_renew(callback: types.CallbackQuery):
) )
await callback.answer() await callback.answer()
# ===== Меню покупки подписки (Basic / Pro / Family) =====
# ===== Меню покупки подписки (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):
@@ -448,10 +457,13 @@ async def lark_tariff_callback_handler(callback: types.CallbackQuery):
@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(":", maxsplit=1)[1] data = callback.data.split(":", maxsplit=1)[1]
name, classif, amount_str = data.split("_") name, classif, amount_str = data.split("_")
plan_name = f"{name}_{classif}_{amount_str}" plan_id = f"{name}_{classif}_{amount_str}"
except Exception: except Exception:
await callback.message.edit_text("Некорректные данные тарифа.") await callback.message.edit_text("Некорректные данные тарифа.")
await callback.answer() await callback.answer()
@@ -460,7 +472,7 @@ async def confirm_callback_handler(callback: types.CallbackQuery):
result = await call_api( result = await call_api(
"POST", "POST",
"/subscription/buy", "/subscription/buy",
{"telegram_id": str(callback.from_user.id), "plan_name": plan_name}, {"telegram_id": str(callback.from_user.id), "plan_name": plan_id},
) )
if result == "ERROR" or not isinstance(result, dict): if result == "ERROR" or not isinstance(result, dict):
@@ -487,10 +499,12 @@ async def confirm_callback_handler(callback: types.CallbackQuery):
uri = result.get("message", "") uri = result.get("message", "")
if uri: if uri:
escaped_uri = escape_markdown_v2(uri)
answer_text = f"Подписка успешно оформлена!\n\nТвой конфиг:\n```{escaped_uri}```"
await callback.message.edit_text("Подписка успешно оформлена.") await callback.message.edit_text("Подписка успешно оформлена.")
await callback.message.answer( await callback.message.answer(
f"Подписка успешно оформлена!\n\nТвой конфиг:\n<code>{uri}</code>", answer_text,
parse_mode=ParseMode.HTML, parse_mode=ParseMode.MARKDOWN_V2,
) )
else: else:
await callback.message.edit_text("Подписка успешно оформлена.") await callback.message.edit_text("Подписка успешно оформлена.")

View File

@@ -1,26 +1,20 @@
from aiogram import Router, types from aiogram import Router, types
from aiogram.filters import Command from aiogram.filters import Command
import logging import logging
from datetime import datetime
from instences.config import BASE_URL_FASTAPI from instences.config import BASE_URL_FASTAPI
import aiohttp import aiohttp
from aiogram.fsm.context import FSMContext from aiogram.fsm.context import FSMContext
from aiogram.fsm.state import State, StatesGroup from aiogram.fsm.state import State, StatesGroup
from keyboard.keyboards import ( from keyboard.keyboards import faq_keyboard, sup_keyboard, ticket_list_keyboard, ticket_keyboard
faq_keyboard,
sup_keyboard,
ticket_list_keyboard,
ticket_keyboard,
)
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
router = Router() router = Router()
class TicketState(StatesGroup): class TicketState(StatesGroup):
subject = State() subject = State()
message = State() message = State()
async def call_api(method, endpoint, data=None): async def call_api(method, endpoint, data=None):
""" """
Выполняет HTTP-запрос к FastAPI. Выполняет HTTP-запрос к FastAPI.
@@ -31,8 +25,7 @@ async def call_api(method, endpoint, data=None):
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( logger.info(f"Получен ответ от {url}: статус {response.status}")
f"Получен ответ от {url}: статус {response.status}")
if response.status in {200, 201}: if response.status in {200, 201}:
result = await response.json() result = await response.json()
@@ -41,15 +34,12 @@ async def call_api(method, endpoint, data=None):
if response.status == 404: if response.status == 404:
logger.debug(f"Код {response.status}, возвращаю ничего") logger.debug(f"Код {response.status}, возвращаю ничего")
return None return None
logger.error( logger.error(f"Ошибка в запросе: статус {response.status}, причина {response.reason}")
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"Исключение при выполнении запроса к {url}: {e}")
return "ERROR" return "ERROR"
@router.callback_query(lambda callback: callback.data == "faq") @router.callback_query(lambda callback: callback.data == "faq")
async def faq_callback_handler(callback: types.CallbackQuery): async def faq_callback_handler(callback: types.CallbackQuery):
""" """
@@ -57,64 +47,46 @@ async def faq_callback_handler(callback: types.CallbackQuery):
""" """
await callback.message.edit_text( await callback.message.edit_text(
"FAQ YOU", "FAQ YOU",
reply_markup=faq_keyboard(), reply_markup=faq_keyboard()
) )
await callback.answer()
@router.message(Command("support")) @router.message(Command("support"))
async def supp(message: types.Message): async def supp(message: types.Message):
""" """
Меню сапп системы (через команду). Меню сапп системы
""" """
await message.answer( await message.answer(
"Добро пожаловать в саппорт систему!", "Добро пожаловать в саппорт систему!",
reply_markup=sup_keyboard(), reply_markup=sup_keyboard()
) )
@router.callback_query(lambda callback: callback.data == "main_sup") @router.callback_query(lambda callback: callback.data == "main_sup")
async def supp_callback(callback: types.CallbackQuery): async def supp_callback(callback: types.CallbackQuery):
""" """
Меню сапп системы (callback-версия). Меню сапп системы (callback версия)
""" """
await callback.message.edit_text( await callback.message.answer(
"Добро пожаловать в саппорт систему!", "Добро пожаловать в саппорт систему!",
reply_markup=sup_keyboard(), reply_markup=sup_keyboard()
) )
await callback.answer()
@router.callback_query(lambda callback: callback.data == "my_tickets") @router.callback_query(lambda callback: callback.data == "my_tickets")
async def list_tickets_callback(callback: types.CallbackQuery): async def list_tickets_callback(callback: types.CallbackQuery):
user_id = callback.from_user.id user_id = callback.from_user.id
user_data = await call_api("GET", f"/user/{user_id}") user_data = await call_api("GET", f"/user/{user_id}")
if not user_data: if not user_data:
user_data = await call_api( user_data = await call_api("POST", f"/user/create", {"telegram_id": f"{user_id}"})
"POST",
"/user/create",
{"telegram_id": f"{user_id}"},
)
tickets = await call_api( tickets = await call_api("GET", f"/support/tickets?user_id={user_data['id']}")
"GET",
f"/support/tickets?user_id={user_data['id']}",
)
if tickets == "ERROR" or not tickets: if tickets == "ERROR" or not tickets:
await callback.message.edit_text( await callback.message.edit_text("У вас нет тикетов.", reply_markup=sup_keyboard())
"У вас нет тикетов.",
reply_markup=sup_keyboard(),
)
await callback.answer()
return return
await callback.message.edit_text( await callback.message.edit_text(
"Ваши тикеты:", "Ваши тикеты:",
reply_markup=ticket_list_keyboard(tickets), reply_markup=ticket_list_keyboard(tickets)
) )
await callback.answer()
@router.callback_query(lambda callback: callback.data == "make_ticket") @router.callback_query(lambda callback: callback.data == "make_ticket")
async def start_ticket_creation(callback: types.CallbackQuery, state: FSMContext): async def start_ticket_creation(callback: types.CallbackQuery, state: FSMContext):
@@ -123,11 +95,9 @@ async def start_ticket_creation(callback: types.CallbackQuery, state: FSMContext
""" """
await callback.message.answer( await callback.message.answer(
"Введите тему тикета (или нажмите 'Отмена', чтобы выйти):", "Введите тему тикета (или нажмите 'Отмена', чтобы выйти):",
reply_markup=ticket_keyboard(), reply_markup=ticket_keyboard()
) )
await state.set_state(TicketState.subject) await state.set_state(TicketState.subject)
await callback.answer()
@router.message() @router.message()
async def handle_ticket_input(message: types.Message, state: FSMContext): async def handle_ticket_input(message: types.Message, state: FSMContext):
@@ -137,10 +107,7 @@ async def handle_ticket_input(message: types.Message, state: FSMContext):
current_state = await state.get_state() current_state = await state.get_state()
if current_state == TicketState.subject: if current_state == TicketState.subject:
await state.update_data(subject=message.text) await state.update_data(subject=message.text)
await message.answer( await message.answer("Введите описание проблемы:", reply_markup=ticket_keyboard())
"Введите описание проблемы:",
reply_markup=ticket_keyboard(),
)
await state.set_state(TicketState.message) await state.set_state(TicketState.message)
elif current_state == TicketState.message: elif current_state == TicketState.message:
@@ -150,23 +117,14 @@ async def handle_ticket_input(message: types.Message, state: FSMContext):
await create_ticket(message, subject, message_text, state) await create_ticket(message, subject, message_text, state)
await state.clear() await state.clear()
async def create_ticket(message: types.Message, subject: str, message_text: str, state: FSMContext):
async def create_ticket(
message: types.Message,
subject: str,
message_text: str,
state: FSMContext,
):
""" """
Отправляет запрос на создание тикета через FastAPI. Отправляет запрос на создание тикета через FastAPI.
""" """
user_id = message.from_user.id user_id = message.from_user.id
try: try:
logger.info( logger.info(f"Создание тикета для пользователя {user_id}: Тема - {subject}, Сообщение - {message_text}")
f"Создание тикета для пользователя {user_id}: "
f"Тема - {subject}, Сообщение - {message_text}"
)
user_data = await call_api("GET", f"/user/{user_id}") user_data = await call_api("GET", f"/user/{user_id}")
if not user_data: if not user_data:
@@ -176,12 +134,12 @@ async def create_ticket(
ticket_data = await call_api( ticket_data = await call_api(
"POST", "POST",
f"/support/tickets?user_id={user_data['id']}", f"/support/tickets?user_id={user_data['id']}",
data={"subject": subject, "message": message_text}, data={"subject": subject, "message": message_text}
) )
if ticket_data != "ERROR": if ticket_data != "ERROR":
await message.answer( await message.answer(
"✅ Тикет успешно создан!\n" f"✅ Тикет успешно создан!\n"
f"📌 Тема: {ticket_data['subject']}\n" f"📌 Тема: {ticket_data['subject']}\n"
f"📊 Статус: {ticket_data['status']}\n" f"📊 Статус: {ticket_data['status']}\n"
f"📅 Дата создания: {ticket_data['created_at']}" f"📅 Дата создания: {ticket_data['created_at']}"
@@ -190,20 +148,13 @@ async def create_ticket(
await message.answer("❌ Ошибка создания тикета. Попробуйте позже.") await message.answer("❌ Ошибка создания тикета. Попробуйте позже.")
except Exception as e: except Exception as e:
logger.exception( logger.exception(f"Ошибка при создании тикета для пользователя {user_id}: {e}")
f"Ошибка при создании тикета для пользователя {user_id}: {e}"
)
await message.answer("❌ Произошла ошибка при создании тикета.") await message.answer("❌ Произошла ошибка при создании тикета.")
@router.callback_query(lambda callback: callback.data == "cancel_ticket") @router.callback_query(lambda callback: callback.data == "cancel_ticket")
async def cancel_ticket_creation(callback: types.CallbackQuery, state: FSMContext): async def cancel_ticket_creation(callback: types.CallbackQuery, state: FSMContext):
""" """
Отмена создания тикета. Отмена создания тикета.
""" """
await state.clear() await state.clear()
await callback.message.answer( await callback.message.answer("Создание тикета отменено.", reply_markup=types.ReplyKeyboardRemove())
"Создание тикета отменено.",
reply_markup=types.ReplyKeyboardRemove(),
)
await callback.answer()

View File

@@ -13,6 +13,7 @@ def main_keyboard():
callback_data="profile", callback_data="profile",
) )
) )
# ------
builder.row( builder.row(
InlineKeyboardButton( InlineKeyboardButton(
text="❔ FAQ ❔", text="❔ FAQ ❔",
@@ -101,7 +102,7 @@ def popup_keyboard():
builder.row( builder.row(
InlineKeyboardButton( InlineKeyboardButton(
text="🔙 Назад", text="🔙 Назад",
callback_data="profile", callback_data="profile", # назад в профиль
) )
) )
return builder.as_markup() return builder.as_markup()
@@ -120,7 +121,7 @@ def payment_methods_keyboard(amount: int):
) )
builder.row( builder.row(
InlineKeyboardButton( InlineKeyboardButton(
text="💵 СБП", text="💵 YooKassa",
callback_data=f"method_ykassa_{amount}", callback_data=f"method_ykassa_{amount}",
) )
) )
@@ -179,7 +180,7 @@ def ticket_keyboard():
builder.row( builder.row(
InlineKeyboardButton( InlineKeyboardButton(
text="🔙 Отмена", text="🔙 Отмена",
callback_data="cancel_ticket", # БЫЛО "cancel" callback_data="cancel",
) )
) )
return builder.as_markup() return builder.as_markup()
@@ -316,11 +317,10 @@ 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,22 +339,28 @@ 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="base", # было "profile" изменил callback_data="profile",
) )
) )
return builder.as_markup() return builder.as_markup()