fix: 状态卡片百分比错误

feat: 群聊Bot开关,防止Bot乱窜
This commit is contained in:
远野千束 2024-04-12 00:32:08 +08:00
parent f9e5742821
commit 33dd2f104d
11 changed files with 129 additions and 25 deletions

View File

@ -28,6 +28,8 @@ tag:
你需要使用Onebot标准的实现端来连接到轻雪并将消息上报给轻雪下面已经列出一些推荐的实现端 你需要使用Onebot标准的实现端来连接到轻雪并将消息上报给轻雪下面已经列出一些推荐的实现端
- `Playwright`安装失败 - `Playwright`安装失败
- 输入`playwright install`安装浏览器 - 输入`playwright install`安装浏览器
- 其他问题
- 加入QQ群[775840726](http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=SzmDYbfR6jY94o9KFNon7AwelRyI6M_u&authKey=ygeBdEmdFNyCWuNR4w0M1M8%2B5oDg7k%2FDfN0tzBkYcnbB%2FGHNnlVEnCIGbdftsnn7&noverify=0&group_code=775840726)
#### 推荐方案(QQ) #### 推荐方案(QQ)

View File

@ -179,6 +179,7 @@ async def _(matcher: Matcher):
# system hook # system hook
@Bot.on_calling_api # 图片模式检测 @Bot.on_calling_api # 图片模式检测
async def test_for_md_image(bot: T_Bot, api: str, data: dict): async def test_for_md_image(bot: T_Bot, api: str, data: dict):
# 截获大图发送转换为markdown发送
if api in ["send_msg", "send_private_msg", "send_group_msg"] and markdown_image and data.get("user_id") != bot.self_id: if api in ["send_msg", "send_private_msg", "send_group_msg"] and markdown_image and data.get("user_id") != bot.self_id:
if api == "send_msg" and data.get("message_type") == "private" or api == "send_private_msg": if api == "send_msg" and data.get("message_type") == "private" or api == "send_private_msg":
session_type = "private" session_type = "private"
@ -207,6 +208,7 @@ async def test_for_md_image(bot: T_Bot, api: str, data: dict):
@driver.on_startup @driver.on_startup
async def on_startup(): async def on_startup():
temp_data = common_db.first(TempConfig(), default=TempConfig()) temp_data = common_db.first(TempConfig(), default=TempConfig())
# 储存重启信息
if temp_data.data.get("reload", False): if temp_data.data.get("reload", False):
delta_time = time.time() - temp_data.data.get("reload_time", 0) delta_time = time.time() - temp_data.data.get("reload_time", 0)
temp_data.data["delta_time"] = delta_time temp_data.data["delta_time"] = delta_time
@ -221,6 +223,7 @@ async def on_shutdown():
@driver.on_bot_connect @driver.on_bot_connect
async def _(bot: T_Bot): async def _(bot: T_Bot):
temp_data = common_db.first(TempConfig(), default=TempConfig()) temp_data = common_db.first(TempConfig(), default=TempConfig())
# 用于重启计时
if temp_data.data.get("reload", False): if temp_data.data.get("reload", False):
temp_data.data["reload"] = False temp_data.data["reload"] = False
reload_bot_id = temp_data.data.get("reload_bot_id", 0) reload_bot_id = temp_data.data.get("reload_bot_id", 0)

View File

@ -123,3 +123,17 @@ def get_plugin_can_be_toggle(plugin_name: str) -> bool:
""" """
plug = nonebot.plugin.get_plugin(plugin_name) plug = nonebot.plugin.get_plugin(plugin_name)
return plug.metadata.extra.get("toggleable", True) if plug and plug.metadata else True return plug.metadata.extra.get("toggleable", True) if plug and plug.metadata else True
def get_group_enable(group_id: str) -> bool:
"""
获取群组是否启用插机器人
Args:
group_id (str): 群组ID
Returns:
bool: 群组是否启用插件
"""
session: Group = group_db.first(Group(), "group_id = ?", group_id, default=Group(group_id=group_id))
return session.enable

