PagerMaid_Plugins_Pyro/alpha/auto_send_reactions/main.py
2024-02-04 15:56:06 +08:00

264 lines
11 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

"""
自动回复Emoji插件
Author: SuperManito
"""
from pagermaid import bot, log
from pagermaid.single_utils import sqlite
from pagermaid.enums import Client, Message
from pagermaid.utils import lang, edit_delete, pip_install
from pagermaid.listener import listener
pip_install("emoji")
import emoji
# 获取内容中的表情符号,并用|分割
def get_emoji(text):
if emojiArr := emoji.distinct_emoji_list(text):
delimiter = "|"
return delimiter.join(emojiArr)
else:
return False
@listener(
is_plugin=False,
outgoing=True,
command="auto_send_reactions",
description="\n自动回复Emoji插件",
parameters="`\n自动回复Emoji插件支持同时设置多个目标生效用户目前仅支持回复用户在群组中发送的消息默认在所有已加入的群组中生效\n"
"\n**设置插件状态**:"
"\n启用:`,auto_send_reactions enable`"
"\n停用:`,auto_send_reactions disable`"
"\n\n**设置目标生效用户**:"
"\n添加:`,auto_send_reactions set <用户id/用户名> <表情内容>`"
"\n移除:`,auto_send_reactions unset <用户id/用户名>`"
"\n支持直接回复消息以进行快速设置,可以省略用户标识参数"
"\n\n**设置生效群组黑名单**:"
"\n添加:`,auto_send_reactions block <群组id/群组用户名>`"
"\n移除:`,auto_send_reactions unblock <群组id/群组用户名>\n",
)
async def AutoSendReactions(client: Client, message: Message):
reply = message.reply_to_message
# 启用插件
if message.parameter[0] == "enable":
# for ID in ID_FROM_ARRAY :
# # 检查来源频道/群组
# try:
# channel = await bot.get_chat(ID)
# except Exception as e:
# errorMsg = f"第{e.__traceback__.tb_lineno}行:{e}"
# await message.edit(f"出错了呜呜呜 ~ 无法识别的来源对话。\n\n{errorMsg}")
# return
if not sqlite.get("AutoSendReactions.Enable"):
sqlite["AutoSendReactions.Enable"] = "yes"
# 返回消息
await edit_delete(message, "✅ **已启用自动回复表情插件**")
elif message.parameter[0] == "disable":
if sqlite.get("AutoSendReactions.Enable"):
del sqlite["AutoSendReactions.Enable"]
# 返回消息
await edit_delete(message, "❌ **已停用自动回复表情插件**")
elif message.parameter[0] in ["set", "unset"]:
## 设置插件
if message.parameter[0] == "set":
if not reply and (len(message.parameter) == 3):
target = message.parameter[1]
content = message.parameter[2]
try:
# 获取用户信息
user_info = await bot.get_users(target)
# 判断参数合法性
if not user_info:
return await message.edit("❌ **目标用户不存在或参数有误**")
if not content or not get_emoji(content):
return await message.edit("❌ **Emoji参数不能为空或不合法**")
user_name = (
f"{user_info.first_name} {user_info.last_name}"
if user_info.last_name
else user_info.first_name
)
# 设置或更新
hasSetted = sqlite.get(f"AutoSendReactions.{target}")
sqlite[f"AutoSendReactions.{target}"] = content
await edit_delete(
message,
f"✅ 已{'更新' if hasSetted else '添加'}对 __{user_name}__ 的自动回复Emoji设置**",
)
except Exception as e:
await message.edit(f"❌ **在设置中遇到了一些错误** > {e}")
await log(e) # 打印错误日志
elif reply and (len(message.parameter) == 2):
from_user = reply.from_user
target = from_user.id
content = message.parameter[1]
try:
# 判断参数合法性
if not content or not get_emoji(content):
return await message.edit("❌ **Emoji参数不能为空或不合法**")
user_name = (
f"{from_user.first_name} {from_user.last_name}"
if from_user.last_name
else from_user.first_name
)
# 设置或更新
hasSetted = sqlite.get(f"AutoSendReactions.{target}")
sqlite[f"AutoSendReactions.{target}"] = content
await edit_delete(
message,
f"✅ 已{'更新' if hasSetted else '添加'}对 __{user_name}__ 的自动回复Emoji设置",
)
except Exception as e:
await message.edit(f"❌ **在设置中遇到了一些错误** > {e}")
await log(e) # 打印错误日志
else:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
elif message.parameter[0] == "unset":
if not reply and (len(message.parameter) == 2):
target = message.parameter[1]
try:
# 获取用户信息
user_info = await bot.get_users(target)
# 判断参数合法性
if not user_info:
return await message.edit("❌ **目标用户不存在或参数有误**")
user_id = user_info.id
user_name = (
f"{user_info.first_name} {user_info.last_name}"
if user_info.last_name
else user_info.first_name
)
if hasSetted := sqlite.get(f"AutoSendReactions.{user_id}"):
del sqlite[f"AutoSendReactions.{user_id}"]
await edit_delete(
message, f"✅ 对 __{user_name}__ 的自动回复Emoji设置已删除"
)
else:
await edit_delete(
message, f"❌ 还没有对 __{user_name}__ 设置自动回复Emoji哦~"
)
except Exception as e:
await message.edit(f"❌ **在设置中遇到了一些错误** > {e}")
await log(e) # 打印错误日志
elif reply and (len(message.parameter) == 1):
from_user = reply.from_user
target = from_user.id
try:
user_name = (
f"{from_user.first_name} {from_user.last_name}"
if from_user.last_name
else from_user.first_name
)
if hasSetted := sqlite.get(f"AutoSendReactions.{target}"):
del sqlite[f"AutoSendReactions.{target}"]
await edit_delete(
message, f"✅ 已删除对 __{user_name}__ 的自动回复Emoji设置"
)
else:
await edit_delete(
message, f"❌ 还没有对 __{user_name}__ 设置自动回复Emoji哦~"
)
except Exception as e:
await message.edit(f"❌ **在设置中遇到了一些错误** > {e}")
await log(e) # 打印错误日志
else:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
else:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
elif (
(message.parameter[0] == "block")
or (message.parameter[0] == "unblock")
and (len(message.parameter) == 2)
):
group = message.parameter[1]
group_info = await bot.get_chat(chat_id=group)
if not group_info:
return await message.edit("❌ **目标群组不存在或参数有误**")
group_id = group_info.id
group_name = group_info.title
if message.parameter[0] == "block":
if hasBlocked := sqlite.get(f"AutoSendReactionsBlock.{group_id}"):
await edit_delete(
message, f"❌ 已经将 __{group_name}__ 加入自动回复Emoji黑名单群组了哦~"
)
else:
sqlite[f"AutoSendReactionsBlock.{group_id}"] = "yes"
await edit_delete(message, f"✅ 已将 __{group_name}__ 加入至自动回复Emoji黑名单群组")
elif message.parameter[0] == "unblock":
if hasBlocked := sqlite.get(f"AutoSendReactionsBlock.{group_id}"):
del sqlite[f"AutoSendReactionsBlock.{group_id}"]
await edit_delete(message, f"✅ 已将 __{group_name}__ 从自动回复Emoji黑名单群组中移除")
else:
await edit_delete(
message, f"❌ 还没有将 __{group_name}__ 加入进自动回复Emoji群组黑名单哦~"
)
else:
return await message.edit(f"{lang('error_prefix')}{lang('arg_error')}")
@listener(is_plugin=False, incoming=True, ignore_edited=True)
async def AutoSendReactions(message: Message):
from_user = ""
try:
# 判断是否启用了本插件
if not sqlite.get("AutoSendReactions.Enable"):
return
if "GROUP" not in str(message.chat.type):
return
# 判断是否在黑名单中
if sqlite.get(f"AutoSendReactionsBlock.{message.chat.id}"):
return
if not message.from_user:
# 过滤匿名管理员
return
from_user = message.from_user
if not sqlite.get(f"AutoSendReactions.{from_user.id}"):
return
# 判断群组是否启用了表情回应功能
group_info = await bot.get_chat(chat_id=message.chat.id)
if not group_info.available_reactions:
return
# 发送表情
emoji = sqlite.get(f"AutoSendReactions.{from_user.id}").split("|")[0]
user_name = (
f"{from_user.first_name} {from_user.last_name}"
if from_user.last_name
else from_user.first_name
)
try:
await bot.send_reaction(message.chat.id, message.id, emoji)
except Exception as e:
errorMsg = f"❌ 自动回复Emoji失败{user_name} {emoji}> {e}"
await log(errorMsg)
return False
# 打印日志
# text = message.text.markdown
# await log(f"AutoSendReactions 监控到来自 {user_name} 的消息:{str(text)}")
except Exception as e:
errorMsg = f"❌ 第{e.__traceback__.tb_lineno}行:{e}"
await log(errorMsg)
return False
## ⬆️ 不懂勿动 ⬆️