Исправил ошибку, надеюсь
This commit is contained in:
@@ -1,21 +1,21 @@
|
|||||||
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.enums.parse_mode import ParseMode
|
from aiogram.enums.parse_mode import ParseMode
|
||||||
import logging
|
import logging
|
||||||
import aiohttp
|
import aiohttp
|
||||||
|
|
||||||
from aiogram.types import Message, CallbackQuery
|
|
||||||
|
|
||||||
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: dict | None = None):
|
||||||
"""Мини-обёртка для запросов к backend API."""
|
"""
|
||||||
|
Мини-обёртка для запросов к backend API.
|
||||||
|
"""
|
||||||
url = f"{BASE_URL_FASTAPI}{endpoint}"
|
url = f"{BASE_URL_FASTAPI}{endpoint}"
|
||||||
logger.info(f"API {method} {url} data={data}")
|
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()
|
return await resp.json()
|
||||||
except Exception:
|
except Exception:
|
||||||
return None
|
return None
|
||||||
logger.error(
|
logger.error(f"Unexpected status {resp.status}: {await resp.text()}")
|
||||||
f"Unexpected status {resp.status}: {await resp.text()}"
|
|
||||||
)
|
|
||||||
return "ERROR"
|
return "ERROR"
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.exception(f"API exception {url}: {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:
|
def _welcome_text(username: str | None) -> str:
|
||||||
"""Текст приветствия в /start и в главном меню."""
|
"""
|
||||||
|
Текст приветствия в /start и в главном меню.
|
||||||
|
"""
|
||||||
return "🥚 Lark Security\n\nВыберите действие из меню ниже."
|
return "🥚 Lark Security\n\nВыберите действие из меню ниже."
|
||||||
|
|
||||||
|
|
||||||
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
|
||||||
@@ -72,11 +73,12 @@ def _parse_referrer_id(message: Message) -> int | None:
|
|||||||
|
|
||||||
@router.message(Command("start"))
|
@router.message(Command("start"))
|
||||||
async def start_command(message: Message):
|
async def start_command(message: Message):
|
||||||
"""/start с обработкой реферального параметра.
|
"""
|
||||||
|
/start c обработкой реферального параметра.
|
||||||
|
|
||||||
Логика:
|
Логика:
|
||||||
- проверяем, есть ли пользователь в БД по telegram_id;
|
- проверяем, есть ли пользователь в БД по telegram_id;
|
||||||
- если нет — создаём через /user/create;
|
- если нет — создаём /user/create;
|
||||||
- если есть корректный ref_ и пользователь новый — вызываем
|
- если есть корректный ref_ и пользователь новый — вызываем
|
||||||
/user/{referrer_id}/add_referral с invited_id = telegram_id.
|
/user/{referrer_id}/add_referral с invited_id = telegram_id.
|
||||||
"""
|
"""
|
||||||
@@ -90,11 +92,9 @@ 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 = existing not in (None, "ERROR")
|
user_exists = existing not in (None, "ERROR")
|
||||||
|
|
||||||
# 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,9 +107,7 @@ 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} попытался зайти "
|
||||||
@@ -119,7 +117,6 @@ 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} уже есть в БД, "
|
||||||
@@ -130,7 +127,6 @@ async def start_command(message: Message):
|
|||||||
"реферальная ссылка не сработает."
|
"реферальная ссылка не сработает."
|
||||||
)
|
)
|
||||||
|
|
||||||
# 3.3. Новый пользователь + чужая рефералка → регистрируем реферал
|
|
||||||
else:
|
else:
|
||||||
payload = {
|
payload = {
|
||||||
"invited_id": user_id,
|
"invited_id": user_id,
|
||||||
@@ -152,7 +148,6 @@ async def start_command(message: Message):
|
|||||||
)
|
)
|
||||||
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(),
|
||||||
@@ -166,31 +161,11 @@ 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):
|
||||||
"""Callback с data="base" — возврат в главное меню."""
|
"""
|
||||||
|
Callback с data="base" — возврат в главное меню.
|
||||||
|
"""
|
||||||
try:
|
try:
|
||||||
user_id = callback.from_user.id
|
user_id = callback.from_user.id
|
||||||
username = callback.from_user.username
|
username = callback.from_user.username
|
||||||
|
|||||||
@@ -78,7 +78,7 @@ async def _fetch_user_and_subs(telegram_id: int):
|
|||||||
if not user or not isinstance(user, dict):
|
if not user or not isinstance(user, dict):
|
||||||
return None, []
|
return None, []
|
||||||
|
|
||||||
user_id = user["telegram_id"]
|
user_id = user["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):
|
||||||
|
|||||||
Reference in New Issue
Block a user