View File

@ -5,8 +5,8 @@ import nonebot.plugin
import pip import pip
from io import StringIO from io import StringIO
from arclet.alconna import MultiVar from arclet.alconna import MultiVar
from nonebot import require from nonebot import Bot, require
from nonebot.exception import FinishedException, IgnoredException from nonebot.exception import FinishedException, IgnoredException, MockApiException
from nonebot.internal.adapter import Event from nonebot.internal.adapter import Event
from nonebot.internal.matcher import Matcher from nonebot.internal.matcher import Matcher
from nonebot.message import run_preprocessor from nonebot.message import run_preprocessor
@ -37,32 +37,31 @@ disable = "disable"
Subcommand( Subcommand(
"enable", "enable",
Args["plugin_name", str], Args["plugin_name", str],
alias=["启用"], alias=["e", "启用"],
), ),
Subcommand( Subcommand(
"disable", "disable",
Args["plugin_name", str], Args["plugin_name", str],
alias=["停用"], alias=["d", "停用"],
), ),
Subcommand( Subcommand(
enable_global, enable_global,
Args["plugin_name", str], Args["plugin_name", str],
alias=["全局启用"], alias=["eg", "全局启用"],
), ),
Subcommand( Subcommand(
disable_global, disable_global,
Args["plugin_name", str], Args["plugin_name", str],
alias=["全局停用"], alias=["dg", "全局停用"],
), ),
# 安装部分 # 安装部分
Subcommand( Subcommand(
"update", "update",
alias=["u"], alias=["u", "更新"],
), ),
Subcommand( Subcommand(
"search", "search",
Args["keywords", MultiVar(str)]["show_num", int, 15], Args["keywords", MultiVar(str)],
alias=["s", "搜索"], alias=["s", "搜索"],
), ),
Subcommand( Subcommand(
@ -79,11 +78,6 @@ disable = "disable"
"list", "list",
Args["page", int, 1]["num", int, 10], Args["page", int, 1]["num", int, 10],
alias=["ls", "列表"], alias=["ls", "列表"],
),
Subcommand(
"usage",
Args["plugin_name", str],
alias=["详情"],
) )
) )
).handle() ).handle()
@ -306,7 +300,7 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot, npm: Matcher):
# 检查是否有 metadata 属性 # 检查是否有 metadata 属性
# 添加帮助按钮 # 添加帮助按钮
btn_usage = md.btn_cmd(ulang.get("npm.usage"), f"npm usage {storePlugin.name}", False) btn_usage = md.btn_cmd(ulang.get("npm.usage"), f"help {storePlugin.name}", False)
store_plugin = await get_store_plugin(storePlugin.name) store_plugin = await get_store_plugin(storePlugin.name)
session_enable = get_plugin_session_enable(event, storePlugin.name) session_enable = get_plugin_session_enable(event, storePlugin.name)
if store_plugin: if store_plugin:
@ -353,11 +347,49 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot, npm: Matcher):
reply += "\n\n***\n" reply += "\n\n***\n"
await md.send_md(reply, bot, event=event) await md.send_md(reply, bot, event=event)
elif sc.get("usage"):
# TODO
pass
else: else:
pass await npm.finish(ulang.get("liteyuki.invalid_command"))
@on_alconna(
aliases={"群聊"},
command=Alconna(
"gm",
Subcommand(
enable,
Args["group_id", str, None],
alias=["e", "启用"],
),
Subcommand(
disable,
Args["group_id", str, None],
alias=["d", "停用"],
),
),
).handle()
async def _(event: T_MessageEvent, gm: Matcher, result: Arparma):
ulang = get_user_lang(str(event.user_id))
to_enable = result.subcommands.get(enable) is not None
group_id = result.subcommands.get(enable, result.subcommands.get(disable)).args.get("group_id")
if group_id is None and event.message_type == "group":
group_id = str(event.group_id)
else:
await gm.finish(ulang.get("liteyuki.invalid_command"), liteyuki_pass=True)
enabled = get_group_enable(group_id)
if enabled == to_enable:
await gm.finish(ulang.get("liteyuki.group_already", STATUS=ulang.get("npm.enable") if to_enable else ulang.get("npm.disable"), GROUP=group_id), liteyuki_pass=True)
else:
group: Group = group_db.first(Group(), "group_id = ?", group_id, default=Group(group_id=group_id))
if to_enable:
group.enable = True
else:
group.enable = False
group_db.upsert(group)
await gm.finish(
ulang.get("liteyuki.group_success", STATUS=ulang.get("npm.enable") if to_enable else ulang.get("npm.disable"), GROUP=group_id),
liteyuki_pass=True
)
@run_preprocessor @run_preprocessor
@ -372,6 +404,15 @@ async def pre_handle(event: Event, matcher: Matcher):
raise IgnoredException("Plugin disabled in session") raise IgnoredException("Plugin disabled in session")
@Bot.on_calling_api
async def block_disable_session(bot: Bot, api: str, args: dict):
if "group_id" in args and not args.get("liteyuki_pass", False):
group_id = args["group_id"]
if not get_group_enable(group_id):
nonebot.logger.debug(f"Group {group_id} disabled")
raise MockApiException(f"Group {group_id} disabled")
async def npm_update() -> bool: async def npm_update() -> bool:
""" """
更新本地插件json缓存 更新本地插件json缓存

