mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
236 lines
7.3 KiB
Python
236 lines
7.3 KiB
Python
from datetime import datetime
|
||
from typing import List
|
||
|
||
from nonebot.adapters import Bot
|
||
from nonebot.permission import SUPERUSER
|
||
from nonebot.plugin import PluginMetadata
|
||
from nonebot_plugin_alconna import Alconna, Args, Arparma, Match, Option, on_alconna
|
||
from nonebot_plugin_session import EventSession
|
||
|
||
from zhenxun.configs.config import BotConfig
|
||
from zhenxun.configs.utils import PluginExtraData, RegisterConfig
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.utils.enum import PluginType
|
||
from zhenxun.utils.image_utils import BuildImage
|
||
from zhenxun.utils.message import MessageUtils
|
||
|
||
from .data_source import set_user_punish, show_black_text_image
|
||
|
||
__plugin_meta__ = PluginMetadata(
|
||
name="敏感词检测",
|
||
description="请注意你的发言！",
|
||
usage="""
|
||
惩罚机制: 检测内容提示
|
||
设置惩罚 [uid] [id] [level]: 设置惩罚内容, 此id需要通过`记录名单 -u:uid`来获取
|
||
记录名单: 查看检测记录名单
|
||
记录名单:
|
||
-u [uid] 指定用户记录名单
|
||
-g [gid] 指定群组记录名单
|
||
-d [date] 指定日期
|
||
-dt ['=', '>', '<'] 大于小于等于指定日期
|
||
|
||
示例:
|
||
设置惩罚 123123123 0 1
|
||
记录名单 -u 123123123
|
||
记录名单 -g 333333
|
||
记录名单 -d 2022-11-11
|
||
记录名单 -d 2022-11-11 -dt >
|
||
""".strip(),
|
||
extra=PluginExtraData(
|
||
author="HibiKier",
|
||
version="0.1",
|
||
plugin_type=PluginType.SUPERUSER,
|
||
menu_type="其他",
|
||
configs=[
|
||
RegisterConfig(
|
||
key="CYCLE_DAYS",
|
||
value=30,
|
||
help="黑名单词汇记录周期",
|
||
default_value=30,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="TOLERATE_COUNT",
|
||
value=[5, 1, 1, 1, 1],
|
||
help="各个级别惩罚的容忍次数, 依次为: 1, 2, 3, 4, 5",
|
||
default_value=[5, 1, 1, 1, 1],
|
||
type=List[int],
|
||
),
|
||
RegisterConfig(
|
||
key="AUTO_PUNISH",
|
||
value=True,
|
||
help="是否启动自动惩罚机制",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="BAN_4_DURATION",
|
||
value=360,
|
||
help="Ban时长（分钟），四级惩罚，可以为指定数字或指定列表区间(随机)，例如 [30, 360]",
|
||
default_value=360,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="BAN_3_DURATION",
|
||
value=7,
|
||
help="Ban时长（天），三级惩罚，可以为指定数字或指定列表区间(随机)，例如 [7, 30]",
|
||
default_value=7,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="WARNING_RESULT",
|
||
value=f"请注意对{BotConfig.self_nickname}的发言内容",
|
||
help="口头警告内容",
|
||
default_value=None,
|
||
),
|
||
RegisterConfig(
|
||
key="AUTO_ADD_PUNISH_LEVEL",
|
||
value=360,
|
||
help="自动提级机制，当周期内处罚次数大于某一特定值就提升惩罚等级",
|
||
default_value=360,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="ADD_PUNISH_LEVEL_TO_COUNT",
|
||
value=3,
|
||
help="在CYCLE_DAYS周期内触发指定惩罚次数后提升惩罚等级",
|
||
default_value=3,
|
||
type=int,
|
||
),
|
||
RegisterConfig(
|
||
key="ALAPI_CHECK_FLAG",
|
||
value=False,
|
||
help="当未检测到已收录的敏感词时，开启ALAPI文本检测并将疑似文本发送给超级用户",
|
||
default_value=False,
|
||
type=bool,
|
||
),
|
||
RegisterConfig(
|
||
key="CONTAIN_BLACK_STOP_PROPAGATION",
|
||
value=True,
|
||
help="当文本包含任意敏感词时，停止向下级插件传递，即不触发ai",
|
||
default_value=True,
|
||
type=bool,
|
||
),
|
||
],
|
||
).dict(),
|
||
)
|
||
|
||
|
||
_punish_matcher = on_alconna(
|
||
Alconna("设置惩罚", Args["uid", str]["id", int]["punish_level", int]),
|
||
priority=1,
|
||
permission=SUPERUSER,
|
||
block=True,
|
||
)
|
||
|
||
|
||
_show_matcher = on_alconna(
|
||
Alconna(
|
||
"记录名单",
|
||
Option("-u|--uid", Args["uid", str]),
|
||
Option("-g|--group", Args["gid", str]),
|
||
Option("-d|--date", Args["date", str]),
|
||
Option("-dt|--type", Args["date_type", ["=", ">", "<"]], default="="),
|
||
),
|
||
priority=1,
|
||
permission=SUPERUSER,
|
||
block=True,
|
||
)
|
||
|
||
_show_punish_matcher = on_alconna(
|
||
Alconna("惩罚机制"), aliases={"敏感词检测"}, priority=1, block=True
|
||
)
|
||
|
||
|
||
@_show_matcher.handle()
|
||
async def _(
|
||
bot: Bot, uid: Match[str], gid: Match[str], date: Match[str], date_type: Match[str]
|
||
):
|
||
user_id = None
|
||
group_id = None
|
||
date_ = None
|
||
date_str = None
|
||
date_type_ = "="
|
||
if uid.available:
|
||
user_id = uid.result
|
||
if gid.available:
|
||
group_id = gid.result
|
||
if date.available:
|
||
date_str = date.result
|
||
if date_type.available:
|
||
date_type_ = date_type.result
|
||
if date_str:
|
||
try:
|
||
date_ = datetime.strptime(date_str, "%Y-%m-%d")
|
||
except ValueError:
|
||
await MessageUtils.build_message("日期格式错误，需要：年-月-日").finish()
|
||
result = await show_black_text_image(
|
||
user_id,
|
||
group_id,
|
||
date_,
|
||
date_type_,
|
||
)
|
||
await MessageUtils.build_message(result).send()
|
||
|
||
|
||
@_show_punish_matcher.handle()
|
||
async def _():
|
||
text = f"""
|
||
** 惩罚机制 **
|
||
|
||
惩罚前包含容忍机制，在指定周期内会容忍偶尔少次数的敏感词只会进行警告提醒
|
||
|
||
多次触发同级惩罚会使惩罚等级提高，即惩罚自动提级机制
|
||
|
||
目前公开的惩罚等级：
|
||
|
||
1级：永久ban
|
||
|
||
2级：删除好友
|
||
|
||
3级：ban指定/随机天数
|
||
|
||
4级：ban指定/随机时长
|
||
|
||
5级：警告
|
||
|
||
备注：
|
||
|
||
该功能为测试阶段，如果你有被误封情况，请联系管理员，会从数据库中提取出你的数据进行审核后判断
|
||
|
||
目前该功能暂不完善，部分情况会由管理员鉴定，请注意对真寻的发言
|
||
|
||
关于敏感词：
|
||
|
||
记住不要骂{BotConfig.self_nickname}就对了！
|
||
""".strip()
|
||
max_width = 0
|
||
for m in text.split("\n"):
|
||
max_width = len(m) * 20 if len(m) * 20 > max_width else max_width
|
||
max_height = len(text.split("\n")) * 24
|
||
A = BuildImage(
|
||
max_width, max_height, font="CJGaoDeGuo.otf", font_size=24, color="#E3DBD1"
|
||
)
|
||
await A.text((10, 10), text)
|
||
await MessageUtils.build_message(A).send()
|
||
|
||
|
||
@_punish_matcher.handle()
|
||
async def _(
|
||
bot: Bot,
|
||
session: EventSession,
|
||
arparma: Arparma,
|
||
uid: str,
|
||
id: int,
|
||
punish_level: int,
|
||
):
|
||
result = await set_user_punish(
|
||
bot, uid, session.id2 or session.id3, id, punish_level
|
||
)
|
||
await MessageUtils.build_message(result).send(reply_to=True)
|
||
logger.info(
|
||
f"设置惩罚 uid:{uid} id_:{id} punish_level:{punish_level} --> {result}",
|
||
arparma.header_result,
|
||
session=session,
|
||
)
|