mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
104 lines
2.8 KiB
Python
104 lines
2.8 KiB
Python
from datetime import datetime
|
||
|
||
from nonebot.adapters import Bot
|
||
|
||
from zhenxun.models.friend_user import FriendUser
|
||
from zhenxun.models.group_member_info import GroupInfoUser
|
||
from zhenxun.utils.image_utils import BuildImage, ImageTemplate
|
||
|
||
from .model import BlackWord
|
||
from .utils import Config, _get_punish
|
||
|
||
|
||
async def show_black_text_image(
|
||
user_id: str | None,
|
||
group_id: str | None,
|
||
date: datetime | None,
|
||
data_type: str = "=",
|
||
) -> BuildImage:
|
||
"""展示记录名单
|
||
|
||
参数:
|
||
bot: bot
|
||
user: 用户id
|
||
group_id: 群组id
|
||
date: 日期
|
||
data_type: 日期搜索类型
|
||
|
||
返回:
|
||
BuildImage: 数据图片
|
||
"""
|
||
data_list = await BlackWord.get_black_data(user_id, group_id, date, data_type)
|
||
column_name = [
|
||
"ID",
|
||
"昵称",
|
||
"UID",
|
||
"GID",
|
||
"文本",
|
||
"检测内容",
|
||
"检测等级",
|
||
"惩罚",
|
||
"平台",
|
||
"记录日期",
|
||
]
|
||
column_list = []
|
||
uid_list = [u for u in data_list]
|
||
uid2name = {
|
||
u.user_id: u.user_name for u in await FriendUser.filter(user_id__in=uid_list)
|
||
}
|
||
for i, data in enumerate(data_list):
|
||
uname = uid2name.get(data.user_id)
|
||
if not uname:
|
||
if u := await GroupInfoUser.get_or_none(
|
||
user_id=data.user_id, group_id=data.group_id
|
||
):
|
||
uname = u.user_name
|
||
if len(data.plant_text) > 30:
|
||
data.plant_text = data.plant_text[:30] + "..."
|
||
column_list.append(
|
||
[
|
||
i,
|
||
uname or data.user_id,
|
||
data.user_id,
|
||
data.group_id,
|
||
data.plant_text,
|
||
data.black_word,
|
||
data.punish_level,
|
||
data.punish,
|
||
data.platform,
|
||
data.create_time,
|
||
]
|
||
)
|
||
A = await ImageTemplate.table_page(
|
||
"记录名单", "一个都不放过!", column_name, column_list
|
||
)
|
||
return A
|
||
|
||
|
||
async def set_user_punish(
|
||
bot: Bot, user_id: str, group_id: str | None, id_: int, punish_level: int
|
||
) -> str:
|
||
"""设置惩罚
|
||
|
||
参数:
|
||
user_id: 用户id
|
||
group_id: 群组id或频道id
|
||
id_: 记录下标
|
||
punish_level: 惩罚等级
|
||
|
||
返回:
|
||
str: 结果
|
||
"""
|
||
result = await _get_punish(bot, punish_level, user_id, group_id)
|
||
punish = {
|
||
1: "永久ban",
|
||
2: "删除好友",
|
||
3: f"ban {result} 天",
|
||
4: f"ban {result} 分钟",
|
||
5: "口头警告",
|
||
}
|
||
if await BlackWord.set_user_punish(user_id, punish[punish_level], id_=id_):
|
||
return f"已对 USER {user_id} 进行 {punish[punish_level]} 处罚。"
|
||
else:
|
||
return "操作失败，可能未找到用户，id或敏感词"
|