Update referral.py

.env.example

@@ -1,113 +1,112 @@
-# TELEGRAM BOT CONFIGURATION
+# ===============================================
+# 🤖 REMNAWAVE BEDOLAGA BOT CONFIGURATION
+# ===============================================
+
+# ===== TELEGRAM BOT =====
 BOT_TOKEN=
 ADMIN_IDS=
-SUPPORT_USERNAME=
+SUPPORT_USERNAME=@support
 
-# DATABASE CONFIGURATION
-DATABASE_URL=sqlite+aiosqlite:///./bot.db
-REDIS_URL=redis://localhost:6379/0
+# ===== DATABASE =====
+# Для Docker используйте PostgreSQL:
+DATABASE_URL=postgresql+asyncpg://remnawave_user:secure_password_123@postgres:5432/remnawave_bot
+# Для локального запуска без Docker используйте SQLite: sqlite+aiosqlite:///./bot.db 
+# DATABASE_URL=postgresql+asyncpg://remnawave_user:secure_password_123@postgres:5432/remnawave_bot
 
-# REMNAWAVE API CONFIGURATION
+REDIS_URL=redis://redis:6379/0
+
+# Пароли для Docker (PostgreSQL/Redis)
+POSTGRES_DB=remnawave_bot
+POSTGRES_USER=remnawave_user
+POSTGRES_PASSWORD=secure_password_123
+
+# ===== REMNAWAVE API =====
 REMNAWAVE_API_URL=
 REMNAWAVE_API_KEY=
 
-# === NEW: Traffic Selection Mode Settings ===
+# ===== ТРИАЛ ПОДПИСКА =====
+TRIAL_DURATION_DAYS=3
+TRIAL_TRAFFIC_LIMIT_GB=10
+TRIAL_DEVICE_LIMIT=1
+TRIAL_SQUAD_UUID=
+DEFAULT_TRAFFIC_RESET_STRATEGY=MONTH
+DEFAULT_TRAFFIC_LIMIT_GB=100
+DEFAULT_DEVICE_LIMIT=1
+
+# ===== НАСТРОЙКИ ТРАФИКА =====
 # Режим выбора трафика:
 # "selectable" - пользователи выбирают пакеты трафика (по умолчанию)
-# "fixed" - фиксированный лимит трафика для всех подписок, доступно 5/10/25/50/100/250/0 (0 безлимит) гб 
+# "fixed" - фиксированный лимит трафика для всех подписок
 TRAFFIC_SELECTION_MODE=selectable
 
 # Фиксированный лимит трафика в ГБ (используется только в режиме "fixed")
 # 0 = безлимит
-# для "fixed" обязательно должы быть проставлены цены на пакеты 5/10/25/50/100/250/0 можно постать 0 руб - будет беслпатно
-FIXED_TRAFFIC_LIMIT_GB=0
+FIXED_TRAFFIC_LIMIT_GB=100
 
-# TRIAL SUBSCRIPTION SETTINGS
-TRIAL_DURATION_DAYS=3
-TRIAL_TRAFFIC_LIMIT_GB=10
-TRIAL_DEVICE_LIMIT=2
-TRIAL_SQUAD_UUID=
-DEFAULT_TRAFFIC_RESET_STRATEGY=MONTH
-
-# === УПРАВЛЕНИЕ ПЕРИОДАМИ ПОДПИСКИ ===
+# ===== ПЕРИОДЫ ПОДПИСКИ =====
 # Доступные периоды подписки (через запятую)
 # Возможные значения: 14,30,60,90,180,360
-AVAILABLE_SUBSCRIPTION_PERIODS=14,30,60,90,180,360
+AVAILABLE_SUBSCRIPTION_PERIODS=30,90,180
 AVAILABLE_RENEWAL_PERIODS=30,90,180
 
-# SUBSCRIPTION PRICING (в копейках для точности)
-BASE_SUBSCRIPTION_PRICE=50000
+# ===== ЦЕНЫ (в копейках) =====
+BASE_SUBSCRIPTION_PRICE=0
 
-PRICE_14_DAYS=5000
-PRICE_30_DAYS=9900  
-PRICE_60_DAYS=18900
-PRICE_90_DAYS=26900
-PRICE_180_DAYS=49900
-PRICE_360_DAYS=89900
+# Цены за периоды
+PRICE_14_DAYS=7000
+PRICE_30_DAYS=9900
+PRICE_60_DAYS=25900
+PRICE_90_DAYS=36900
+PRICE_180_DAYS=69900
+PRICE_360_DAYS=109900
 
-PRICE_TRAFFIC_5GB=2000
-PRICE_TRAFFIC_10GB=4000
-PRICE_TRAFFIC_25GB=6000
-PRICE_TRAFFIC_50GB=10000
-PRICE_TRAFFIC_100GB=15000
-PRICE_TRAFFIC_250GB=20000
-PRICE_TRAFFIC_UNLIMITED=25000
+# Цены за трафик
+PRICE_TRAFFIC_5GB=1000
+PRICE_TRAFFIC_10GB=2000
+PRICE_TRAFFIC_25GB=3000
+PRICE_TRAFFIC_50GB=4000
+PRICE_TRAFFIC_100GB=6000
+PRICE_TRAFFIC_250GB=10000
+PRICE_TRAFFIC_UNLIMITED=15000
 
+# Цена за дополнительное устройство
 PRICE_PER_DEVICE=5000
 
-# REFERRAL SYSTEM SETTINGS
-REFERRAL_REGISTRATION_REWARD=5000
+# ===== РЕФЕРАЛЬНАЯ СИСТЕМА =====
+REFERRAL_REGISTRATION_REWARD=10000
 REFERRED_USER_REWARD=10000
 REFERRAL_COMMISSION_PERCENT=25
 
-# Режим работы кнопки "Подключиться"
-# guide - открывает гайд подключения (режим 1)
-# miniapp_subscription - открывает ссылку подписки в мини-приложении (режим 2)
-# miniapp_custom - открывает заданную ссылку в мини-приложении (режим 3)
-CONNECT_BUTTON_MODE=miniapp_subscription
-# URL для режима miniapp_custom (обязателен при CONNECT_BUTTON_MODE=miniapp_custom)
-# MINIAPP_CUSTOM_URL=
-
-# AUTO-PAYMENT SETTINGS
+# ===== АВТОПРОДЛЕНИЕ =====
 AUTOPAY_WARNING_DAYS=3,1
+DEFAULT_AUTOPAY_DAYS_BEFORE=3
+MIN_BALANCE_FOR_AUTOPAY_KOPEKS=10000
 
