6 Commits

Author SHA1 Message Date
unknown
54530c8521 Исправлен субскрпи 2025-12-06 14:33:32 +03:00
unknown
a4d863c8d3 Исправил пару вещей 2025-12-06 14:19:58 +03:00
unknown
206e5b2fd9 Исправленные саппорты 2025-12-06 14:00:58 +03:00
unknown
ebb523acec помогите.. функции.. бот.. проекты... ларк(( 2025-12-06 13:54:59 +03:00
unknown
8050855b33 исправил одну строку 2025-12-06 13:38:48 +03:00
unknown
26b2bd5fdf Исправил ошибку, надеюсь 2025-12-06 12:16:25 +03:00
5 changed files with 107 additions and 101 deletions

View File

@@ -3,7 +3,7 @@ from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode
import logging
from .start import call_api # используем уже готовый HTTP-хелпер
from .start import call_api
router = Router()
logger = logging.getLogger(__name__)

View File

@@ -1,21 +1,21 @@
from aiogram import Router, types
from aiogram.types import Message, CallbackQuery
from aiogram.filters import Command
from aiogram.enums.parse_mode import ParseMode
import logging
import aiohttp
from aiogram.types import Message, CallbackQuery
from instences.config import BASE_URL_FASTAPI
from keyboard.keyboards import main_keyboard
from .referrals import _build_referral_text
router = Router()
logger = logging.getLogger(__name__)
async def call_api(method: str, endpoint: str, data: dict | None = None):
"""Мини-обёртка для запросов к backend API."""
"""
Мини-обёртка для запросов к backend API.
"""
url = f"{BASE_URL_FASTAPI}{endpoint}"
logger.info(f"API {method} {url} data={data}")
@@ -30,9 +30,7 @@ async def call_api(method: str, endpoint: str, data: dict | None = None):
return await resp.json()
except Exception:
return None
logger.error(
f"Unexpected status {resp.status}: {await resp.text()}"
)
logger.error(f"Unexpected status {resp.status}: {await resp.text()}")
return "ERROR"
except Exception as e:
logger.exception(f"API exception {url}: {e}")
@@ -40,12 +38,15 @@ async def call_api(method: str, endpoint: str, data: dict | None = None):
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
@@ -72,11 +73,12 @@ def _parse_referrer_id(message: Message) -> int | None:
@router.message(Command("start"))
async def start_command(message: Message):
"""/start с обработкой реферального параметра.
"""
/start c обработкой реферального параметра.
Логика:
- проверяем, есть ли пользователь в БД по telegram_id;
- если нет — создаём через /user/create;
- если нет — создаём /user/create;
- если есть корректный ref_ и пользователь новый — вызываем
/user/{referrer_id}/add_referral с invited_id = telegram_id.
"""
@@ -90,11 +92,9 @@ async def start_command(message: Message):
)
try:
# 1. Проверяем, есть ли пользователь в БД
existing = await call_api("GET", f"/user/{user_id}")
user_exists = existing not in (None, "ERROR")
# 2. Если пользователя нет — создаём
user_exists = isinstance(
existing, dict) and existing.get("id") is not None
if not user_exists:
logger.debug(f"[start] Пользователь {user_id} не найден, создаю.")
create_result = await call_api(
@@ -107,9 +107,7 @@ async def start_command(message: Message):
f"[start] Не удалось создать пользователя {user_id} в БД"
)
# 3. Обработка рефералки, если параметр есть
if referrer_id is not None:
# 3.1. Самореферал
if referrer_id == user_id:
logger.info(
f"[start] Пользователь {user_id} попытался зайти "
@@ -119,7 +117,6 @@ async def start_command(message: Message):
"Нельзя переходить по своей же реферальной ссылке."
)
# 3.2. Пользователь уже зарегистрирован в боте
elif user_exists:
logger.info(
f"[start] Пользователь {user_id} уже есть в БД, "
@@ -130,7 +127,6 @@ async def start_command(message: Message):
"реферальная ссылка не сработает."
)
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
else:
payload = {
"invited_id": user_id,
@@ -152,7 +148,6 @@ async def start_command(message: Message):
)
await message.answer("Вы вошли по реферальной ссылке.")
# 4. В любом случае показываем главное меню
await message.answer(
_welcome_text(username),
reply_markup=main_keyboard(),
@@ -166,31 +161,11 @@ async def start_command(message: Message):
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")
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

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

View File

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

View File

@@ -13,7 +13,6 @@ def main_keyboard():
callback_data="profile",
)
)
# ------
builder.row(
InlineKeyboardButton(
text="❔ FAQ ❔",
@@ -102,7 +101,7 @@ def popup_keyboard():
builder.row(
InlineKeyboardButton(
text="🔙 Назад",
callback_data="profile", # назад в профиль
callback_data="profile",
)
)
return builder.as_markup()
@@ -180,7 +179,7 @@ def ticket_keyboard():
builder.row(
InlineKeyboardButton(
text="🔙 Отмена",
callback_data="cancel",
callback_data="cancel_ticket", # БЫЛО "cancel"
)
)
return builder.as_markup()