Исправил ошибку, надеюсь

This commit is contained in:
unknown
2025-12-06 12:16:25 +03:00
parent ead48389b7
commit 26b2bd5fdf
2 changed files with 17 additions and 42 deletions

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. Если пользователя нет — создаём
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

@@ -78,7 +78,7 @@ async def _fetch_user_and_subs(telegram_id: int):
if not user or not isinstance(user, dict):
return None, []
user_id = user["telegram_id"]
user_id = user["id"]
subs = await call_api("GET", f"/subscriptions/{user_id}")
if subs == "ERROR" or not isinstance(subs, list):