PagerMaid_Plugins_Pyro/dc/main.py

52 lines
2.3 KiB
Python
Raw Normal View History

# -*- coding: UTF-8 -*-
'''
@File main.py
@Author 汐洛 @guimc233
@Date 2022/6/23 2:49
'''
from pyrogram import Client
from pagermaid.listener import listener
from pagermaid.utils import Message
from pyrogram.enums import ParseMode
@listener(command="dc",
description="查看本群dc分布, 查看你回复的人在哪个dc")
async def dc(bot: Client, context: Message):
context = await context.edit("Please wait...")
if context.reply_to_message:
user = (
context.reply_to_message.from_user
or context.reply_to_message.sender_chat
)
if not user:
return await context.edit("出错啦!")
try:
return await context.edit(f"您所在的位置: DC{user.dc_id}")
except:
return await context.edit("无法查询! 您是否设置了头像呢?我是否可以看到你的头像呢?")
count = await bot.get_chat_members_count(context.chat.id)
if count >= 10000 and context.arguments != "force":
return await context.edit("太...太多人了... 我会...会...会坏掉的...\n\n如果您执意要运行的的话,您可以使用指令 ,dc force")
users = bots_deleted = 0
dc_ids = {"1": 0, "2": 0, "3": 0, "4": 0, "5": 0, "failed": 0}
async for m in bot.get_chat_members(context.chat.id, limit=9999):
if not m.user.is_bot and not m.user.is_deleted:
users += 1
try:
dc_ids[str(m.user.dc_id)] += 1
except:
dc_ids["failed"] += 1
else:
bots_deleted += 1
await context.edit(f"""DC:
> DC1用户: **{dc_ids["1"]}** 分遗产占比: **{round((dc_ids["1"]/users)*100, 2)}%**
> DC2用户: **{dc_ids["2"]}** 分遗产占比: **{round((dc_ids["2"]/users)*100, 2)}%**
> DC3用户: **{dc_ids["3"]}** 分遗产占比: **{round((dc_ids["3"]/users)*100, 2)}%**
> DC4用户: **{dc_ids["4"]}** 分遗产占比: **{round((dc_ids["4"]/users)*100, 2)}%**
> DC5用户: **{dc_ids["5"]}** 分遗产占比: **{round((dc_ids["5"]/users)*100, 2)}%**
> 无法获取在哪个DC的用户: **{dc_ids["failed"]}**
> 已自动过滤掉 **{bots_deleted}** Bot / 死号
{'***请注意: 由于tg限制 我们只能遍历前10k人 此次获得到的数据并不完整***' if count >= 10000 else ''}""", parse_mode = ParseMode.MARKDOWN)