mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-28 16:20:56 +08:00
130 lines
3.6 KiB
Python
130 lines
3.6 KiB
Python
import time
|
|
|
|
from pydantic import BaseModel, Field
|
|
import ujson as json
|
|
|
|
from zhenxun.configs.config import Config
|
|
from zhenxun.configs.path_config import DATA_PATH
|
|
|
|
base_config = Config.get("mute_setting")
|
|
|
|
|
|
class GroupData(BaseModel):
|
|
count: int
|
|
"""次数"""
|
|
time: int
|
|
"""检测时长"""
|
|
duration: int
|
|
"""禁言时长"""
|
|
message_data: dict = Field(default_factory=dict)
|
|
"""消息存储"""
|
|
|
|
|
|
class MuteManager:
|
|
file = DATA_PATH / "group_mute_data.json"
|
|
|
|
def __init__(self) -> None:
|
|
self._group_data: dict[str, GroupData] = {}
|
|
if self.file.exists():
|
|
with open(self.file, encoding="utf-8") as f:
|
|
_data = json.load(f)
|
|
for gid, gdata in _data.items():
|
|
self._group_data[gid] = GroupData(
|
|
count=gdata["count"],
|
|
time=gdata["time"],
|
|
duration=gdata["duration"],
|
|
)
|
|
|
|
def get_group_data(self, group_id: str) -> GroupData:
|
|
"""获取群组数据
|
|
|
|
参数:
|
|
group_id: 群组id
|
|
|
|
返回:
|
|
GroupData: GroupData
|
|
"""
|
|
if group_id not in self._group_data:
|
|
self._group_data[group_id] = GroupData(
|
|
count=base_config.get("MUTE_DEFAULT_COUNT", 10) or 10,
|
|
time=base_config.get("MUTE_DEFAULT_TIME", 7) or 7,
|
|
duration=base_config.get("MUTE_DEFAULT_DURATION", 10) or 10,
|
|
)
|
|
return self._group_data[group_id]
|
|
|
|
def reset(self, user_id: str, group_id: str):
|
|
"""重置用户检查次数
|
|
|
|
参数:
|
|
user_id: 用户id
|
|
group_id: 群组id
|
|
"""
|
|
if group_data := self._group_data.get(group_id):
|
|
if user_id in group_data.message_data:
|
|
group_data.message_data[user_id]["count"] = 0
|
|
|
|
def save_data(self):
|
|
"""保存数据"""
|
|
data = {
|
|
gid: {
|
|
"count": gdata.count,
|
|
"time": gdata.time,
|
|
"duration": gdata.duration,
|
|
}
|
|
for gid, gdata in self._group_data.items()
|
|
}
|
|
with open(self.file, "w", encoding="utf-8") as f:
|
|
json.dump(data, f, indent=4, ensure_ascii=False)
|
|
|
|
def add_message(self, user_id: str, group_id: str, message: str) -> int:
|
|
"""添加消息
|
|
|
|
参数:
|
|
user_id: 用户id
|
|
group_id: 群组id
|
|
message: 消息内容
|
|
|
|
返回:
|
|
int: 禁言时长
|
|
"""
|
|
group_data = self.get_group_data(group_id)
|
|
if group_data.duration == 0:
|
|
return 0
|
|
|
|
message_data = group_data.message_data
|
|
user_data = message_data.get(user_id)
|
|
now = time.time()
|
|
|
|
if not user_data:
|
|
message_data[user_id] = {
|
|
"time": now,
|
|
"count": 1,
|
|
"message": message,
|
|
}
|
|
return 0
|
|
|
|
# 超过检测时间窗口，重置计数
|
|
if now - user_data["time"] > group_data.time:
|
|
user_data["time"] = now
|
|
user_data["count"] = 1
|
|
user_data["message"] = message
|
|
return 0
|
|
|
|
# 消息内容相似（包含之前的消息），累加计数
|
|
if user_data["message"] in message:
|
|
user_data["count"] += 1
|
|
else:
|
|
user_data["time"] = now
|
|
user_data["count"] = 1
|
|
|
|
user_data["message"] = message
|
|
|
|
# 检测是否触发刷屏
|
|
if user_data["count"] > group_data.count:
|
|
return group_data.duration
|
|
|
|
return 0
|
|
|
|
|
|
mute_manager = MuteManager()
|