View File

@ -122,3 +122,6 @@ rpm.move_down=下移
rpm.move_top=置顶 rpm.move_top=置顶
npm.update=更新 npm.update=更新
liteyuki.group_already=群 {GROUP} 已经是 {STATUS} 状态,无需重复操作
liteyuki.group_success=群 {GROUP} {STATUS} 成功

View File

@ -0,0 +1,18 @@
#data-storage {
display: none;
}
body {
background-repeat: repeat-y;
background-size: cover;
background-position: center;
margin: 20px;
}
.info-box {
border-radius: 30px;
padding: 30px;
backdrop-filter: blur(10px);
background-color: rgba(0, 0, 0, 0.5);
margin-bottom: 20px;
}

View File

@ -5,7 +5,6 @@ body {
color: white; color: white;
/ / 上10px左右10px下0px / / margin: 24 px; / / 上10px左右10px下0px / / margin: 24 px;
margin: 20px; margin: 20px;
} }
.info-box { .info-box {

View File

@ -0,0 +1,9 @@
const data = JSON.parse(document.getElementById('data').innerText);
function createPieChartOption(title, data){
// data为各项占比列表
}
function createBarChartOption(title, percent){
// percent为百分比最大值为100
}

View File

@ -132,8 +132,7 @@
used += item.value used += item.value
} }
}) })
console.log(used, total) return (1 - used / total) * 100
return used / total * 100
} }
@ -146,6 +145,7 @@
top: 'center', top: 'center',
textStyle: { textStyle: {
//文字颜色 //文字颜色
lineHeight: 36, lineHeight: 36,
color: '#fff', color: '#fff',
fontSize: 30 fontSize: 30

View File

@ -0,0 +1,14 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Liteyuki Status</title>
<link rel="stylesheet" href="css/card.css">
</head>
<body>
<div class="data-storage" id="data">{{ data | tojson }}</div>
<div class="info-box" id="device-info"></div>
<div class="info-box" id="disk-info"></div>
<div class="info-box" id="motto-info"></div>
</body>
</html>

View File

@ -28,6 +28,7 @@ class Group(LiteModel):
group_name: str = Field(str(), alias="group_name") group_name: str = Field(str(), alias="group_name")
enabled_plugins: list[str] = Field([], alias="enabled_plugins") enabled_plugins: list[str] = Field([], alias="enabled_plugins")
disabled_plugins: list[str] = Field([], alias="disabled_plugins") disabled_plugins: list[str] = Field([], alias="disabled_plugins")
enable: bool = Field(True, alias="enable") # 群聊全局机器人是否启用
class InstalledPlugin(LiteModel): class InstalledPlugin(LiteModel):