Vladimir/Telegram-bot-old
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Compare commits

base: Vladimir:front-misha-ui
Branches Tags
Vladimir:main Vladimir:misha-ui Vladimir:front-misha-ui Vladimir:temp Vladimir:feature/billing-integration
..
compare: Vladimir:temp
Branches Tags
Vladimir:misha-ui Vladimir:front-misha-ui Vladimir:temp Vladimir:feature/billing-integration Vladimir:main

1 Commits
front-mish ... temp

Author SHA1 Message Date
Vladimir
d3e47a662f Временные изменения 2025-12-05 18:45:25 +03:00
7 changed files with 1261 additions and 1168 deletions
Expand all files Collapse all files

32
handlers/__init__.py
View File

@@ -1,16 +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 from .referrals import router as referrals_router
# Список всех роутеров бота # Список всех роутеров бота
routers = [ routers = [
start_router, start_router,
profile_router, profile_router,
subscriptions_router, subscriptions_router,
support_router, support_router,
referrals_router, referrals_router,
] ]
__all__ = ["routers"] __all__ = ["routers"]

1046
handlers/profile.py
View File

File diff suppressed because it is too large Load Diff

186
handlers/referrals.py
View File

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

98
handlers/start.py
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 None return result
logger.error(f"Unexpected status {resp.status}: {await resp.text()}") if response.status == 404:
logger.debug("[start] Получен 404, возвращаю None")
return None
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):
""" """

68
handlers/subscriptions.py
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("Подписка успешно оформлена.")

95
handlers/support.py
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,80 +34,59 @@ 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):
""" """
Обработчик callback_query с data="faq". Обработчик callback_query с data="faq".
""" """
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()

904
keyboard/keyboards.py
View File

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