Merge pull request 'Fix UserService method call in P2P chat handler' (#16) from v2_functions into master

Reviewed-on: #16

src/handlers/p2p_chat.py

@@ -30,6 +30,35 @@ def is_admin(user_id: int) -> bool:
     return user_id in ADMIN_IDS
 
 
+def format_sender_name(user: User, is_current_user: bool = False, current_user_is_admin: bool = False) -> str:
+    """
+    Форматирует имя отправителя для отображения в чате
+    
+    Args:
+        user: Объект пользователя
+        is_current_user: Текущий ли это пользователь
+        current_user_is_admin: Админ ли текущий пользователь
+    
+    Returns:
+        Отформатированное имя
+    """
+    if is_current_user:
+        return "🔵 Вы"
+    
+    # Если это администратор и текущий пользователь не админ - показываем "Админ"
+    if user.is_admin and not current_user_is_admin:
+        return "🔵 Админ"
+    
+    # Формируем базовое имя (используем nickname из профиля)
+    name = user.nickname or user.first_name or f"@{user.username}" or "Unknown"
+    
+    # Добавляем информацию о карте если пользователь админ и текущий юзер админ
+    if current_user_is_admin and user.club_card_number:
+        name += f" (карта: {user.club_card_number})"
+    
+    return f"🔵 {name}"
+
+
 @router.message(CaseInsensitiveCommand("chat"))
 async def show_chat_menu(message: Message, state: FSMContext):
     """
@@ -106,7 +135,7 @@ async def select_recipient(callback: CallbackQuery, state: FSMContext):
     # Создаём кнопки с пользователями (по 1 на строку)
     buttons = []
     for user in users[:20]:  # Ограничение 20 пользователей на странице
-        display_name = f"@{user.username}" if user.username else user.first_name
+        display_name = user.nickname or f"@{user.username}" or user.first_name or "Unknown"
         if user.club_card_number:
             display_name += f" (карта: {user.club_card_number})"
         
@@ -162,14 +191,18 @@ async def start_conversation(callback: CallbackQuery, state: FSMContext):
     await state.update_data(recipient_id=recipient.id, recipient_telegram_id=recipient.telegram_id)
     await state.set_state(P2PChatStates.chatting)
     
-    recipient_name = f"@{recipient.username}" if recipient.username else recipient.first_name
+    recipient_name = recipient.nickname or f"@{recipient.username}" or recipient.first_name or "Unknown"
     
     text = f"💬 <b>Диалог с {recipient_name}</b>\n\n"
     
     if messages:
         text += "📝 <b>Последние сообщения:</b>\n\n"
         for msg in reversed(messages[-5:]):  # Последние 5 сообщений
-            sender_name = "Вы" if msg.sender_id == sender.id else recipient_name
+            # Определяем имя отправителя
+            is_current = msg.sender_id == sender.id
+            user_for_display = sender if is_current else recipient
+            sender_name = format_sender_name(user_for_display, is_current, is_admin(sender.telegram_id))
+            
             msg_text = msg.text[:50] + "..." if msg.text and len(msg.text) > 50 else (msg.text or f"[{msg.message_type}]")
             text += f"• {sender_name}: {msg_text}\n"
         text += "\n"
@@ -204,7 +237,7 @@ async def show_conversations(callback: CallbackQuery):
             last_name=callback.from_user.last_name
         )
         
-        conversations = await P2PMessageService.get_recent_conversations(session, user.id, limit=10)
+        conversations = await P2PMessageService.get_recent_conversations(session, sender.id, limit=10)
     
     if not conversations:
         await callback.message.edit_text(
@@ -217,7 +250,7 @@ async def show_conversations(callback: CallbackQuery):
     
     buttons = []
     for peer, last_msg, unread in conversations:
-        peer_name = f"@{peer.username}" if peer.username else peer.first_name
+        peer_name = peer.nickname or f"@{peer.username}" or peer.first_name or "Unknown"
         
         # Иконка в зависимости от непрочитанных
         icon = "🔴" if unread > 0 else "💬"
@@ -234,7 +267,11 @@ async def show_conversations(callback: CallbackQuery):
             callback_data=f"p2p:user:{peer.id}"
         )])
         
-        text += f"{icon} <b>{peer_name}</b>\n"
+        text += f"{icon} <b>{peer_name}</b>"
+        # Показываем номер карты если есть
+        if peer.club_card_number:
+            text += f" (карта: {peer.club_card_number})"
+        text += "\n"
         text += f"   {preview}\n"
         if unread > 0:
             text += f"   📨 Непрочитанных: {unread}\n"
@@ -268,12 +305,53 @@ async def end_conversation(callback: CallbackQuery, state: FSMContext):
 async def back_to_menu(callback: CallbackQuery, state: FSMContext):
     """Вернуться в главное меню"""
     await callback.answer()
+    await state.clear()
     
-    # Имитируем команду /chat
+    async with async_session_maker() as session:
-    fake_message = callback.message
+        user = await UserService.get_or_create_user(
-    fake_message.from_user = callback.from_user
+            session,
+            callback.from_user.id,
+            username=callback.from_user.username,
+            first_name=callback.from_user.first_name,
+            last_name=callback.from_user.last_name
+        )
         
-    await show_chat_menu(fake_message, state)
+        if not user:
+            await callback.message.edit_text("❌ Вы не зарегистрированы. Используйте /start")
+            return
+        
+        # Получаем количество непрочитанных сообщений
+        unread_count = await P2PMessageService.get_unread_count(session, user.id)
+        
+        # Получаем последние диалоги
+        recent = await P2PMessageService.get_recent_conversations(session, user.id, limit=5)
+    
+    text = "💬 <b>Чат</b>\n\n"
+    
+    if unread_count > 0:
+        text += f"📨 У вас <b>{unread_count}</b> непрочитанных сообщений\n\n"
+    
+    text += "Выберите действие:"
+    
+    buttons = [
+        [InlineKeyboardButton(
+            text="✉️ Написать пользователю",
+            callback_data="p2p:select_user"
+        )],
+        [InlineKeyboardButton(
+            text="📋 Мои диалоги",
+            callback_data="p2p:my_conversations"
+        )]
+    ]
+    
+    if recent:
+        text += "\n\n<b>Последние диалоги:</b>\n"
+        for peer, last_msg, unread in recent:
+            unread_badge = f" ({unread})" if unread > 0 else ""
+            text += f"  • @{peer.username or peer.first_name}{unread_badge}\n"
+    
+    kb = InlineKeyboardMarkup(inline_keyboard=buttons)
+    await callback.message.edit_text(text, reply_markup=kb, parse_mode="HTML")
 
 
 # Обработчик сообщений в состоянии chatting
@@ -301,7 +379,19 @@ async def handle_p2p_message(message: Message, state: FSMContext):
             first_name=message.from_user.first_name,
             last_name=message.from_user.last_name
         )
-        sender_name = f"@{sender.username}" if sender.username else sender.first_name
+        
+        # Получаем информацию о получателе для определения как подписать сообщение
+        recipient = await UserService.get_user_by_telegram_id(session, recipient_telegram_id)
+        
+        # Формируем подпись сообщения для получателя
+        if sender.is_admin:
+            sender_name = "АДМИН"
+        else:
+            sender_name = sender.nickname or f"@{sender.username}" or sender.first_name or "Unknown"
+        
+        # Добавляем карту если получатель админ
+        if recipient and recipient.is_admin and sender.club_card_number:
+            sender_name += f" (карта: {sender.club_card_number})"
         
         # Определяем тип сообщения
         message_type = "text"
@@ -326,28 +416,28 @@ async def handle_p2p_message(message: Message, state: FSMContext):
             if message_type == "text":
                 sent = await message.bot.send_message(
                     recipient_telegram_id,
-                    f"💬 <b>Сообщение от {sender_name}:</b>\n\n{text}",
+                    f"<b>{sender_name}</b>\n\n{text}",
                     parse_mode="HTML"
                 )
             elif message_type == "photo":
                 sent = await message.bot.send_photo(
                     recipient_telegram_id,
                     photo=file_id,
-                    caption=f"💬 <b>Фото от {sender_name}</b>\n\n{text or ''}" ,
+                    caption=f"<b>{sender_name}</b>\n\n{text or ''}" ,
                     parse_mode="HTML"
                 )
             elif message_type == "video":
                 sent = await message.bot.send_video(
                     recipient_telegram_id,
                     video=file_id,
-                    caption=f"💬 <b>Видео от {sender_name}</b>\n\n{text or ''}",
+                    caption=f"<b>{sender_name}</b>\n\n{text or ''}",
                     parse_mode="HTML"
                 )
             elif message_type == "document":
                 sent = await message.bot.send_document(
                     recipient_telegram_id,
                     document=file_id,
-                    caption=f"💬 <b>Документ от {sender_name}</b>\n\n{text or ''}",
+                    caption=f"<b>{sender_name}</b>\n\n{text or ''}",
                     parse_mode="HTML"
                 )