from datetime import datetime, timedelta

from pyrogram.errors import FloodWait, ChatAdminRequired, UserAdminInvalid

from pagermaid.listener import listener
from pagermaid.enums import Client, Message


@listener(command="getdel",
          groups_only=True,
          need_admin=True,
          parameters="清理",
          description="获取当前群组的死号数。")
async def get_del(client: Client, message: Message):
    """ PagerMaid get_del. """
    need_kick = message.arguments
    member_count = 0
    try:
        await message.edit('遍历成员中。。。')
        if need_kick:
            user = await client.get_chat_member(message.chat.id, (await client.get_me()).id)
            need_kick = bool(user.privileges and user.privileges.can_restrict_members)
        async for member in client.get_chat_members(message.chat.id):
            if member.user.is_deleted:
                member_count += 1
                if need_kick:
                    try:
                        await client.ban_chat_member(
                            message.chat.id,
                            member.user.id,
                            datetime.now() + timedelta(minutes=5))
                    except FloodWait:
                        return await message.edit('处理失败，您已受到 TG 服务器限制。')
                    except UserAdminInvalid:
                        pass
        if need_kick:
            await message.edit(f'此群组的死号数：`{member_count}`，并且已经清理完毕。')
        else:
            await message.edit(f'此群组的死号数：`{member_count}`。')
    except ChatAdminRequired:
        await message.edit("你好像并不拥有封禁用户权限。")