TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-11-25 10:02:17 +00:00
Code Issues Actions Projects Releases Wiki Activity

shift 支持所有对话类型并且支持设置备份选项 (#177)

Browse Source
This commit is contained in:
canvex 2024-02-24 20:38:51 +08:00 committed by GitHub
parent 312aa42ded
commit 639d795e59
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 35 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
shift/main.py
View File

@ -3,7 +3,7 @@
import contextlib import contextlib
from asyncio import sleep from asyncio import sleep
from random import uniform from random import uniform
from typing import Any from typing import Any, List, Union, Set
from pyrogram.enums.chat_type import ChatType from pyrogram.enums.chat_type import ChatType
from pyrogram.errors.exceptions.flood_420 import FloodWait from pyrogram.errors.exceptions.flood_420 import FloodWait
@ -28,8 +28,8 @@ def try_cast_or_fallback(val: Any, t: type) -> Any:
def check_chat_available(chat: Chat): def check_chat_available(chat: Chat):
assert ( assert (
chat.type in [ChatType.CHANNEL, ChatType.GROUP] chat.type in [ChatType.CHANNEL, ChatType.GROUP, ChatType.SUPERGROUP, ChatType.BOT, ChatType.PRIVATE]
and not chat.has_protected_content and not chat.has_protected_content
) )
@ -37,11 +37,11 @@ def check_chat_available(chat: Chat):
command="shift", command="shift",
description="开启转发频道新消息功能", description="开启转发频道新消息功能",
parameters="set [from channel] [to channel] (silent) 自动转发频道新消息（可以使用频道用户名或者 id）\n" parameters="set [from channel] [to channel] (silent) 自动转发频道新消息（可以使用频道用户名或者 id）\n"
"del [from channel] 删除转发\n" "del [from channel] 删除转发\n"
"backup [from channel] [to channel] (silent) 备份频道（可以使用频道用户名或者 id）\n" "backup [from channel] [to channel] (silent) 备份频道（可以使用频道用户名或者 id）\n"
"list 顯示目前轉發的頻道\n\n" "list 顯示目前轉發的頻道\n\n"
"选项说明：\n" "选项说明：\n"
"silent: 禁用通知, text: 文字, all: 全部訊息都傳, photo: 圖片, document: 檔案, video: 影片", "silent: 禁用通知, text: 文字, all: 全部訊息都傳, photo: 圖片, document: 檔案, video: 影片",
) )
async def shift_set(client: Client, message: Message): async def shift_set(client: Client, message: Message):
if not message.parameter: if not message.parameter:
@ -129,12 +129,17 @@ async def shift_set(client: Client, message: Message):
return await message.edit("出错了呜呜呜 ~ 此对话位于白名单中。") return await message.edit("出错了呜呜呜 ~ 此对话位于白名单中。")
# 开始遍历消息 # 开始遍历消息
await message.edit(f"开始备份频道 {source.id} 到 {target.id} 。") await message.edit(f"开始备份频道 {source.id} 到 {target.id} 。")
# 如果有把get_chat_history方法merge進去就可以實現從舊訊息到新訊息,https://github.com/pyrogram/pyrogram/pull/1046
# async for msg in client.get_chat_history(source.id,reverse=True):
async for msg in client.search_messages(source.id): # type: ignore async for msg in client.search_messages(source.id): # type: ignore
await sleep(uniform(0.5, 1.0)) await sleep(uniform(0.5, 1.0))
await loosely_forward( await loosely_forward(
message, message,
msg, msg,
target.id, target.id,
options,
disable_notification="silent" in options, disable_notification="silent" in options,
) )
await message.edit(f"备份频道 {source.id} 到 {target.id} 已完成。") await message.edit(f"备份频道 {source.id} 到 {target.id} 已完成。")
@ -173,8 +178,7 @@ async def shift_channel_message(message: Message):
source = message.chat.id source = message.chat.id
# 找訊息類型video、document... # 找訊息類型video、document...
media_type = message.media.name.lower() if message.media else "text" media_type = message.media.value if message.media else "text"
target = d.get(f"shift.{source}") target = d.get(f"shift.{source}")
if not target: if not target:
return return
@ -199,18 +203,33 @@ async def shift_channel_message(message: Message):
async def loosely_forward( async def loosely_forward(
notifier: Message, notifier: Message,
message: Message, message: Message,
chat_id: int, chat_id: int,
disable_notification: bool = False, options: Union[List[str], Set[str]],
disable_notification: bool = False,
): ):
# 找訊息類型video、document...
media_type = message.media.value if message.media else "text"
try: try:
await message.forward(chat_id, disable_notification=disable_notification) if (not options) or "all" in options:
await message.forward(
chat_id,
disable_notification=disable_notification,
)
elif media_type in options:
await message.forward(
chat_id,
disable_notification=disable_notification,
)
else:
logs.debug("skip message type: %s", media_type)
# await message.forward(chat_id, disable_notification=disable_notification)
except FloodWait as ex: except FloodWait as ex:
min: int = ex.value # type: ignore min: int = ex.value # type: ignore
delay = min + uniform(0.5, 1.0) delay = min + uniform(0.5, 1.0)
await notifier.edit(f"触发 Flood ，暂停 {delay} 秒。") await notifier.edit(f"触发 Flood ，暂停 {delay} 秒。")
await sleep(delay) await sleep(delay)
await loosely_forward(notifier, message, chat_id, disable_notification) await loosely_forward(notifier, message, chat_id, options, disable_notification)
except Exception: except Exception:
pass # drop other errors pass # drop other errors