TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-11-26 05:04:28 +00:00
Code Issues Actions Projects Releases Wiki Activity

restore_pin 修复问题

Browse Source
This commit is contained in:
xtaodada 2023-10-21 16:12:31 +08:00
parent caffd78f8d
commit 44c42fd129
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
restore_pin/main.py
View File

@ -3,9 +3,10 @@ from typing import Dict, List
from pyrogram import filters from pyrogram import filters
from pyrogram.errors import ChatAdminRequired, UserAdminInvalid, FloodWait from pyrogram.errors import ChatAdminRequired, UserAdminInvalid, FloodWait
from pyrogram.raw.functions.channels import GetAdminLog from pyrogram.raw.functions.channels import GetAdminLog, GetMessages
from pyrogram.raw.types import ChannelAdminLogEventsFilter, ChannelAdminLogEventActionUpdatePinned from pyrogram.raw.types import ChannelAdminLogEventsFilter, ChannelAdminLogEventActionUpdatePinned, InputMessageID
from pyrogram.raw.types.channels import AdminLogResults from pyrogram.raw.types.channels import AdminLogResults
from pyrogram.raw.types.messages import Messages
from pagermaid.enums import Message from pagermaid.enums import Message
from pagermaid.listener import listener from pagermaid.listener import listener
@ -73,9 +74,20 @@ async def pin_one(message: Message, mid: int):
await pin_one(message, mid) await pin_one(message, mid)
async def get_unpin_messages(cid: int, ids: List[int]) -> List[int]:
ids = [InputMessageID(id=i) for i in ids]
r: Messages = await bot.invoke(GetMessages(channel=await bot.resolve_peer(cid), id=ids))
new_ids = []
for i in r.messages:
if not i.pinned:
new_ids.append(i.id)
return new_ids
async def try_restore_pin(message: Message, ids: List[int]): async def try_restore_pin(message: Message, ids: List[int]):
msgs = await bot.get_messages(message.chat.id, ids) new_ids = await get_unpin_messages(message.chat.id, ids)
new_ids = [i.id for i in msgs if not i.pinned_message] if not new_ids:
return await message.edit("没有需要恢复的置顶")
error = "" error = ""
for idx, i in enumerate(new_ids): for idx, i in enumerate(new_ids):
if (idx + 1) % 5 == 0: if (idx + 1) % 5 == 0: