diff --git a/app/database/crud/user.py b/app/database/crud/user.py
index 8613ad20..e0295e73 100644
--- a/app/database/crud/user.py
+++ b/app/database/crud/user.py
@@ -62,10 +62,34 @@ async def get_user_by_telegram_id(db: AsyncSession, telegram_id: int) -> Optiona
.where(User.telegram_id == telegram_id)
)
user = result.scalar_one_or_none()
-
+
if user and user.subscription:
_ = user.subscription.is_active
-
+
+ return user
+
+
+async def get_user_by_username(db: AsyncSession, username: str) -> Optional[User]:
+ if not username:
+ return None
+
+ normalized = username.lower()
+
+ result = await db.execute(
+ select(User)
+ .options(
+ selectinload(User.subscription),
+ selectinload(User.promo_group),
+ selectinload(User.referrer),
+ )
+ .where(func.lower(User.username) == normalized)
+ )
+
+ user = result.scalar_one_or_none()
+
+ if user and user.subscription:
+ _ = user.subscription.is_active
+
return user
diff --git a/app/handlers/admin/users.py b/app/handlers/admin/users.py
index ed18af5d..32d3b02e 100644
--- a/app/handlers/admin/users.py
+++ b/app/handlers/admin/users.py
@@ -1,6 +1,7 @@
import logging
+import re
from datetime import datetime, timedelta
-from typing import Optional
+from typing import Optional, List, Tuple
from aiogram import Dispatcher, types, F
from aiogram.exceptions import TelegramForbiddenError, TelegramBadRequest
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
@@ -10,7 +11,12 @@ from sqlalchemy.ext.asyncio import AsyncSession
from app.config import settings
from app.states import AdminStates
from app.database.models import User, UserStatus, Subscription, SubscriptionStatus, TransactionType
-from app.database.crud.user import get_user_by_id
+from app.database.crud.user import (
+ get_user_by_id,
+ get_user_by_telegram_id,
+ get_user_by_username,
+ get_referrals,
+)
from app.database.crud.campaign import (
get_campaign_registration_by_user,
get_campaign_statistics,
@@ -1489,6 +1495,395 @@ async def show_user_management(
await callback.answer()
+async def _build_user_referrals_view(
+ db: AsyncSession,
+ language: str,
+ user_id: int,
+ limit: int = 30,
+) -> Optional[Tuple[str, InlineKeyboardMarkup]]:
+ texts = get_texts(language)
+
+ user = await get_user_by_id(db, user_id)
+ if not user:
+ return None
+
+ referrals = await get_referrals(db, user_id)
+
+ header = texts.t(
+ "ADMIN_USER_REFERRALS_TITLE",
+ "🤝 Рефералы пользователя",
+ )
+ summary = texts.t(
+ "ADMIN_USER_REFERRALS_SUMMARY",
+ "👤 {name} (ID: {telegram_id})\n👥 Всего рефералов: {count}",
+ ).format(
+ name=user.full_name,
+ telegram_id=user.telegram_id,
+ count=len(referrals),
+ )
+
+ lines: List[str] = [header, summary]
+
+ if referrals:
+ lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_LIST_HEADER",
+ "Список рефералов:",
+ )
+ )
+ items = []
+ for referral in referrals[:limit]:
+ username_part = (
+ f", @{referral.username}"
+ if referral.username
+ else ""
+ )
+ items.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_LIST_ITEM",
+ "• {name} (ID: {telegram_id}{username_part})",
+ ).format(
+ name=referral.full_name,
+ telegram_id=referral.telegram_id,
+ username_part=username_part,
+ )
+ )
+
+ lines.append("\n".join(items))
+
+ if len(referrals) > limit:
+ remaining = len(referrals) - limit
+ lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_LIST_TRUNCATED",
+ "• … и ещё {count} рефералов",
+ ).format(count=remaining)
+ )
+ else:
+ lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_EMPTY",
+ "Рефералов пока нет.",
+ )
+ )
+
+ lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_EDIT_HINT",
+ "✏️ Чтобы изменить список, нажмите «✏️ Редактировать» ниже.",
+ )
+ )
+
+ keyboard = InlineKeyboardMarkup(
+ inline_keyboard=[
+ [
+ InlineKeyboardButton(
+ text=texts.t(
+ "ADMIN_USER_REFERRALS_EDIT_BUTTON",
+ "✏️ Редактировать",
+ ),
+ callback_data=f"admin_user_referrals_edit_{user_id}",
+ )
+ ],
+ [
+ InlineKeyboardButton(
+ text=texts.BACK,
+ callback_data=f"admin_user_manage_{user_id}",
+ )
+ ],
+ ]
+ )
+
+ return "\n\n".join(lines), keyboard
+
+
+@admin_required
+@error_handler
+async def show_user_referrals(
+ callback: types.CallbackQuery,
+ db_user: User,
+ db: AsyncSession,
+ state: FSMContext,
+):
+ user_id = int(callback.data.split('_')[-1])
+
+ current_state = await state.get_state()
+ if current_state == AdminStates.editing_user_referrals:
+ data = await state.get_data()
+ preserved_data = {
+ key: value
+ for key, value in data.items()
+ if key not in {"editing_referrals_user_id", "referrals_message_id"}
+ }
+ await state.clear()
+ if preserved_data:
+ await state.update_data(**preserved_data)
+
+ view = await _build_user_referrals_view(db, db_user.language, user_id)
+ if not view:
+ await callback.answer("❌ Пользователь не найден", show_alert=True)
+ return
+
+ text, keyboard = view
+
+ await callback.message.edit_text(
+ text,
+ reply_markup=keyboard,
+ )
+ await callback.answer()
+
+
+@admin_required
+@error_handler
+async def start_edit_user_referrals(
+ callback: types.CallbackQuery,
+ db_user: User,
+ state: FSMContext,
+ db: AsyncSession,
+):
+ user_id = int(callback.data.split('_')[-1])
+
+ user = await get_user_by_id(db, user_id)
+ if not user:
+ await callback.answer("❌ Пользователь не найден", show_alert=True)
+ return
+
+ texts = get_texts(db_user.language)
+
+ prompt = texts.t(
+ "ADMIN_USER_REFERRALS_EDIT_PROMPT",
+ (
+ "✏️ Редактирование рефералов\n\n"
+ "Отправьте список рефералов для пользователя {name} (ID: {telegram_id}):\n"
+ "• Используйте TG ID или @username\n"
+ "• Значения можно указывать через запятую, пробел или с новой строки\n"
+ "• Чтобы очистить список, отправьте 0 или слово 'нет'\n\n"
+ "Или нажмите кнопку ниже, чтобы отменить."
+ ),
+ ).format(
+ name=user.full_name,
+ telegram_id=user.telegram_id,
+ )
+
+ await state.update_data(
+ editing_referrals_user_id=user_id,
+ referrals_message_id=callback.message.message_id,
+ )
+
+ await callback.message.edit_text(
+ prompt,
+ reply_markup=InlineKeyboardMarkup(
+ inline_keyboard=[
+ [
+ InlineKeyboardButton(
+ text=texts.BACK,
+ callback_data=f"admin_user_referrals_{user_id}",
+ )
+ ]
+ ]
+ ),
+ )
+
+ await state.set_state(AdminStates.editing_user_referrals)
+ await callback.answer()
+
+
+@admin_required
+@error_handler
+async def process_edit_user_referrals(
+ message: types.Message,
+ db_user: User,
+ state: FSMContext,
+ db: AsyncSession,
+):
+ texts = get_texts(db_user.language)
+ data = await state.get_data()
+
+ user_id = data.get("editing_referrals_user_id")
+ if not user_id:
+ await message.answer(
+ texts.t(
+ "ADMIN_USER_REFERRALS_STATE_LOST",
+ "❌ Не удалось определить пользователя. Попробуйте начать сначала.",
+ )
+ )
+ await state.clear()
+ return
+
+ raw_text = message.text.strip()
+ lower_text = raw_text.lower()
+ clear_keywords = {"0", "нет", "none", "пусто", "clear"}
+ clear_requested = lower_text in clear_keywords
+
+ tokens: List[str] = []
+ if not clear_requested:
+ parts = re.split(r"[,\n]+", raw_text)
+ for part in parts:
+ for token in part.split():
+ cleaned = token.strip()
+ if cleaned and cleaned not in tokens:
+ tokens.append(cleaned)
+
+ found_users: List[User] = []
+ not_found: List[str] = []
+ skipped_self: List[str] = []
+ duplicate_tokens: List[str] = []
+
+ seen_ids = set()
+
+ for token in tokens:
+ normalized = token.strip()
+ if not normalized:
+ continue
+
+ if normalized.startswith("@"):
+ normalized = normalized[1:]
+
+ user = None
+ if normalized.isdigit():
+ try:
+ user = await get_user_by_telegram_id(db, int(normalized))
+ except ValueError:
+ user = None
+ else:
+ user = await get_user_by_username(db, normalized)
+
+ if not user:
+ not_found.append(token)
+ continue
+
+ if user.id == user_id:
+ skipped_self.append(token)
+ continue
+
+ if user.id in seen_ids:
+ duplicate_tokens.append(token)
+ continue
+
+ seen_ids.add(user.id)
+ found_users.append(user)
+
+ if not found_users and not clear_requested:
+ error_lines = [
+ texts.t(
+ "ADMIN_USER_REFERRALS_NO_VALID",
+ "❌ Не удалось найти ни одного пользователя по введённым данным.",
+ )
+ ]
+ if not_found:
+ error_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_INVALID_ENTRIES",
+ "Не найдены: {values}",
+ ).format(values=", ".join(not_found))
+ )
+ if skipped_self:
+ error_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_SELF_SKIPPED",
+ "Пропущены значения пользователя: {values}",
+ ).format(values=", ".join(skipped_self))
+ )
+ await message.answer("\n".join(error_lines))
+ return
+
+ user_service = UserService()
+
+ new_referral_ids = [user.id for user in found_users] if not clear_requested else []
+
+ success, details = await user_service.update_user_referrals(
+ db,
+ user_id,
+ new_referral_ids,
+ db_user.id,
+ )
+
+ if not success:
+ await message.answer(
+ texts.t(
+ "ADMIN_USER_REFERRALS_UPDATE_ERROR",
+ "❌ Не удалось обновить рефералов. Попробуйте позже.",
+ )
+ )
+ return
+
+ response_lines = [
+ texts.t(
+ "ADMIN_USER_REFERRALS_UPDATED",
+ "✅ Список рефералов обновлён.",
+ )
+ ]
+
+ total_referrals = details.get("total", len(new_referral_ids))
+ added = details.get("added", 0)
+ removed = details.get("removed", 0)
+
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_UPDATED_TOTAL",
+ "• Текущий список: {total}",
+ ).format(total=total_referrals)
+ )
+
+ if added > 0:
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_UPDATED_ADDED",
+ "• Добавлено: {count}",
+ ).format(count=added)
+ )
+
+ if removed > 0:
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_UPDATED_REMOVED",
+ "• Удалено: {count}",
+ ).format(count=removed)
+ )
+
+ if not_found:
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_INVALID_ENTRIES",
+ "Не найдены: {values}",
+ ).format(values=", ".join(not_found))
+ )
+
+ if skipped_self:
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_SELF_SKIPPED",
+ "Пропущены значения пользователя: {values}",
+ ).format(values=", ".join(skipped_self))
+ )
+
+ if duplicate_tokens:
+ response_lines.append(
+ texts.t(
+ "ADMIN_USER_REFERRALS_DUPLICATES",
+ "Игнорированы дубли: {values}",
+ ).format(values=", ".join(duplicate_tokens))
+ )
+
+ view = await _build_user_referrals_view(db, db_user.language, user_id)
+ message_id = data.get("referrals_message_id")
+
+ if view and message_id:
+ try:
+ await message.bot.edit_message_text(
+ view[0],
+ chat_id=message.chat.id,
+ message_id=message_id,
+ reply_markup=view[1],
+ )
+ except TelegramBadRequest:
+ await message.answer(view[0], reply_markup=view[1])
+ elif view:
+ await message.answer(view[0], reply_markup=view[1])
+
+ await message.answer("\n".join(response_lines))
+ await state.clear()
+
async def _render_user_promo_group(
message: types.Message,
language: str,
@@ -4159,6 +4554,21 @@ def register_handlers(dp: Dispatcher):
AdminStates.editing_user_balance
)
+ dp.callback_query.register(
+ show_user_referrals,
+ F.data.startswith("admin_user_referrals_") & ~F.data.contains("_edit")
+ )
+
+ dp.callback_query.register(
+ start_edit_user_referrals,
+ F.data.startswith("admin_user_referrals_edit_")
+ )
+
+ dp.message.register(
+ process_edit_user_referrals,
+ AdminStates.editing_user_referrals
+ )
+
dp.callback_query.register(
start_send_user_message,
F.data.startswith("admin_user_send_message_")
diff --git a/app/keyboards/admin.py b/app/keyboards/admin.py
index 5d993543..fb0b21b1 100644
--- a/app/keyboards/admin.py
+++ b/app/keyboards/admin.py
@@ -760,6 +760,12 @@ def get_user_management_keyboard(user_id: int, user_status: str, language: str =
callback_data=f"admin_user_promo_group_{user_id}"
)
],
+ [
+ InlineKeyboardButton(
+ text=_t(texts, "ADMIN_USER_REFERRALS_BUTTON", "🤝 Рефералы"),
+ callback_data=f"admin_user_referrals_{user_id}"
+ )
+ ],
[
InlineKeyboardButton(
text=_t(texts, "ADMIN_USER_STATISTICS", "📊 Статистика"),
diff --git a/app/localization/locales/ru.json b/app/localization/locales/ru.json
index 67437d35..fc188245 100644
--- a/app/localization/locales/ru.json
+++ b/app/localization/locales/ru.json
@@ -713,6 +713,26 @@
"ADMIN_USER_PROMO_GROUP_ALREADY": "ℹ️ Пользователь уже состоит в этой промогруппе.",
"ADMIN_USER_PROMO_GROUP_BACK": "⬅️ К пользователю",
"ADMIN_USER_PROMO_GROUP_BUTTON": "👥 Промогруппа",
+ "ADMIN_USER_REFERRALS_BUTTON": "🤝 Рефералы",
+ "ADMIN_USER_REFERRALS_TITLE": "🤝 Рефералы пользователя",
+ "ADMIN_USER_REFERRALS_SUMMARY": "👤 {name} (ID: {telegram_id})\n👥 Всего рефералов: {count}",
+ "ADMIN_USER_REFERRALS_LIST_HEADER": "Список рефералов:",
+ "ADMIN_USER_REFERRALS_LIST_ITEM": "• {name} (ID: {telegram_id}{username_part})",
+ "ADMIN_USER_REFERRALS_LIST_TRUNCATED": "• … и ещё {count} рефералов",
+ "ADMIN_USER_REFERRALS_EMPTY": "Рефералов пока нет.",
+ "ADMIN_USER_REFERRALS_EDIT_HINT": "✏️ Чтобы изменить список, нажмите «✏️ Редактировать» ниже.",
+ "ADMIN_USER_REFERRALS_EDIT_BUTTON": "✏️ Редактировать",
+ "ADMIN_USER_REFERRALS_EDIT_PROMPT": "✏️ Редактирование рефералов\n\nОтправьте список рефералов для пользователя {name} (ID: {telegram_id}):\n• Используйте TG ID или @username\n• Значения можно указывать через запятую, пробел или с новой строки\n• Чтобы очистить список, отправьте 0 или слово 'нет'\n\nИли нажмите кнопку ниже, чтобы отменить.",
+ "ADMIN_USER_REFERRALS_STATE_LOST": "❌ Не удалось определить пользователя. Попробуйте начать сначала.",
+ "ADMIN_USER_REFERRALS_NO_VALID": "❌ Не удалось найти ни одного пользователя по введённым данным.",
+ "ADMIN_USER_REFERRALS_INVALID_ENTRIES": "Не найдены: {values}",
+ "ADMIN_USER_REFERRALS_SELF_SKIPPED": "Пропущены значения пользователя: {values}",
+ "ADMIN_USER_REFERRALS_DUPLICATES": "Игнорированы дубли: {values}",
+ "ADMIN_USER_REFERRALS_UPDATE_ERROR": "❌ Не удалось обновить рефералов. Попробуйте позже.",
+ "ADMIN_USER_REFERRALS_UPDATED": "✅ Список рефералов обновлён.",
+ "ADMIN_USER_REFERRALS_UPDATED_TOTAL": "• Текущий список: {total}",
+ "ADMIN_USER_REFERRALS_UPDATED_ADDED": "• Добавлено: {count}",
+ "ADMIN_USER_REFERRALS_UPDATED_REMOVED": "• Удалено: {count}",
"ADMIN_USER_PROMO_GROUP_CURRENT": "Текущая группа: {name}",
"ADMIN_USER_PROMO_GROUP_CURRENT_NONE": "Текущая группа: не назначена",
"ADMIN_USER_PROMO_GROUP_DISCOUNTS": "Скидки — серверы: {servers}%, трафик: {traffic}%, устройства: {devices}%",
diff --git a/app/services/user_service.py b/app/services/user_service.py
index 6c19472b..9d551fec 100644
--- a/app/services/user_service.py
+++ b/app/services/user_service.py
@@ -9,7 +9,7 @@ from app.database.crud.user import (
get_user_by_id, get_user_by_telegram_id, get_users_list,
get_users_count, get_users_statistics, get_inactive_users,
add_user_balance, subtract_user_balance, update_user, delete_user,
- get_users_spending_stats
+ get_users_spending_stats, get_referrals
)
from app.database.crud.promo_group import get_promo_group_by_id
from app.database.crud.transaction import get_user_transactions_count
@@ -411,6 +411,72 @@ class UserService:
logger.error(f"Ошибка обновления промогруппы пользователя {user_id}: {e}")
return False, None, None, None
+ async def update_user_referrals(
+ self,
+ db: AsyncSession,
+ user_id: int,
+ referral_user_ids: List[int],
+ admin_id: int,
+ ) -> Tuple[bool, Dict[str, int]]:
+ try:
+ user = await get_user_by_id(db, user_id)
+ if not user:
+ return False, {"error": "user_not_found"}
+
+ unique_ids: List[int] = []
+ for referral_id in referral_user_ids:
+ if referral_id == user_id:
+ continue
+ if referral_id not in unique_ids:
+ unique_ids.append(referral_id)
+
+ current_referrals = await get_referrals(db, user_id)
+ current_ids = {ref.id for ref in current_referrals}
+
+ to_assign = unique_ids
+ to_remove = [rid for rid in current_ids if rid not in unique_ids]
+ to_add = [rid for rid in unique_ids if rid not in current_ids]
+
+ if to_assign:
+ await db.execute(
+ update(User)
+ .where(User.id.in_(to_assign))
+ .values(referred_by_id=user_id)
+ )
+
+ if to_remove:
+ await db.execute(
+ update(User)
+ .where(User.id.in_(to_remove))
+ .values(referred_by_id=None)
+ )
+
+ await db.commit()
+
+ logger.info(
+ "Админ %s обновил рефералов пользователя %s: добавлено %s, удалено %s, всего %s",
+ admin_id,
+ user_id,
+ len(to_add),
+ len(to_remove),
+ len(unique_ids),
+ )
+
+ return True, {
+ "added": len(to_add),
+ "removed": len(to_remove),
+ "total": len(unique_ids),
+ }
+
+ except Exception as e:
+ await db.rollback()
+ logger.error(
+ "Ошибка обновления рефералов пользователя %s: %s",
+ user_id,
+ e,
+ )
+ return False, {"error": "update_failed"}
+
async def block_user(
self,
db: AsyncSession,
diff --git a/app/states.py b/app/states.py
index 549e314f..981c314a 100644
--- a/app/states.py
+++ b/app/states.py
@@ -90,6 +90,7 @@ class AdminStates(StatesGroup):
editing_device_price = State()
editing_user_devices = State()
editing_user_traffic = State()
+ editing_user_referrals = State()
editing_rules_page = State()
editing_privacy_policy = State()