Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 06:12:53 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fix mute

Browse Source
This commit is contained in:
HibiKier 2022-05-21 13:46:29 +08:00
parent c0d055b825
commit 81e4bca9ef
2 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@ -238,7 +238,7 @@ __Docker 最新版本由 [Sakuracio](https://github.com/Sakuracio) 提供__
## 更新 ## 更新
### 2022/5/ ### 2022/5/21
* 修复搜番无结果时报错无正确反馈 * 修复搜番无结果时报错无正确反馈
* 解锁了windows上无法使用playwright的限制 * 解锁了windows上无法使用playwright的限制
@ -248,6 +248,8 @@ __Docker 最新版本由 [Sakuracio](https://github.com/Sakuracio) 提供__
* 修复了bilibili订阅直播间订阅up名称不一致的问题 * 修复了bilibili订阅直播间订阅up名称不一致的问题
* 修复原神玩家查询没开地图时报错 * 修复原神玩家查询没开地图时报错
* 最低priority修改为 999 * 最低priority修改为 999
* 修复刷屏检测失效
* 优化刷屏显示设置禁言时长显示，并改为分钟
### 2022/5/19 ### 2022/5/19

17
plugins/mute.py
View File

@ -1,8 +1,7 @@
from nonebot import on_message, on_command from nonebot import on_message, on_command
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message
from nonebot.adapters.onebot.v11.permission import GROUP from nonebot.adapters.onebot.v11.permission import GROUP
from utils.utils import is_number, get_message_img from utils.utils import is_number, get_message_img, get_message_text
from nonebot.typing import T_State
from nonebot.adapters.onebot.v11.exception import ActionFailed from nonebot.adapters.onebot.v11.exception import ActionFailed
from configs.path_config import DATA_PATH, TEMP_PATH from configs.path_config import DATA_PATH, TEMP_PATH
from utils.image_utils import get_img_hash from utils.image_utils import get_img_hash
@ -10,7 +9,7 @@ from services.log import logger
from configs.config import NICKNAME, Config from configs.config import NICKNAME, Config
from utils.http_utils import AsyncHttpx from utils.http_utils import AsyncHttpx
from nonebot.params import CommandArg, Command from nonebot.params import CommandArg, Command
from typing import Tuple from typing import Tuple, Dict, Any
import time import time
try: try:
@ -57,7 +56,7 @@ mute_setting = on_command(
) )
def get_data(): def get_data() -> Dict[Any]:
try: 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) data = json.load(f)
@ -72,7 +71,7 @@ def save_data():
json.dump(mute_data, f, indent=4) json.dump(mute_data, f, indent=4)
async def download_img_and_hash(url, group_id): async def download_img_and_hash(url, group_id) -> str:
if await AsyncHttpx.download_file( if await AsyncHttpx.download_file(
url, TEMP_PATH / f"mute_{group_id}_img.jpg" url, TEMP_PATH / f"mute_{group_id}_img.jpg"
): ):
@ -85,9 +84,9 @@ mute_data = get_data()
@mute.handle() @mute.handle()
async def _(bot: Bot, event: GroupMessageEvent, cmd: Tuple[str, ...] = Command(), arg: Message = CommandArg()): async def _(bot: Bot, event: GroupMessageEvent):
group_id = str(event.group_id) group_id = str(event.group_id)
msg = arg.extract_plain_text().strip() msg = get_message_text(event.json())
img_list = get_message_img(event.json()) img_list = get_message_img(event.json())
img_hash = "" img_hash = ""
for img in img_list: for img in img_list:
@ -102,8 +101,6 @@ async def _(bot: Bot, event: GroupMessageEvent, cmd: Tuple[str, ...] = Command()
if not mute_dict.get(event.user_id): 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: else:
if cmd or not msg:
return
if msg and msg.find(mute_dict[event.user_id]["msg"]) != -1: if msg and msg.find(mute_dict[event.user_id]["msg"]) != -1:
mute_dict[event.user_id]["count"] += 1 mute_dict[event.user_id]["count"] += 1
else: else:
@ -123,7 +120,7 @@ async def _(bot: Bot, event: GroupMessageEvent, cmd: Tuple[str, ...] = Command()
await bot.set_group_ban( await bot.set_group_ban(
group_id=event.group_id, group_id=event.group_id,
user_id=event.user_id, user_id=event.user_id,
duration=mute_data[group_id]["duration"], duration=mute_data[group_id]["duration"] * 60,
) )
await mute.send(f"检测到恶意刷屏，{NICKNAME}要把你关进小黑屋！", at_sender=True) await mute.send(f"检测到恶意刷屏，{NICKNAME}要把你关进小黑屋！", at_sender=True)
mute_dict[event.user_id]["count"] = 0 mute_dict[event.user_id]["count"] = 0