mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-09-28 07:25:06 +00:00
102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
# yapppbc
|
|
# Hello 2023
|
|
|
|
from os.path import getmtime
|
|
from datetime import datetime
|
|
from time import time
|
|
|
|
from pyrogram.enums import ChatType
|
|
from pyrogram.raw.functions.contacts import GetBlocked
|
|
from pyrogram.raw.types.contacts import BlockedSlice
|
|
|
|
from pagermaid.listener import listener
|
|
from pagermaid.enums import Message, AsyncClient, Client
|
|
from pagermaid.modules import __list_plugins
|
|
from pagermaid.services import bot, sqlite
|
|
|
|
|
|
async def get_chat_count():
|
|
private, group, bots, channel = 0, 0, 0, 0
|
|
for dialog in await bot.get_dialogs_list():
|
|
if dialog.chat.type == ChatType.BOT:
|
|
bots += 1
|
|
if dialog.chat.type == ChatType.CHANNEL:
|
|
channel += 1
|
|
if dialog.chat.type == ChatType.GROUP or ChatType.SUPERGROUP:
|
|
group += 1
|
|
if dialog.chat.type == ChatType.PRIVATE:
|
|
private += 1
|
|
return private, group, bots, channel
|
|
|
|
|
|
async def get_blocked_count():
|
|
blocked = await bot.invoke(GetBlocked(offset=0, limit=1))
|
|
if isinstance(blocked, BlockedSlice):
|
|
return blocked.count
|
|
elif blocked.users:
|
|
return len(blocked.users)
|
|
else:
|
|
return 0
|
|
|
|
|
|
async def get_hitokoto(request: AsyncClient):
|
|
try:
|
|
htk = (await request.get("https://v1.hitokoto.cn/?charset=utf-8")).json()
|
|
text = f"\"{htk['hitokoto']}\" —— "
|
|
if htk["from_who"]:
|
|
text += f"{htk['from_who']}"
|
|
if htk["from"]:
|
|
text += f"「{htk['from']}」"
|
|
except Exception:
|
|
text = '"用代码表达言语的魅力，用代码书写山河的壮丽。" —— 一言「一言开发者中心」'
|
|
return text
|
|
|
|
|
|
async def get_year() -> str:
|
|
now = datetime.now()
|
|
year = now.year
|
|
if now.month == 1:
|
|
year -= 1
|
|
return str(year)
|
|
|
|
|
|
@listener(command="annualreport", description="TG年度报告")
|
|
async def annualreport(client: Client, message: Message, request: AsyncClient):
|
|
await message.edit("加载中请稍候。。。")
|
|
year = await get_year()
|
|
private, group, bots, channel = await get_chat_count()
|
|
days = int((time() - getmtime("LICENSE")) / 86400)
|
|
plg = sorted(__list_plugins())
|
|
blocked = await get_blocked_count()
|
|
blt = "你的账户真的很干净" if blocked < 20 else "愿明年的spam少一些"
|
|
u = await client.get_me()
|
|
if u.username:
|
|
user = f"@{u.username}"
|
|
elif u.last_name:
|
|
user = f"{u.first_name} {u.last_name}"
|
|
else:
|
|
user = u.first_name
|
|
pre = "你已成为TG大会员用户,不知遗产分到没有" if u.is_premium else ""
|
|
pmct = ""
|
|
if pmc := sqlite.get("pmcaptcha", {}):
|
|
pmcu = "" if "pmcaptcha" in plg else "不过此插件已经被卸载了,是spam变少了吗?\n"
|
|
pmct = (
|
|
f'pmcaptcha 已帮助你拦截了 {pmc.get("banned", 0)} 次私聊\n你的清净由 pagermaid 守护\n{pmcu}'
|
|
)
|
|
htks = await get_hitokoto(request)
|
|
msg = f"""{user} 的年度报告
|
|
{year} 一路上,你充实而满足
|
|
此Pagermaid-Pyro实例陪伴了你的TG {days} 天,安装了 {len(plg)} 个插件
|
|
为你的TG使用体验增光添彩
|
|
|
|
每次遇见,皆需万般幸运
|
|
你邂逅了 {channel} 个频道, {group} 个群组, 遇见了 {private} 个有趣的灵魂, 使用了 {bots} 个机器人
|
|
愿你的生活每天都像庆典一样开心
|
|
|
|
你的黑名单里有 {blocked} 人
|
|
{blt}
|
|
{pmct}
|
|
{pre}
|
|
{htks}"""
|
|
await message.edit(msg)
|