-# MONITORING SETTINGS
-MONITORING_INTERVAL=60
-INACTIVE_USER_DELETE_MONTHS=3
+# ===== ПЛАТЕЖНЫЕ СИСТЕМЫ =====
 
-TRIAL_WARNING_HOURS=2
-ENABLE_NOTIFICATIONS=true
-NOTIFICATION_RETRY_ATTEMPTS=3
-MONITORING_LOGS_RETENTION_DAYS=30
-
-# PAYMENT SYSTEMS
+# Telegram Stars (работает автоматически)
 TELEGRAM_STARS_ENABLED=true
+
+# Tribute (https://tribute.app)
 TRIBUTE_ENABLED=false
 TRIBUTE_API_KEY=
 TRIBUTE_WEBHOOK_SECRET=your_webhook_secret
-TRIBUTE_DONATE_LINK=https://t.me/tribute/app?startapp=XXXX
+TRIBUTE_DONATE_LINK=
 TRIBUTE_WEBHOOK_PATH=/tribute-webhook
 TRIBUTE_WEBHOOK_PORT=8081
 
-# === НОВЫЕ НАСТРОЙКИ YOOKASSA ===
-# Включение/выключение YooKassa
+# YooKassa (https://yookassa.ru)
 YOOKASSA_ENABLED=false
-
-# Основные настройки YooKassa (получить в личном кабинете)
-YOOKASSA_SHOP_ID=your_shop_id_here
-YOOKASSA_SECRET_KEY=your_secret_key_here
-
-# URL для возврата после оплаты (необязательно, по умолчанию t.me/your_bot)
-YOOKASSA_RETURN_URL=https://yourdomain.com/payment-success
-
-# Email по умолчанию для чеков (если пользователь не указал свой)
+YOOKASSA_SHOP_ID=
+YOOKASSA_SECRET_KEY=
+YOOKASSA_RETURN_URL=
 YOOKASSA_DEFAULT_RECEIPT_EMAIL=receipts@yourdomain.com
 
 # Настройки чеков для налоговой
 YOOKASSA_VAT_CODE=1
 # Коды НДС:
 # 1 - НДС не облагается
-# 2 - НДС 0%  
+# 2 - НДС 0%
 # 3 - НДС 10%
 # 4 - НДС 20%
 # 5 - НДС 10/110
@@ -116,7 +115,7 @@ YOOKASSA_VAT_CODE=1
 YOOKASSA_PAYMENT_MODE=full_payment
 # Способы расчета:
 # full_payment - полная оплата
-# partial_payment - частичная оплата  
+# partial_payment - частичная оплата
 # advance - аванс
 # full_prepayment - полная предоплата
 # partial_prepayment - частичная предоплата
@@ -126,7 +125,7 @@ YOOKASSA_PAYMENT_MODE=full_payment
 YOOKASSA_PAYMENT_SUBJECT=service
 # Предметы расчета:
 # commodity - товар
-# excise - подакцизный товар  
+# excise - подакцизный товар
 # job - работа
 # service - услуга
 # gambling_bet - ставка в азартной игре
@@ -139,21 +138,54 @@ YOOKASSA_PAYMENT_SUBJECT=service
 # composite - составной предмет расчета
 # another - другое
 
-# Webhook для получения уведомлений от YooKassa
+# Webhook настройки
 YOOKASSA_WEBHOOK_PATH=/yookassa-webhook
 YOOKASSA_WEBHOOK_PORT=8082
-YOOKASSA_WEBHOOK_SECRET=ваш_секретный_ключ_для_webhook
+YOOKASSA_WEBHOOK_SECRET=your_webhook_secret
 
-WEBHOOK_URL=https://example.com
+# ===== ИНТЕРФЕЙС И UX =====
+
+# Режим работы кнопки "Подключиться"
+# guide - открывает гайд подключения (режим 1)
+# miniapp_subscription - открывает ссылку подписки в мини-приложении (режим 2)
+# miniapp_custom - открывает заданную ссылку в мини-приложении (режим 3)
+CONNECT_BUTTON_MODE=guide
+
+# URL для режима miniapp_custom (обязателен при CONNECT_BUTTON_MODE=miniapp_custom)
+MINIAPP_CUSTOM_URL=
+
+# ===== МОНИТОРИНГ И УВЕДОМЛЕНИЯ =====
+MONITORING_INTERVAL=60
+INACTIVE_USER_DELETE_MONTHS=3
+
+# Уведомления
+TRIAL_WARNING_HOURS=2
+ENABLE_NOTIFICATIONS=true
+NOTIFICATION_RETRY_ATTEMPTS=3
+MONITORING_LOGS_RETENTION_DAYS=30
+NOTIFICATION_CACHE_HOURS=24
+
+# ===== РЕЖИМ ТЕХНИЧЕСКИХ РАБОТ =====
+MAINTENANCE_MODE=false
+MAINTENANCE_CHECK_INTERVAL=30
+MAINTENANCE_AUTO_ENABLE=true
+MAINTENANCE_MESSAGE=Ведутся технические работы. Сервис временно недоступен. Попробуйте позже.
+
+# ===== ЛОКАЛИЗАЦИЯ =====
+DEFAULT_LANGUAGE=ru
+AVAILABLE_LANGUAGES=ru,en
+
+# ===== ЛОГИРОВАНИЕ =====
+LOG_LEVEL=INFO
+LOG_FILE=logs/bot.log
+
+# ===== РАЗРАБОТКА =====
+DEBUG=false
+WEBHOOK_URL=
 WEBHOOK_PATH=/webhook
 
-# LOCALIZATION
-DEFAULT_LANGUAGE=ru
-AVAILABLE_LANGUAGES=ru
-
-# LOGGING
-LOG_LEVEL=INFO
-LOG_FILE=/tmp/bot.log
-
-# DEVELOPMENT
-DEBUG=false
+# ===== ДОПОЛНИТЕЛЬНЫЕ НАСТРОЙКИ =====
+# Конфигурация приложений для гайда подключения
+APP_CONFIG_PATH=app-config.json
+ENABLE_DEEP_LINKS=true
+APP_CONFIG_CACHE_TTL=3600

app/database/crud/promocode.py

@@ -5,7 +5,7 @@ from sqlalchemy import select, and_, func
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import selectinload
 
-from app.database.models import PromoCode, PromoCodeUse, PromoCodeType
+from app.database.models import PromoCode, PromoCodeUse, PromoCodeType, User
 
 logger = logging.getLogger(__name__)
 
