first

handlers/group_handlers.py

@@ -0,0 +1,185 @@
+"""Handlers for group events."""
+import telebot
+from typing import Optional
+from telebot import types
+from sqlalchemy.orm import Session
+from database import get_db_session, User, Chat, UserChat
+
+
+def register_group_handlers(bot: telebot.TeleBot) -> None:
+    """Register all group event handlers."""
+    
+    @bot.message_handler(content_types=['new_chat_members'])
+    def handle_new_member(message: telebot.types.Message) -> None:
+        """Handle bot being added to a chat."""
+        # Check if bot was added
+        bot_me = bot.get_me()
+        if not bot_me:
+            return
+        
+        for member in message.new_chat_members:
+            if member.id == bot_me.id:
+                chat_id = message.chat.id
+                chat_title: str = message.chat.title or "Unknown"
+                
+                db = get_db_session()
+                try:
+                    # Check if chat exists
+                    chat = db.query(Chat).filter(Chat.chat_id == chat_id).first()
+                    if not chat:
+                        chat = Chat(chat_id=chat_id, chat_title=chat_title, bot_is_admin=False)
+                        db.add(chat)
+                        db.commit()
+                    
+                    # Check if bot is admin
+                    try:
+                        bot_me = bot.get_me()
+                        if not bot_me:
+                            return
+                        bot_member = bot.get_chat_member(chat_id, bot_me.id)
+                        is_admin = bot_member.status in ['administrator', 'creator']
+                        chat.bot_is_admin = is_admin
+                        db.commit()
+                        
+                        if not is_admin:
+                            # Request admin rights
+                            bot.reply_to(
+                                message,
+                                "Привет! Мне нужны права администратора, чтобы видеть список участников чата. "
+                                "Пожалуйста, выдайте мне права администратора."
+                            )
+                        else:
+                            # Sync users and show statistics
+                            sync_chat_users(bot, db, chat_id)
+                            show_statistics(bot, message.chat)
+                    except Exception:
+                        # Bot might not have permission to check
+                        chat.bot_is_admin = False
+                        db.commit()
+                        bot.reply_to(
+                            message,
+                            "Привет! Мне нужны права администратора, чтобы видеть список участников чата. "
+                            "Пожалуйста, выдайте мне права администратора."
+                        )
+                finally:
+                    db.close()
+                break
+    
+    @bot.chat_member_handler()
+    def handle_chat_member_update(message: telebot.types.ChatMemberUpdated) -> None:
+        """Handle chat member updates (including bot becoming admin)."""
+        bot_me = bot.get_me()
+        if not bot_me:
+            return
+        
+        if message.new_chat_member.user.id == bot_me.id:
+            # Bot's status changed
+            chat_id = message.chat.id
+            chat_title: str = message.chat.title or "Unknown"
+            
+            db = get_db_session()
+            try:
+                chat = db.query(Chat).filter(Chat.chat_id == chat_id).first()
+                if not chat:
+                    chat = Chat(chat_id=chat_id, chat_title=chat_title, bot_is_admin=False)
+                    db.add(chat)
+                
+                # Check if bot became admin
+                is_admin = message.new_chat_member.status in ['administrator', 'creator']
+                was_admin = chat.bot_is_admin if chat else False
+                if chat:
+                    chat.bot_is_admin = is_admin
+                    chat.chat_title = chat_title
+                    db.commit()
+                
+                if is_admin and not was_admin:
+                    # Bot just became admin - sync users and show statistics
+                    sync_chat_users(bot, db, chat_id)
+                    show_statistics(bot, message.chat)
+            finally:
+                db.close()
+
+
+def sync_chat_users(bot: telebot.TeleBot, db: Session, chat_id: int) -> None:
+    """Sync users from chat with database."""
+    try:
+        # Get all chat members (this requires admin rights)
+        # Note: This is a simplified approach - in production you might want to
+        # use get_chat_administrators or iterate through members differently
+        # For now, we'll sync users as they interact with the bot
+        
+        # Get all users who already shared birthday
+        existing_users = db.query(User).all()
+        
+        # Try to check if they're in this chat and add relationships
+        for user in existing_users:
+            try:
+                member = bot.get_chat_member(chat_id, user.user_id)
+                if member.status not in ['left', 'kicked']:
+                    # User is in chat - ensure relationship exists
+                    user_chat = db.query(UserChat).filter(
+                        UserChat.user_id == user.user_id,
+                        UserChat.chat_id == chat_id
+                    ).first()
+                    
+                    if not user_chat:
+                        user_chat = UserChat(user_id=user.user_id, chat_id=chat_id)
+                        db.add(user_chat)
+            except Exception:
+                # User might have blocked bot or is not in chat
+                pass
+        
+        db.commit()
+    except Exception:
+        # If sync fails, continue anyway
+        pass
+
+
+def show_statistics(bot: telebot.TeleBot, chat: telebot.types.Chat) -> None:
+    """Show statistics about users who shared their birthday."""
+    db = get_db_session()
+    try:
+        chat_id = chat.id
+        
+        # Get all chat members
+        try:
+            members_count = bot.get_chat_member_count(chat_id)
+        except Exception:
+            members_count = 0
+        
+        # Get users from this chat who shared birthday
+        users_with_birthday = db.query(User).join(UserChat).filter(
+            UserChat.chat_id == chat_id
+        ).distinct().count()
+        
+        # Create message
+        if members_count > 0:
+            percentage = (users_with_birthday / members_count) * 100
+            message_text = (
+                f"Отлично! Теперь я админ этого чата.\n\n"
+                f"📊 Статистика:\n"
+                f"• Поделились днем рождения: {users_with_birthday} из {members_count} участников\n"
+                f"• Процент: {percentage:.1f}%"
+            )
+        else:
+            message_text = (
+                f"Отлично! Теперь я админ этого чата.\n\n"
+                f"📊 Поделились днем рождения: {users_with_birthday} участников"
+            )
+        
+        # Create inline keyboard with button to start private chat
+        bot_me = bot.get_me()
+        if not bot_me or not bot_me.username:
+            bot.send_message(chat_id, message_text)
+            return
+        
+        keyboard = types.InlineKeyboardMarkup()
+        start_button = types.InlineKeyboardButton(
+            text="Поделиться днем рождения",
+            url=f"https://t.me/{bot_me.username}?start=share_birthday"
+        )
+        keyboard.add(start_button)
+        
+        bot.send_message(chat_id, message_text, reply_markup=keyboard)
+    finally:
+        db.close()