shift 自动转发/备份频道消息

This commit is contained in:
xtaodada 2022-06-20 22:40:15 +08:00
parent d5414fb00e
commit bdcef64b7d
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659
2 changed files with 153 additions and 0 deletions

View File

@ -169,6 +169,16 @@
"supported": true, "supported": true,
"des-short": "vip 捐赠用户功能", "des-short": "vip 捐赠用户功能",
"des": "命令天气weather weather_pic weather_he\nDuckduckgoduckduckgo\n彩云翻译caiyun\n文本转语音tts_nan tts_nv tts_tw tts_ne tts_en。与 weather 插件冲突" "des": "命令天气weather weather_pic weather_he\nDuckduckgoduckduckgo\n彩云翻译caiyun\n文本转语音tts_nan tts_nv tts_tw tts_ne tts_en。与 weather 插件冲突"
},
{
"name": "shift",
"version": "1.031",
"section": "daily",
"maintainer": "xtaodada",
"size": "5.3 kb",
"supported": true,
"des-short": "自动转发/备份频道消息",
"des": "自动转发频道新消息或者从零开始备份频道。\n指令-shift\n查看帮助-help shift"
} }
] ]
} }

143
shift/main.py Normal file
View File

@ -0,0 +1,143 @@
""" PagerMaid module for channel help. """
from asyncio import sleep
from random import uniform
from pyrogram.enums.chat_type import ChatType
from pyrogram.errors.exceptions.flood_420 import FloodWait
from pagermaid import log
from pagermaid.single_utils import sqlite, Message
from pagermaid.utils import lang
from pagermaid.listener import listener
import contextlib
@listener(is_plugin=False, outgoing=True, command="shift",
description='开启转发频道新消息功能',
parameters="set <from channel> <to channel> 自动转发频道新消息(可以使用频道用户名或者 id\n"
"del <from channel> 删除转发\n"
"backup <from channel> <to channel> 备份频道(可以使用频道用户名或者 id")
async def shift_set(message: Message):
if not 1 < len(message.parameter) < 4:
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
return
if message.parameter[0] == "set":
if len(message.parameter) != 3:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
# 检查来源频道
try:
channel = await message.bot.get_chat(int(message.parameter[1]))
if channel.type != ChatType.CHANNEL:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.has_protected_content:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
except Exception:
try:
channel = await message.bot.get_chat(message.parameter[1])
if channel.type != ChatType.CHANNEL:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.has_protected_content:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
except Exception:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.id in [-1001441461877]:
return await message.edit('出错了呜呜呜 ~ 此对话位于白名单中。')
# 检查目标频道
try:
to = await message.bot.get_chat(int(message.parameter[2]))
except Exception:
try:
to = await message.bot.get_chat(message.parameter[2])
except Exception:
return await message.edit("出错了呜呜呜 ~ 无法识别的目标对话。")
if to.id in [-1001441461877]:
await message.edit('出错了呜呜呜 ~ 此对话位于白名单中。')
return
sqlite[f"shift.{channel.id}"] = to.id
await message.edit(f"已成功配置将对话 {channel.id} 的新消息转发到 {to.id}")
await log(f"已成功配置将对话 {channel.id} 的新消息转发到 {to.id}")
elif message.parameter[0] == "del":
if len(message.parameter) != 2:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
# 检查来源频道
try:
channel = await message.bot.get_chat(int(message.parameter[1]))
except Exception:
try:
channel = await message.bot.get_chat(message.parameter[1])
except Exception:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
try:
del sqlite[f"shift.{channel.id}"]
except Exception as e:
return await message.edit('emm...当前对话不存在于自动转发列表中。')
await message.edit(f"已成功关闭对话 {str(channel.id)} 的自动转发功能。")
await log(f"已成功关闭对话 {str(channel.id)} 的自动转发功能。")
elif message.parameter[0] == "backup":
if len(message.parameter) != 3:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
# 检查来源频道
try:
channel = await message.bot.get_chat(int(message.parameter[1]))
if channel.type != ChatType.CHANNEL:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.has_protected_content:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
except Exception:
try:
channel = await message.bot.get_chat(message.parameter[1])
if channel.type != ChatType.CHANNEL:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.has_protected_content:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
except Exception:
return await message.edit("出错了呜呜呜 ~ 无法识别的来源对话。")
if channel.id in [-1001441461877]:
return await message.edit('出错了呜呜呜 ~ 此对话位于白名单中。')
# 检查目标频道
try:
to = await message.bot.get_chat(int(message.parameter[2]))
except Exception:
try:
to = await message.bot.get_chat(message.parameter[2])
except Exception:
return await message.edit("出错了呜呜呜 ~ 无法识别的目标对话。")
if to.id in [-1001441461877]:
return await message.edit('出错了呜呜呜 ~ 此对话位于白名单中。')
# 开始遍历消息
await message.edit(f'开始备份频道 {channel.id}{to.id}')
async for msg in message.bot.search_messages(channel.id):
await sleep(uniform(0.5, 1.0))
await forward_msg(message, msg, to.id)
await message.edit(f'备份频道 {channel.id}{to.id} 已完成。')
else:
await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
return
@listener(is_plugin=True, incoming=True, ignore_edited=True)
async def shift_channel_message(message):
""" Event handler to auto forward channel messages. """
cid = sqlite.get(f"shift.{message.chat.id}", None)
if not cid:
return
if message.chat.id in [-1001441461877]:
return
if message.chat.has_protected_content:
del sqlite[f"shift.{message.chat.id}"]
return
with contextlib.suppress(Exception):
await message.forward(cid)
async def forward_msg(message, msg, cid):
try:
await msg.forward(msg, cid)
except FloodWait as e:
await message.edit(f'触发 Flood ,暂停 {e.value + uniform(0.5, 1.0)} 秒。')
try:
await sleep(e.value + uniform(0.5, 1.0))
except Exception as e:
print(f"Wait flood error: {e}")
return
await forward_msg(message, msg, cid)