@@ -187,8 +187,13 @@ async def update_promocode(
 
 
 async def delete_promocode(db: AsyncSession, promocode: PromoCode) -> bool:
-    
     try:
+        from sqlalchemy import delete as sql_delete
+        
+        await db.execute(
+            sql_delete(PromoCodeUse).where(PromoCodeUse.promocode_id == promocode.id)
+        )
+        
         await db.delete(promocode)
         await db.commit()
         
@@ -222,15 +227,24 @@ async def get_promocode_statistics(db: AsyncSession, promocode_id: int) -> dict:
     today_uses = today_uses_result.scalar()
     
     recent_uses_result = await db.execute(
-        select(PromoCodeUse)
+        select(PromoCodeUse, User)
+        .join(User, PromoCodeUse.user_id == User.id)
         .where(PromoCodeUse.promocode_id == promocode_id)
         .order_by(PromoCodeUse.used_at.desc())
         .limit(10)
     )
-    recent_uses = recent_uses_result.scalars().all()
+    recent_uses_data = recent_uses_result.all()
+    
+    recent_uses = []
+    for use, user in recent_uses_data:
+        use.user_username = user.username
+        use.user_full_name = user.full_name
+        use.user_telegram_id = user.telegram_id
+        recent_uses.append(use)
     
     return {
         "total_uses": total_uses,
         "today_uses": today_uses,
         "recent_uses": recent_uses
-    }
+    }
+

app/handlers/admin/promocodes.py

@@ -6,7 +6,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from app.config import settings
 from app.states import AdminStates
