From 7b4b3fb4c7e495dbbecd9f3c3a0b3b1c1a55b7bd Mon Sep 17 00:00:00 2001 From: canvex <44216326+canvex@users.noreply.github.com> Date: Sat, 27 Jan 2024 23:31:10 +0800 Subject: [PATCH] =?UTF-8?q?shift=20=E6=94=AF=E6=8C=81=E8=BF=87=E6=BB=A4?= =?UTF-8?q?=E6=B6=88=E6=81=AF=E7=B1=BB=E5=9E=8B=EF=BC=8C=E5=B1=95=E7=A4=BA?= =?UTF-8?q?=E6=89=80=E6=9C=89=E8=BD=AC=E5=8F=91=E4=BB=BB=E5=8A=A1?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: omg-xtao <100690902+omg-xtao@users.noreply.github.com> --- shift/DES.md | 11 +++++++- shift/main.py | 69 ++++++++++++++++++++++++++++++++++++--------------- 2 files changed, 59 insertions(+), 21 deletions(-) diff --git a/shift/DES.md b/shift/DES.md index 9a2a100..3f5e9fe 100644 --- a/shift/DES.md +++ b/shift/DES.md @@ -10,7 +10,7 @@ 需要加入 from_channel_id 和 to_channel_id。 -`, shift set [from channel_id] [to channel_id] (silent)` +`, shift set [from channel_id] [to channel_id] (silent, all, text, photo, document, video)` ### 取消转发 @@ -20,6 +20,15 @@ `, shift backup [from channel_id] [to channel_id] (silent)` +### 顯示要轉發的頻道 + +`, shift list` + ## 选项解释 - `silent`: 禁用通知 +- `all`: 全部都轉發(預設會有all) +- `text`: 轉發文字 +- `photo`: 轉發圖片 +- `document`: 轉發檔案 +- `video`: 轉發影片 diff --git a/shift/main.py b/shift/main.py index 4fd1a50..9d51108 100644 --- a/shift/main.py +++ b/shift/main.py @@ -5,17 +5,18 @@ from asyncio import sleep from random import uniform from typing import Any -from pagermaid import log -from pagermaid.enums import Client, Message -from pagermaid.listener import listener -from pagermaid.single_utils import sqlite -from pagermaid.utils import lang from pyrogram.enums.chat_type import ChatType from pyrogram.errors.exceptions.flood_420 import FloodWait from pyrogram.types import Chat +from pagermaid import log, logs +from pagermaid.enums import Client, Message +from pagermaid.listener import listener +from pagermaid.single_utils import sqlite +from pagermaid.utils import lang + WHITELIST = [-1001441461877] -AVAILABLE_OPTIONS = {"silent"} +AVAILABLE_OPTIONS = {"silent", "text", "all", "photo", "document", "video"} def try_cast_or_fallback(val: Any, t: type) -> Any: @@ -26,17 +27,18 @@ def try_cast_or_fallback(val: Any, t: type) -> Any: def check_chat_available(chat: Chat): - assert chat.type == ChatType.CHANNEL and not chat.has_protected_content + assert (chat.type in [ChatType.CHANNEL, ChatType.GROUP] and not chat.has_protected_content) @listener( command="shift", description="开启转发频道新消息功能", parameters="set [from channel] [to channel] (silent) 自动转发频道新消息（可以使用频道用户名或者 id）\n" - "del [from channel] 删除转发\n" - "backup [from channel] [to channel] (silent) 备份频道（可以使用频道用户名或者 id）\n\n" - "选项说明：\n" - "silent: 禁用通知", + "del [from channel] 删除转发\n" + "backup [from channel] [to channel] (silent) 备份频道（可以使用频道用户名或者 id）\n" + "list 顯示目前轉發的頻道\n\n" + "选项说明：\n" + "silent: 禁用通知, none: 文字, all: 全部訊息都傳, photo: 圖片, document: 檔案, video: 影片", ) async def shift_set(client: Client, message: Message): if not message.parameter: @@ -72,7 +74,7 @@ async def shift_set(client: Client, message: Message): return sqlite[f"shift.{source.id}"] = target.id sqlite[f"shift.{source.id}.options"] = ( - message.parameter[3:] if len(message.parameter) > 3 else [] + message.parameter[3:] if len(message.parameter) > 3 else ["all"] ) await message.edit(f"已成功配置将对话 {source.id} 的新消息转发到 {target.id} 。") await log(f"已成功配置将对话 {source.id} 的新消息转发到 {target.id} 。") @@ -133,16 +135,35 @@ async def shift_set(client: Client, message: Message): disable_notification="silent" in options, ) await message.edit(f"备份频道 {source.id} 到 {target.id} 已完成。") + # 列出要轉存的頻道 + elif message.parameter[0] == "list": + from_ids = list(filter(lambda x: (x.startswith("shift.") and (not x.endswith("options"))), list(sqlite.keys()))) + if not from_ids: + return await message.edit("沒有要轉存的頻道") + output = "總共有 %d 個頻道要轉存\n\n" % len(from_ids) + for from_id in from_ids: + to_id = sqlite[from_id] + output += "%s -> %s\n" % (format_channel_id(from_id[6:]), format_channel_id(to_id)) + await message.edit(output) else: await message.edit(f"{lang('error_prefix')}{lang('arg_error')}") return +def format_channel_id(channel_id: str): + short_channel_id = str(channel_id)[4:] + return f"[{channel_id}](https://t.me/c/{short_channel_id})" + + @listener(is_plugin=True, incoming=True, ignore_edited=True) async def shift_channel_message(message: Message): """Event handler to auto forward channel messages.""" d = dict(sqlite) source = message.chat.id + + # 找訊息類型video、document... + media_type = message.media.name.lower() if message.media else "text" + target = d.get(f"shift.{source}") if not target: return @@ -152,17 +173,25 @@ async def shift_channel_message(message: Message): options = d.get(f"shift.{source}.options") or [] with contextlib.suppress(Exception): - await message.forward( - target, - disable_notification="silent" in options, - ) + if (not options) or "all" in options: + await message.forward( + target, + disable_notification="silent" in options, + ) + elif media_type in options: + await message.forward( + target, + disable_notification="silent" in options, + ) + else: + logs.debug("skip message type: %s", media_type) async def loosely_forward( - notifier: Message, - message: Message, - chat_id: int, - disable_notification: bool = False, + notifier: Message, + message: Message, + chat_id: int, + disable_notification: bool = False, ): try: await message.forward(chat_id, disable_notification=disable_notification)