mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-10-06 20:29:59 +08:00
refactor code
This commit is contained in:
hibiki
+67
-56
@@ -6,28 +6,34 @@ from nonebot.typing import T_State
|
||||
import time
|
||||
from nonebot.adapters.cqhttp.exception import ActionFailed
|
||||
from configs.path_config import DATA_PATH, IMAGE_PATH
|
||||
from utils.img_utils import get_img_hash
|
||||
from utils.image_utils import get_img_hash
|
||||
from services.log import logger
|
||||
import aiohttp
|
||||
import aiofiles
|
||||
from configs.config import MUTE_DEFAULT_COUNT, MUTE_DEFAULT_TIME, MUTE_DEFAULT_DURATION
|
||||
|
||||
try:
|
||||
import ujson as json
|
||||
except ModuleNotFoundError:
|
||||
import json
|
||||
|
||||
__plugin_name__ = '刷屏禁言'
|
||||
__plugin_name__ = "刷屏禁言"
|
||||
|
||||
__plugin_usage__ = '刷屏禁言检测'
|
||||
__plugin_usage__ = "刷屏禁言检测"
|
||||
|
||||
|
||||
mute = on_message(priority=1, block=False)
|
||||
mute_setting = on_command('mute_setting', aliases={'设置检测时间', '设置检测次数', '设置禁言时长', '刷屏检测设置'}, permission=GROUP, block=True)
|
||||
mute_setting = on_command(
|
||||
"mute_setting",
|
||||
aliases={"设置检测时间", "设置检测次数", "设置禁言时长", "刷屏检测设置"},
|
||||
permission=GROUP,
|
||||
block=True,
|
||||
)
|
||||
|
||||
|
||||
def get_data():
|
||||
try:
|
||||
with open(DATA_PATH + 'group_mute_data.json', 'r', encoding='utf8') as f:
|
||||
with open(DATA_PATH + "group_mute_data.json", "r", encoding="utf8") as f:
|
||||
data = json.load(f)
|
||||
except (ValueError, FileNotFoundError):
|
||||
data = {}
|
||||
@@ -36,16 +42,18 @@ def get_data():
|
||||
|
||||
def save_data():
|
||||
global mute_data
|
||||
with open(DATA_PATH + 'group_mute_data.json', 'w', encoding='utf8') as f:
|
||||
with open(DATA_PATH + "group_mute_data.json", "w", encoding="utf8") as f:
|
||||
json.dump(mute_data, f, indent=4)
|
||||
|
||||
|
||||
async def download_img_and_hash(url, group_id):
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get(url, proxy=get_local_proxy(), timeout=10) as response:
|
||||
async with aiofiles.open(IMAGE_PATH + f'temp/mute_{group_id}_img.jpg', 'wb') as f:
|
||||
async with aiofiles.open(
|
||||
IMAGE_PATH + f"temp/mute_{group_id}_img.jpg", "wb"
|
||||
) as f:
|
||||
await f.write(await response.read())
|
||||
return str(get_img_hash(IMAGE_PATH + f'temp/mute_{group_id}_img.jpg'))
|
||||
return str(get_img_hash(IMAGE_PATH + f"temp/mute_{group_id}_img.jpg"))
|
||||
|
||||
|
||||
mute_dict = {}
|
||||
@@ -57,44 +65,48 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
group_id = str(event.group_id)
|
||||
msg = get_message_text(event.json())
|
||||
imgs = get_message_imgs(event.json())
|
||||
img_hash = ''
|
||||
img_hash = ""
|
||||
for img in imgs:
|
||||
img_hash += await download_img_and_hash(img, event.group_id)
|
||||
msg += img_hash
|
||||
if not mute_data.get(group_id):
|
||||
mute_data[group_id] = {
|
||||
'count': MUTE_DEFAULT_COUNT,
|
||||
'time': MUTE_DEFAULT_TIME,
|
||||
'duration': MUTE_DEFAULT_DURATION
|
||||
"count": MUTE_DEFAULT_COUNT,
|
||||
"time": MUTE_DEFAULT_TIME,
|
||||
"duration": MUTE_DEFAULT_DURATION,
|
||||
}
|
||||
if not mute_dict.get(event.user_id):
|
||||
mute_dict[event.user_id] = {
|
||||
'time': time.time(),
|
||||
'count': 1,
|
||||
'msg': msg
|
||||
}
|
||||
mute_dict[event.user_id] = {"time": time.time(), "count": 1, "msg": msg}
|
||||
else:
|
||||
if state["_prefix"]["raw_command"] or not msg:
|
||||
return
|
||||
if msg and msg.find(mute_dict[event.user_id]['msg']) != -1:
|
||||
mute_dict[event.user_id]['count'] += 1
|
||||
if msg and msg.find(mute_dict[event.user_id]["msg"]) != -1:
|
||||
mute_dict[event.user_id]["count"] += 1
|
||||
else:
|
||||
mute_dict[event.user_id]['time'] = time.time()
|
||||
mute_dict[event.user_id]['count'] = 1
|
||||
mute_dict[event.user_id]['msg'] = msg
|
||||
if time.time() - mute_dict[event.user_id]['time'] > mute_data[group_id]['time']:
|
||||
mute_dict[event.user_id]['time'] = time.time()
|
||||
mute_dict[event.user_id]['count'] = 1
|
||||
if mute_dict[event.user_id]['count'] > mute_data[group_id]['count'] and\
|
||||
time.time() - mute_dict[event.user_id]['time'] < mute_data[group_id]['time']:
|
||||
mute_dict[event.user_id]["time"] = time.time()
|
||||
mute_dict[event.user_id]["count"] = 1
|
||||
mute_dict[event.user_id]["msg"] = msg
|
||||
if time.time() - mute_dict[event.user_id]["time"] > mute_data[group_id]["time"]:
|
||||
mute_dict[event.user_id]["time"] = time.time()
|
||||
mute_dict[event.user_id]["count"] = 1
|
||||
if (
|
||||
mute_dict[event.user_id]["count"] > mute_data[group_id]["count"]
|
||||
and time.time() - mute_dict[event.user_id]["time"]
|
||||
< mute_data[group_id]["time"]
|
||||
):
|
||||
try:
|
||||
if mute_data[group_id]['duration'] != 0:
|
||||
await bot.set_group_ban(group_id=event.group_id, user_id=event.user_id,
|
||||
duration=mute_data[group_id]['duration'])
|
||||
await mute.send('检测到恶意刷屏，真寻要把你关进小黑屋！', at_sender=True)
|
||||
mute_dict[event.user_id]['count'] = 0
|
||||
logger.info(f'USER {event.user_id} GROUP {event.group_id} '
|
||||
f'检测刷屏 被禁言 {mute_data[group_id]["duration"] / 60} 分钟')
|
||||
if mute_data[group_id]["duration"] != 0:
|
||||
await bot.set_group_ban(
|
||||
group_id=event.group_id,
|
||||
user_id=event.user_id,
|
||||
duration=mute_data[group_id]["duration"],
|
||||
)
|
||||
await mute.send("检测到恶意刷屏，真寻要把你关进小黑屋！", at_sender=True)
|
||||
mute_dict[event.user_id]["count"] = 0
|
||||
logger.info(
|
||||
f"USER {event.user_id} GROUP {event.group_id} "
|
||||
f'检测刷屏 被禁言 {mute_data[group_id]["duration"] / 60} 分钟'
|
||||
)
|
||||
except ActionFailed:
|
||||
pass
|
||||
|
||||
@@ -103,29 +115,28 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
||||
group_id = str(event.group_id)
|
||||
if not mute_data.get(group_id):
|
||||
mute_data[group_id] = {
|
||||
'count': 10,
|
||||
'time': 7,
|
||||
'duration': 0
|
||||
}
|
||||
mute_data[group_id] = {"count": 10, "time": 7, "duration": 0}
|
||||
msg = get_message_text(event.json())
|
||||
if state["_prefix"]["raw_command"] == '刷屏检测设置':
|
||||
await mute_setting.finish(f'最大次数：{mute_data[group_id]["count"]} 次\n'
|
||||
f'规定时间：{mute_data[group_id]["time"]} 秒\n'
|
||||
f'禁言时长：{mute_data[group_id]["duration"] / 60} 分钟\n'
|
||||
f'【在规定时间内发送相同消息超过最大次数则禁言\n当禁言时长为0时关闭此功能】')
|
||||
if state["_prefix"]["raw_command"] == "刷屏检测设置":
|
||||
await mute_setting.finish(
|
||||
f'最大次数：{mute_data[group_id]["count"]} 次\n'
|
||||
f'规定时间：{mute_data[group_id]["time"]} 秒\n'
|
||||
f'禁言时长：{mute_data[group_id]["duration"] / 60} 分钟\n'
|
||||
f"【在规定时间内发送相同消息超过最大次数则禁言\n当禁言时长为0时关闭此功能】"
|
||||
)
|
||||
if not is_number(msg):
|
||||
await mute.finish('设置的参数必须是数字啊！', at_sender=True)
|
||||
if state["_prefix"]["raw_command"] == '设置检测时间':
|
||||
mute_data[group_id]['time'] = int(msg)
|
||||
msg += '秒'
|
||||
if state["_prefix"]["raw_command"] == '设置检测次数':
|
||||
mute_data[group_id]['count'] = int(msg)
|
||||
msg += ' 次'
|
||||
if state["_prefix"]["raw_command"] == '设置禁言时长':
|
||||
mute_data[group_id]['duration'] = int(msg) * 60
|
||||
msg += ' 分钟'
|
||||
await mute.finish("设置的参数必须是数字啊！", at_sender=True)
|
||||
if state["_prefix"]["raw_command"] == "设置检测时间":
|
||||
mute_data[group_id]["time"] = int(msg)
|
||||
msg += "秒"
|
||||
if state["_prefix"]["raw_command"] == "设置检测次数":
|
||||
mute_data[group_id]["count"] = int(msg)
|
||||
msg += " 次"
|
||||
if state["_prefix"]["raw_command"] == "设置禁言时长":
|
||||
mute_data[group_id]["duration"] = int(msg) * 60
|
||||
msg += " 分钟"
|
||||
await mute_setting.send(f'刷屏检测：{state["_prefix"]["raw_command"]}为 {msg}')
|
||||
logger.info(f'USER {event.user_id} GROUP {group_id} {state["_prefix"]["raw_command"]}：{msg}')
|
||||
logger.info(
|
||||
f'USER {event.user_id} GROUP {group_id} {state["_prefix"]["raw_command"]}：{msg}'
|
||||
)
|
||||
save_data()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user