-from app.database.models import User, PromoCodeType
+from app.database.models import PromoCode, PromoCodeUse, PromoCodeType, User
 from app.keyboards.admin import (
     get_admin_promocodes_keyboard, get_promocode_type_keyboard,
     get_admin_pagination_keyboard, get_confirmation_keyboard
@@ -77,6 +77,7 @@ async def show_promocodes_list(
         return
     
     text = f"🎫 <b>Список промокодов</b> (стр. {page}/{total_pages})\n\n"
+    keyboard = []
     
     for promo in promocodes:
         status_emoji = "✅" if promo.is_active else "❌"
@@ -93,9 +94,14 @@ async def show_promocodes_list(
         if promo.valid_until:
             text += f"⏰ До: {format_datetime(promo.valid_until)}\n"
         
-        text += f"🔧 Управление: /promo_{promo.id}\n\n"
-    
-    keyboard = []
+        keyboard.append([
+            types.InlineKeyboardButton(
+                text=f"🎫 {promo.code}", 
+                callback_data=f"promo_manage_{promo.id}"
+            )
+        ])
+        
+        text += "\n" 
     
     if total_pages > 1:
         pagination_row = get_admin_pagination_keyboard(
@@ -182,128 +188,75 @@ async def show_promocode_management(
     )
     await callback.answer()
 
-
 @admin_required
 @error_handler
-async def toggle_promocode_status(
+async def show_promocode_edit_menu(
     callback: types.CallbackQuery,
     db_user: User,
     db: AsyncSession
 ):
-    promo_id = int(callback.data.split('_')[-1])
-    
-    promo = await db.get(PromoCode, promo_id)
-    if not promo:
-        await callback.answer("❌ Промокод не найден", show_alert=True)
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
         return
     
-    new_status = not promo.is_active
-    await update_promocode(db, promo, is_active=new_status)
-    
-    status_text = "активирован" if new_status else "деактивирован"
-    await callback.answer(f"✅ Промокод {status_text}", show_alert=True)
-    
-    await show_promocode_management(callback, db_user, db)
-
-
-@admin_required
-@error_handler
-async def confirm_delete_promocode(
-    callback: types.CallbackQuery,
-    db_user: User,
-    db: AsyncSession
-):
-    promo_id = int(callback.data.split('_')[-1])
-    
     promo = await db.get(PromoCode, promo_id)
     if not promo:
         await callback.answer("❌ Промокод не найден", show_alert=True)
         return
     
     text = f"""
-⚠️ <b>Подтверждение удаления</b>
+✏️ <b>Редактирование промокода</b> <code>{promo.code}</code>
 
-Вы действительно хотите удалить промокод <code>{promo.code}</code>?
-
-<b>Внимание:</b> Это действие нельзя отменить!
+💰 <b>Текущие параметры:</b>
 """
     
-    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+    if promo.type == PromoCodeType.BALANCE.value:
+        text += f"• Бонус: {settings.format_price(promo.balance_bonus_kopeks)}\n"
+    elif promo.type in [PromoCodeType.SUBSCRIPTION_DAYS.value, PromoCodeType.TRIAL_SUBSCRIPTION.value]:
+        text += f"• Дней: {promo.subscription_days}\n"
+    
+    text += f"• Использований: {promo.current_uses}/{promo.max_uses}\n"
+    
+    if promo.valid_until:
+        text += f"• До: {format_datetime(promo.valid_until)}\n"
+    else:
+        text += f"• Срок: бессрочно\n"
+    
+    text += f"\nВыберите параметр для изменения:"
+    
+    keyboard = [
         [
             types.InlineKeyboardButton(
-                text="✅ Да, удалить", 
-                callback_data=f"promo_delete_confirm_{promo.id}"
-            ),
+                text="📅 Дата окончания", 
+                callback_data=f"promo_edit_date_{promo.id}"
+            )
+        ],
+        [
             types.InlineKeyboardButton(
-                text="❌ Отмена", 
-                callback_data=f"promo_manage_{promo.id}"
+                text="📊 Количество использований", 
+                callback_data=f"promo_edit_uses_{promo.id}"
             )
         ]
-    ])
+    ]
     
-    await callback.message.edit_text(text, reply_markup=keyboard)
-    await callback.answer()
-
-
-@admin_required
-@error_handler
-async def delete_promocode_confirmed(
-    callback: types.CallbackQuery,
-    db_user: User,
-    db: AsyncSession
-):
-    promo_id = int(callback.data.split('_')[-1])
+    if promo.type == PromoCodeType.BALANCE.value:
+        keyboard.insert(1, [
+            types.InlineKeyboardButton(
+                text="💰 Сумма бонуса", 
+                callback_data=f"promo_edit_amount_{promo.id}"
+            )
+        ])
+    elif promo.type in [PromoCodeType.SUBSCRIPTION_DAYS.value, PromoCodeType.TRIAL_SUBSCRIPTION.value]:
+        keyboard.insert(1, [
+            types.InlineKeyboardButton(
+                text="📅 Количество дней", 
+                callback_data=f"promo_edit_days_{promo.id}"
+            )
+        ])
     
-    promo = await db.get(PromoCode, promo_id)
-    if not promo:
-        await callback.answer("❌ Промокод не найден", show_alert=True)
-        return
-    
-    code = promo.code
-    success = await delete_promocode(db, promo)
-    
-    if success:
-        await callback.answer(f"✅ Промокод {code} удален", show_alert=True)
-        await show_promocodes_list(callback, db_user, db)
-    else:
-        await callback.answer("❌ Ошибка удаления промокода", show_alert=True)
-
-
-@admin_required
-@error_handler
-async def show_promocode_stats(
-    callback: types.CallbackQuery,
-    db_user: User,
-    db: AsyncSession
-):
-    promo_id = int(callback.data.split('_')[-1])
-    
-    promo = await db.get(PromoCode, promo_id)
-    if not promo:
-        await callback.answer("❌ Промокод не найден", show_alert=True)
-        return
-    
-    stats = await get_promocode_statistics(db, promo_id)
-    
-    text = f"""
-📊 <b>Статистика промокода</b> <code>{promo.code}</code>
-
-📈 <b>Общая статистика:</b>
-- Всего использований: {stats['total_uses']}
-- Использований сегодня: {stats['today_uses']}
-- Осталось использований: {promo.max_uses - promo.current_uses}
-
-📅 <b>Последние использования:</b>
-"""
-    
-    for use in stats['recent_uses'][:5]:
-        use_date = format_datetime(use.used_at)
-        text += f"- {use_date} (ID: {use.user_id})\n"
-    
-    if not stats['recent_uses']:
-        text += "- Пока не было использований\n"
-    
-    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+    keyboard.extend([
         [
             types.InlineKeyboardButton(
                 text="⬅️ Назад", 
@@ -312,7 +265,160 @@ async def show_promocode_stats(
         ]
     ])
     
+    await callback.message.edit_text(
+        text,
+        reply_markup=types.InlineKeyboardMarkup(inline_keyboard=keyboard)
+    )
+    await callback.answer()
+
+
+@admin_required
+@error_handler
+async def start_edit_promocode_date(
+    callback: types.CallbackQuery,
+    db_user: User,
+    state: FSMContext
+):
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    await state.update_data(
+        editing_promo_id=promo_id,
+        edit_action="date"
+    )
+    
+    text = f"""
+📅 <b>Изменение даты окончания промокода</b>
+
+Введите количество дней до окончания (от текущего момента):
+• Введите <b>0</b> для бессрочного промокода
+• Введите положительное число для установки срока
+
+<i>Например: 30 (промокод будет действовать 30 дней)</i>
+
+ID промокода: {promo_id}
+"""
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [types.InlineKeyboardButton(text="❌ Отмена", callback_data=f"promo_edit_{promo_id}")]
+    ])
+    
     await callback.message.edit_text(text, reply_markup=keyboard)
+    await state.set_state(AdminStates.setting_promocode_expiry)
+    await callback.answer()
+
+
+@admin_required
+@error_handler
+async def start_edit_promocode_amount(
+    callback: types.CallbackQuery,
+    db_user: User,
+    state: FSMContext
+):
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    await state.update_data(
+        editing_promo_id=promo_id,
+        edit_action="amount"
+    )
+    
+    text = f"""
+💰 <b>Изменение суммы бонуса промокода</b>
+
+Введите новую сумму в рублях:
+<i>Например: 500</i>
+
+ID промокода: {promo_id}
+"""
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [types.InlineKeyboardButton(text="❌ Отмена", callback_data=f"promo_edit_{promo_id}")]
+    ])
+    
+    await callback.message.edit_text(text, reply_markup=keyboard)
+    await state.set_state(AdminStates.setting_promocode_value)
+    await callback.answer()
+
+@admin_required
+@error_handler
+async def start_edit_promocode_days(
+    callback: types.CallbackQuery,
+    db_user: User,
+    state: FSMContext
+):
+    # ИСПРАВЛЕНИЕ: берем последний элемент как ID
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    await state.update_data(
+        editing_promo_id=promo_id,
+        edit_action="days"
+    )
+    
+    text = f"""
+📅 <b>Изменение количества дней подписки</b>
+
+Введите новое количество дней:
+<i>Например: 30</i>
+
+ID промокода: {promo_id}
+"""
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [types.InlineKeyboardButton(text="❌ Отмена", callback_data=f"promo_edit_{promo_id}")]
+    ])
+    
+    await callback.message.edit_text(text, reply_markup=keyboard)
+    await state.set_state(AdminStates.setting_promocode_value)
+    await callback.answer()
+
+
+@admin_required
+@error_handler
+async def start_edit_promocode_uses(
+    callback: types.CallbackQuery,
+    db_user: User,
+    state: FSMContext
+):
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    await state.update_data(
+        editing_promo_id=promo_id,
+        edit_action="uses"
+    )
+    
+    text = f"""
+📊 <b>Изменение максимального количества использований</b>
+
+Введите новое количество использований:
+• Введите <b>0</b> для безлимитных использований
+• Введите положительное число для ограничения
+
+<i>Например: 100</i>
+
+ID промокода: {promo_id}
+"""
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [types.InlineKeyboardButton(text="❌ Отмена", callback_data=f"promo_edit_{promo_id}")]
+    ])
+    
+    await callback.message.edit_text(text, reply_markup=keyboard)
+    await state.set_state(AdminStates.setting_promocode_uses)
     await callback.answer()
 
 
@@ -375,7 +481,6 @@ async def process_promocode_code(
         await message.answer("❌ Код должен содержать только латинские буквы и цифры (3-20 символов)")
         return
     
-    from app.database.crud.promocode import get_promocode_by_code
     existing = await get_promocode_by_code(db, code)
     if existing:
         await message.answer("❌ Промокод с таким кодом уже существует")
@@ -411,12 +516,18 @@ async def process_promocode_code(
 async def process_promocode_value(
     message: types.Message,
     db_user: User,
-    state: FSMContext
+    state: FSMContext,
+    db: AsyncSession
 ):
+    data = await state.get_data()
+    
+    if data.get('editing_promo_id'):
+        await handle_edit_value(message, db_user, state, db)
+        return
+    
     try:
         value = int(message.text.strip())
         
-        data = await state.get_data()
         promo_type = data.get('promocode_type')
         
         if promo_type == "balance" and (value < 1 or value > 10000):
@@ -437,13 +548,72 @@ async def process_promocode_value(
         await message.answer("❌ Введите корректное число")
 
 
+async def handle_edit_value(
+    message: types.Message,
+    db_user: User,
+    state: FSMContext,
+    db: AsyncSession
+):
+    data = await state.get_data()
+    promo_id = data.get('editing_promo_id')
+    edit_action = data.get('edit_action')
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await message.answer("❌ Промокод не найден")
+        await state.clear()
+        return
+    
+    try:
+        value = int(message.text.strip())
+        
+        if edit_action == "amount":
+            if value < 1 or value > 10000:
+                await message.answer("❌ Сумма должна быть от 1 до 10,000 рублей")
+                return
+            
+            await update_promocode(db, promo, balance_bonus_kopeks=value * 100)
+            await message.answer(
+                f"✅ Сумма бонуса изменена на {value}₽",
+                reply_markup=types.InlineKeyboardMarkup(inline_keyboard=[
+                    [types.InlineKeyboardButton(text="🎫 К промокоду", callback_data=f"promo_manage_{promo_id}")]
+                ])
+            )
+            
+        elif edit_action == "days":
+            if value < 1 or value > 3650:
+                await message.answer("❌ Количество дней должно быть от 1 до 3650")
+                return
+            
+            await update_promocode(db, promo, subscription_days=value)
+            await message.answer(
+                f"✅ Количество дней изменено на {value}",
+                reply_markup=types.InlineKeyboardMarkup(inline_keyboard=[
+                    [types.InlineKeyboardButton(text="🎫 К промокоду", callback_data=f"promo_manage_{promo_id}")]
+                ])
+            )
+        
+        await state.clear()
+        logger.info(f"Промокод {promo.code} отредактирован администратором {db_user.telegram_id}: {edit_action} = {value}")
+        
+    except ValueError:
+        await message.answer("❌ Введите корректное число")
+
+
 @admin_required
 @error_handler
 async def process_promocode_uses(
     message: types.Message,
     db_user: User,
-    state: FSMContext
+    state: FSMContext,
+    db: AsyncSession
 ):
+    data = await state.get_data()
+    
+    if data.get('editing_promo_id'):
+        await handle_edit_uses(message, db_user, state, db)
+        return
+    
     try:
         max_uses = int(message.text.strip())
         
@@ -465,6 +635,54 @@ async def process_promocode_uses(
         await message.answer("❌ Введите корректное число")
 
 
+async def handle_edit_uses(
+    message: types.Message,
+    db_user: User,
+    state: FSMContext,
+    db: AsyncSession
+):
+    data = await state.get_data()
+    promo_id = data.get('editing_promo_id')
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await message.answer("❌ Промокод не найден")
+        await state.clear()
+        return
+    
+    try:
+        max_uses = int(message.text.strip())
+        
+        if max_uses < 0 or max_uses > 100000:
+            await message.answer("❌ Количество использований должно быть от 0 до 100,000")
+            return
+        
+        if max_uses == 0:
+            max_uses = 999999
+        
+        if max_uses < promo.current_uses:
+            await message.answer(
+                f"❌ Новый лимит ({max_uses}) не может быть меньше текущих использований ({promo.current_uses})"
+            )
+            return
+        
+        await update_promocode(db, promo, max_uses=max_uses)
+        
+        uses_text = "безлимитное" if max_uses == 999999 else str(max_uses)
+        await message.answer(
+            f"✅ Максимальное количество использований изменено на {uses_text}",
+            reply_markup=types.InlineKeyboardMarkup(inline_keyboard=[
+                [types.InlineKeyboardButton(text="🎫 К промокоду", callback_data=f"promo_manage_{promo_id}")]
+            ])
+        )
+        
+        await state.clear()
+        logger.info(f"Промокод {promo.code} отредактирован администратором {db_user.telegram_id}: max_uses = {max_uses}")
+        
+    except ValueError:
+        await message.answer("❌ Введите корректное число")
+
+
 @admin_required
 @error_handler
 async def process_promocode_expiry(
@@ -473,6 +691,12 @@ async def process_promocode_expiry(
     state: FSMContext,
     db: AsyncSession
 ):
+    data = await state.get_data()
+    
+    if data.get('editing_promo_id'):
+        await handle_edit_expiry(message, db_user, state, db)
+        return
+    
     try:
         expiry_days = int(message.text.strip())
         
@@ -480,7 +704,6 @@ async def process_promocode_expiry(
             await message.answer("❌ Срок действия должен быть от 0 до 3650 дней")
             return
         
-        data = await state.get_data()
         code = data.get('promocode_code')
         promo_type = data.get('promocode_type')
         value = data.get('promocode_value', 0)
@@ -544,6 +767,209 @@ async def process_promocode_expiry(
         await message.answer("❌ Введите корректное число дней")
 
 
+async def handle_edit_expiry(
+    message: types.Message,
+    db_user: User,
+    state: FSMContext,
+    db: AsyncSession
+):
+    data = await state.get_data()
+    promo_id = data.get('editing_promo_id')
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await message.answer("❌ Промокод не найден")
+        await state.clear()
+        return
+    
+    try:
+        expiry_days = int(message.text.strip())
+        
+        if expiry_days < 0 or expiry_days > 3650:
+            await message.answer("❌ Срок действия должен быть от 0 до 3650 дней")
+            return
+        
+        valid_until = None
+        if expiry_days > 0:
+            valid_until = datetime.utcnow() + timedelta(days=expiry_days)
+        
+        await update_promocode(db, promo, valid_until=valid_until)
+        
+        if valid_until:
+            expiry_text = f"до {format_datetime(valid_until)}"
+        else:
+            expiry_text = "бессрочно"
+            
+        await message.answer(
+            f"✅ Срок действия промокода изменен: {expiry_text}",
+            reply_markup=types.InlineKeyboardMarkup(inline_keyboard=[
+                [types.InlineKeyboardButton(text="🎫 К промокоду", callback_data=f"promo_manage_{promo_id}")]
+            ])
+        )
+        
+        await state.clear()
+        logger.info(f"Промокод {promo.code} отредактирован администратором {db_user.telegram_id}: expiry = {expiry_days} дней")
+        
+    except ValueError:
+        await message.answer("❌ Введите корректное число дней")
+
+
+@admin_required
+@error_handler
+async def toggle_promocode_status(
+    callback: types.CallbackQuery,
+    db_user: User,
+    db: AsyncSession
+):
+    promo_id = int(callback.data.split('_')[-1])
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await callback.answer("❌ Промокод не найден", show_alert=True)
+        return
+    
+    new_status = not promo.is_active
+    await update_promocode(db, promo, is_active=new_status)
+    
+    status_text = "активирован" if new_status else "деактивирован"
+    await callback.answer(f"✅ Промокод {status_text}", show_alert=True)
+    
+    await show_promocode_management(callback, db_user, db)
+
+
+@admin_required
+@error_handler
+async def confirm_delete_promocode(
+    callback: types.CallbackQuery,
+    db_user: User,
+    db: AsyncSession
+):
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await callback.answer("❌ Промокод не найден", show_alert=True)
+        return
+    
+    text = f"""
+⚠️ <b>Подтверждение удаления</b>
+
+Вы действительно хотите удалить промокод <code>{promo.code}</code>?
+
+📊 <b>Информация о промокоде:</b>
+• Использований: {promo.current_uses}/{promo.max_uses}
+• Статус: {'Активен' if promo.is_active else 'Неактивен'}
+
+<b>⚠️ Внимание:</b> Это действие нельзя отменить!
+
+ID: {promo_id}
+"""
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [
+            types.InlineKeyboardButton(
+                text="✅ Да, удалить", 
+                callback_data=f"promo_delete_confirm_{promo.id}"
+            ),
+            types.InlineKeyboardButton(
+                text="❌ Отмена", 
+                callback_data=f"promo_manage_{promo.id}"
+            )
+        ]
+    ])
+    
+    await callback.message.edit_text(text, reply_markup=keyboard)
+    await callback.answer()
+
+@admin_required
+@error_handler
+async def delete_promocode_confirmed(
+    callback: types.CallbackQuery,
+    db_user: User,
+    db: AsyncSession
+):
+    try:
+        promo_id = int(callback.data.split('_')[-1])
+    except (ValueError, IndexError):
+        await callback.answer("❌ Ошибка получения ID промокода", show_alert=True)
+        return
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await callback.answer("❌ Промокод не найден", show_alert=True)
+        return
+    
+    code = promo.code
+    success = await delete_promocode(db, promo)
+    
+    if success:
+        await callback.answer(f"✅ Промокод {code} удален", show_alert=True)
+        await show_promocodes_list(callback, db_user, db)
+    else:
+        await callback.answer("❌ Ошибка удаления промокода", show_alert=True)
+
+
+@admin_required
+@error_handler
+async def show_promocode_stats(
+    callback: types.CallbackQuery,
+    db_user: User,
+    db: AsyncSession
+):
+    promo_id = int(callback.data.split('_')[-1])
+    
+    promo = await db.get(PromoCode, promo_id)
+    if not promo:
+        await callback.answer("❌ Промокод не найден", show_alert=True)
+        return
+    
+    stats = await get_promocode_statistics(db, promo_id)
+    
+    text = f"""
+📊 <b>Статистика промокода</b> <code>{promo.code}</code>
+
+📈 <b>Общая статистика:</b>
+- Всего использований: {stats['total_uses']}
+- Использований сегодня: {stats['today_uses']}
+- Осталось использований: {promo.max_uses - promo.current_uses}
+
+📅 <b>Последние использования:</b>
+"""
+    
+    if stats['recent_uses']:
+        for use in stats['recent_uses'][:5]:
+            use_date = format_datetime(use.used_at)
+            
+            if hasattr(use, 'user_username') and use.user_username:
+                user_display = f"@{use.user_username}"
+            elif hasattr(use, 'user_full_name') and use.user_full_name:
+                user_display = use.user_full_name
+            elif hasattr(use, 'user_telegram_id'):
+                user_display = f"ID{use.user_telegram_id}"
+            else:
+                user_display = f"ID{use.user_id}"
+            
+            text += f"- {use_date} | {user_display}\n"
+    else:
+        text += "- Пока не было использований\n"
+    
+    keyboard = types.InlineKeyboardMarkup(inline_keyboard=[
+        [
+            types.InlineKeyboardButton(
+                text="⬅️ Назад", 
+                callback_data=f"promo_manage_{promo.id}"
+            )
+        ]
+    ])
+    
+    await callback.message.edit_text(text, reply_markup=keyboard)
+    await callback.answer()
+
+
 def register_handlers(dp: Dispatcher):
     dp.callback_query.register(show_promocodes_menu, F.data == "admin_promocodes")
     dp.callback_query.register(show_promocodes_list, F.data == "admin_promo_list")
@@ -552,11 +978,22 @@ def register_handlers(dp: Dispatcher):
     
     dp.callback_query.register(show_promocode_management, F.data.startswith("promo_manage_"))
     dp.callback_query.register(toggle_promocode_status, F.data.startswith("promo_toggle_"))
-    dp.callback_query.register(confirm_delete_promocode, F.data.startswith("promo_delete_"))
-    dp.callback_query.register(delete_promocode_confirmed, F.data.startswith("promo_delete_confirm_"))
     dp.callback_query.register(show_promocode_stats, F.data.startswith("promo_stats_"))
     
+    dp.callback_query.register(start_edit_promocode_date, F.data.startswith("promo_edit_date_"))
+    dp.callback_query.register(start_edit_promocode_amount, F.data.startswith("promo_edit_amount_"))
+    dp.callback_query.register(start_edit_promocode_days, F.data.startswith("promo_edit_days_"))
+    dp.callback_query.register(start_edit_promocode_uses, F.data.startswith("promo_edit_uses_"))
+    
+    dp.callback_query.register(
+        show_promocode_edit_menu, 
+        F.data.regexp(r"^promo_edit_\d+$")
+    )
+    
+    dp.callback_query.register(delete_promocode_confirmed, F.data.startswith("promo_delete_confirm_"))
+    dp.callback_query.register(confirm_delete_promocode, F.data.startswith("promo_delete_"))
+    
     dp.message.register(process_promocode_code, AdminStates.creating_promocode)
     dp.message.register(process_promocode_value, AdminStates.setting_promocode_value)
     dp.message.register(process_promocode_uses, AdminStates.setting_promocode_uses)
-    dp.message.register(process_promocode_expiry, AdminStates.setting_promocode_expiry)
+    dp.message.register(process_promocode_expiry, AdminStates.setting_promocode_expiry)

app/handlers/referral.py

@@ -85,13 +85,7 @@ async def create_invite_message(
         [
             types.InlineKeyboardButton(
                 text="📤 Поделиться",
-                switch_inline_query=invite_text
-            )
-        ],
-        [
-            types.InlineKeyboardButton(
-                text="📋 Скопировать ссылку",
-                callback_data="copy_referral_link"
+                switch_inline_query=invite_text 
             )
         ],
         [
@@ -104,7 +98,7 @@ async def create_invite_message(
     
     await callback.message.edit_text(
         f"📝 <b>Приглашение создано!</b>\n\n"
-        f"Используйте кнопку ниже для отправки приглашения или скопируйте текст:\n\n"
+        f"Нажмите кнопку «📤 Поделиться» чтобы отправить приглашение в любой чат, или скопируйте текст ниже:\n\n"
         f"<code>{invite_text}</code>",
         reply_markup=keyboard,
         parse_mode="HTML"
@@ -112,20 +106,6 @@ async def create_invite_message(
     await callback.answer()
 
 
-async def copy_referral_link(
-    callback: types.CallbackQuery,
-    db_user: User
-):
-    
-    bot_username = (await callback.bot.get_me()).username
-    referral_link = f"https://t.me/{bot_username}?start={db_user.referral_code}"
-    
-    await callback.answer(
-        f"Ссылка скопирована: {referral_link}",
-        show_alert=True
-    )
-
-
 def register_handlers(dp: Dispatcher):
     
     dp.callback_query.register(
@@ -137,8 +117,3 @@ def register_handlers(dp: Dispatcher):
         create_invite_message,
         F.data == "referral_create_invite"
     )
-    
-    dp.callback_query.register(
-        copy_referral_link,
-        F.data == "copy_referral_link"
-    )

app/handlers/start.py

@@ -1,4 +1,5 @@
 import logging
+from datetime import datetime
 from aiogram import Dispatcher, types, F
 from aiogram.filters import Command, StateFilter
 from aiogram.fsm.context import FSMContext
@@ -20,6 +21,62 @@ from app.utils.user_utils import generate_unique_referral_code
 logger = logging.getLogger(__name__)
 
 
+async def handle_potential_referral_code(
+    message: types.Message, 
+    state: FSMContext, 
+    db: AsyncSession
+):
+    current_state = await state.get_state()
+    logger.info(f"🔍 REFERRAL CHECK: Проверка сообщения '{message.text}' в состоянии {current_state}")
+    
+    if current_state not in [
+        RegistrationStates.waiting_for_rules_accept.state,
+        RegistrationStates.waiting_for_referral_code.state,
+        None 
+    ]:
+        return False
+    
+    user = await get_user_by_telegram_id(db, message.from_user.id)
+    if user and user.status == UserStatus.ACTIVE.value:
+        return False
+    
+    potential_code = message.text.strip()
+    if len(potential_code) < 4 or len(potential_code) > 20:
+        return False
+    
+    referrer = await get_user_by_referral_code(db, potential_code)
+    if not referrer:
+        await message.answer(
+            "❌ Неверный реферальный код.\n\n"
+            "💡 Если у вас есть реферальный код, убедитесь что он введен правильно.\n"
+            "⏭️ Для продолжения регистрации без реферального кода используйте команду /start"
+        )
+        return True 
+    
+    data = await state.get_data() or {}
+    data['referral_code'] = potential_code
+    data['referrer_id'] = referrer.id
+    await state.set_data(data)
+    
+    await message.answer("✅ Реферальный код принят!")
+    logger.info(f"✅ Реферальный код {potential_code} применен для пользователя {message.from_user.id}")
+    
+    if current_state != RegistrationStates.waiting_for_referral_code.state:
+        language = data.get('language', 'ru')
+        texts = get_texts(language)
+        
+        await message.answer(
+            texts.RULES_TEXT,
+            reply_markup=get_rules_keyboard(language)
+        )
+        await state.set_state(RegistrationStates.waiting_for_rules_accept)
+        logger.info("📋 Правила отправлены после ввода реферального кода")
+    else:
+        await complete_registration(message, state, db)
+    
+    return True 
+
+
 async def cmd_start(message: types.Message, state: FSMContext, db: AsyncSession, db_user=None):
     logger.info(f"🚀 START: Обработка /start от {message.from_user.id}")
     
@@ -36,6 +93,37 @@ async def cmd_start(message: types.Message, state: FSMContext, db: AsyncSession,
     
     if user and user.status != UserStatus.DELETED.value:
         logger.info(f"✅ Активный пользователь найден: {user.telegram_id}")
+        
+        profile_updated = False
+        
+        if user.username != message.from_user.username:
+            old_username = user.username
+            user.username = message.from_user.username
+            logger.info(f"📝 Username обновлен: '{old_username}' → '{user.username}'")
+            profile_updated = True
+        
+        if user.first_name != message.from_user.first_name:
+            old_first_name = user.first_name
+            user.first_name = message.from_user.first_name
+            logger.info(f"📝 Имя обновлено: '{old_first_name}' → '{user.first_name}'")
+            profile_updated = True
+        
+        if user.last_name != message.from_user.last_name:
+            old_last_name = user.last_name
+            user.last_name = message.from_user.last_name
+            logger.info(f"📝 Фамилия обновлена: '{old_last_name}' → '{user.last_name}'")
+            profile_updated = True
+        
+        user.last_activity = datetime.utcnow()
+        
+        if profile_updated:
+            user.updated_at = datetime.utcnow()
+            await db.commit()
+            await db.refresh(user)
+            logger.info(f"💾 Профиль пользователя {user.telegram_id} обновлен")
+        else:
+            await db.commit()
+        
         texts = get_texts(user.language)
         
         if referral_code and not user.referred_by_id:
@@ -107,6 +195,13 @@ async def cmd_start(message: types.Message, state: FSMContext, db: AsyncSession,
             user.remnawave_uuid = None
             user.has_had_paid_subscription = False
             user.referred_by_id = None
+            
+            user.username = message.from_user.username
+            user.first_name = message.from_user.first_name
+            user.last_name = message.from_user.last_name
+            user.updated_at = datetime.utcnow()
+            user.last_activity = datetime.utcnow()
+            
             from app.utils.user_utils import generate_unique_referral_code
             user.referral_code = await generate_unique_referral_code(db, user.telegram_id)
             
@@ -299,6 +394,10 @@ async def complete_registration_from_callback(
         logger.warning(f"⚠️ Пользователь {callback.from_user.id} уже активен! Показываем главное меню.")
         texts = get_texts(existing_user.language)
         
+        data = await state.get_data()
+        if data.get('referral_code') and not existing_user.referred_by_id:
+            await callback.message.answer("ℹ️ Вы уже зарегистрированы в системе. Реферальная ссылка не может быть применена.")
+        
         has_active_subscription = existing_user.subscription is not None
         subscription_is_active = False
         
@@ -362,7 +461,7 @@ async def complete_registration_from_callback(
         user = existing_user
         logger.info(f"✅ Пользователь {callback.from_user.id} восстановлен")
         
-    else:
+    elif not existing_user:
         logger.info(f"🆕 Создаем нового пользователя {callback.from_user.id}")
         
         referral_code = await generate_unique_referral_code(db, callback.from_user.id)
@@ -377,6 +476,20 @@ async def complete_registration_from_callback(
             referred_by_id=referrer_id,
             referral_code=referral_code 
         )
+    else:
+        logger.info(f"🔄 Обновляем существующего пользователя {callback.from_user.id}")
+        existing_user.status = UserStatus.ACTIVE.value
+        existing_user.language = language
+        if referrer_id and not existing_user.referred_by_id:
+            existing_user.referred_by_id = referrer_id
+        
+        from datetime import datetime
+        existing_user.updated_at = datetime.utcnow()
+        existing_user.last_activity = datetime.utcnow()
+        
+        await db.commit()
+        await db.refresh(existing_user)
+        user = existing_user
     
     if referrer_id:
         try:
@@ -451,6 +564,42 @@ async def complete_registration(
     
     existing_user = await get_user_by_telegram_id(db, message.from_user.id)
     
+    if existing_user and existing_user.status == UserStatus.ACTIVE.value:
+        logger.warning(f"⚠️ Пользователь {message.from_user.id} уже активен! Показываем главное меню.")
+        texts = get_texts(existing_user.language)
+        
+        data = await state.get_data()
+        if data.get('referral_code') and not existing_user.referred_by_id:
+            await message.answer("ℹ️ Вы уже зарегистрированы в системе. Реферальная ссылка не может быть применена.")
+        
+        has_active_subscription = existing_user.subscription is not None
+        subscription_is_active = False
+        
+        if existing_user.subscription:
+            subscription_is_active = existing_user.subscription.is_active
+        
+        try:
+            await message.answer(
+                texts.MAIN_MENU.format(
+                    user_name=existing_user.full_name,
+                    balance=texts.format_price(existing_user.balance_kopeks),
+                    subscription_status=_get_subscription_status(existing_user, texts)
+                ),
+                reply_markup=get_main_menu_keyboard(
+                    language=existing_user.language,
+                    is_admin=settings.is_admin(existing_user.telegram_id),
+                    has_had_paid_subscription=existing_user.has_had_paid_subscription,
+                    has_active_subscription=has_active_subscription,
+                    subscription_is_active=subscription_is_active
+                )
+            )
+        except Exception as e:
+            logger.error(f"Ошибка при показе главного меню существующему пользователю: {e}")
+            await message.answer(f"Добро пожаловать, {existing_user.full_name}!")
+        
+        await state.clear()
+        return
+    
     data = await state.get_data()
     language = data.get('language', 'ru')
     texts = get_texts(language)
@@ -483,7 +632,7 @@ async def complete_registration(
         user = existing_user
         logger.info(f"✅ Пользователь {message.from_user.id} восстановлен")
         
-    else:
+    elif not existing_user:
         logger.info(f"🆕 Создаем нового пользователя {message.from_user.id}")
         
         referral_code = await generate_unique_referral_code(db, message.from_user.id)
@@ -498,6 +647,20 @@ async def complete_registration(
             referred_by_id=referrer_id,
             referral_code=referral_code
         )
+    else:
+        logger.info(f"🔄 Обновляем существующего пользователя {message.from_user.id}")
+        existing_user.status = UserStatus.ACTIVE.value
+        existing_user.language = language
+        if referrer_id and not existing_user.referred_by_id:
+            existing_user.referred_by_id = referrer_id
+        
+        from datetime import datetime
+        existing_user.updated_at = datetime.utcnow()
+        existing_user.last_activity = datetime.utcnow()
+        
+        await db.commit()
+        await db.refresh(existing_user)
+        user = existing_user
     
     if referrer_id:
         try:
@@ -614,4 +777,13 @@ def register_handlers(dp: Dispatcher):
     )
     logger.info("✅ Зарегистрирован process_referral_code_input")
     
+    dp.message.register(
+        handle_potential_referral_code,
+        StateFilter(
+            RegistrationStates.waiting_for_rules_accept,
+            RegistrationStates.waiting_for_referral_code
+        )
+    )
+    logger.info("✅ Зарегистрирован handle_potential_referral_code")
+    
     logger.info("🔧 === КОНЕЦ регистрации обработчиков start.py ===")

app/middlewares/auth.py

@@ -1,4 +1,5 @@
 import logging
+from datetime import datetime
 from typing import Callable, Dict, Any, Awaitable
 from aiogram import BaseMiddleware
 from aiogram.types import Message, CallbackQuery, TelegramObject, User as TgUser
@@ -56,7 +57,7 @@ class AuthMiddleware(BaseMiddleware):
                     )
                     
                     if is_registration_process:
-                        logger.info(f"📝 Пропускаем пользователя {user.id} в процессе регистрации")
+                        logger.info(f"🔍 Пропускаем пользователя {user.id} в процессе регистрации")
                         data['db'] = db
                         data['db_user'] = None
                         data['is_admin'] = False
@@ -124,8 +125,33 @@ class AuthMiddleware(BaseMiddleware):
                             logger.info(f"❌ Удаленный пользователь {user.id} попытался использовать бота без /start")
                             return
                     
-                    from datetime import datetime
+                    
+                    profile_updated = False
+                    
+                    if db_user.username != user.username:
+                        old_username = db_user.username
+                        db_user.username = user.username
+                        logger.info(f"📝 [Middleware] Username обновлен для {user.id}: '{old_username}' → '{db_user.username}'")
+                        profile_updated = True
+                    
+                    if db_user.first_name != user.first_name:
+                        old_first_name = db_user.first_name
+                        db_user.first_name = user.first_name
+                        logger.info(f"📝 [Middleware] Имя обновлено для {user.id}: '{old_first_name}' → '{db_user.first_name}'")
+                        profile_updated = True
+                    
+                    if db_user.last_name != user.last_name:
+                        old_last_name = db_user.last_name
+                        db_user.last_name = user.last_name
+                        logger.info(f"📝 [Middleware] Фамилия обновлена для {user.id}: '{old_last_name}' → '{db_user.last_name}'")
+                        profile_updated = True
+                    
                     db_user.last_activity = datetime.utcnow()
+                    
+                    if profile_updated:
+                        db_user.updated_at = datetime.utcnow()
+                        logger.info(f"💾 [Middleware] Профиль пользователя {user.id} обновлен в middleware")
+                    
                     await db.commit()
 
                 data['db'] = db