mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 06:12:53 +08:00
HibiKier
GitHub
commit
b5ce4fab31
@ -16,68 +16,68 @@ except ModuleNotFoundError:
|
||||
import json
|
||||
|
||||
|
||||
__zx_plugin_name__ = "功能调用统计可视化"
|
||||
__zx_plugin_name__ = "功能调用统计可视化"
|
||||
__plugin_usage__ = """
|
||||
usage：
|
||||
功能调用统计可视化
|
||||
指令：
|
||||
功能调用统计
|
||||
日功能调用统计
|
||||
周功能调用统计 ?[功能]
|
||||
月功能调用统计 ?[功能]
|
||||
我的功能调用统计
|
||||
我的日功能调用统计 ?[功能]
|
||||
我的周功能调用统计 ?[功能]
|
||||
我的月功能调用统计 ?[功能]
|
||||
usage：
|
||||
功能调用统计可视化
|
||||
指令：
|
||||
功能调用统计
|
||||
日功能调用统计
|
||||
周功能调用统计 ?[功能]
|
||||
月功能调用统计 ?[功能]
|
||||
我的功能调用统计
|
||||
我的日功能调用统计 ?[功能]
|
||||
我的周功能调用统计 ?[功能]
|
||||
我的月功能调用统计 ?[功能]
|
||||
""".strip()
|
||||
__plugin_superuser_usage__ = """
|
||||
usage：
|
||||
功能调用统计可视化
|
||||
指令：
|
||||
全局功能调用统计
|
||||
全局日功能调用统计
|
||||
全局周功能调用统计 ?[功能]
|
||||
全局月功能调用统计 ?[功能]
|
||||
usage：
|
||||
功能调用统计可视化
|
||||
指令：
|
||||
全局功能调用统计
|
||||
全局日功能调用统计
|
||||
全局周功能调用统计 ?[功能]
|
||||
全局月功能调用统计 ?[功能]
|
||||
""".strip()
|
||||
__plugin_des__ = "功能调用统计可视化"
|
||||
__plugin_des__ = "功能调用统计可视化"
|
||||
__plugin_cmd__ = [
|
||||
"功能调用统计",
|
||||
"全局功能调用统计 [_superuser]",
|
||||
"全局日功能调用统计 [_superuser]",
|
||||
"全局周功能调用统计 ?[功能] [_superuser]",
|
||||
"全局月功能调用统计 ?[功能] [_superuser]",
|
||||
"周功能调用统计 ?[功能]",
|
||||
"月功能调用统计 ?[功能]",
|
||||
"我的功能调用统计",
|
||||
"我的日功能调用统计 ?[功能]",
|
||||
"我的周功能调用统计 ?[功能]",
|
||||
"我的月功能调用统计 ?[功能]",
|
||||
"功能调用统计",
|
||||
"全局功能调用统计 [_superuser]",
|
||||
"全局日功能调用统计 [_superuser]",
|
||||
"全局周功能调用统计 ?[功能] [_superuser]",
|
||||
"全局月功能调用统计 ?[功能] [_superuser]",
|
||||
"周功能调用统计 ?[功能]",
|
||||
"月功能调用统计 ?[功能]",
|
||||
"我的功能调用统计",
|
||||
"我的日功能调用统计 ?[功能]",
|
||||
"我的周功能调用统计 ?[功能]",
|
||||
"我的月功能调用统计 ?[功能]",
|
||||
]
|
||||
__plugin_type__ = ("功能调用统计可视化", 1)
|
||||
__plugin_type__ = ("功能调用统计可视化", 1)
|
||||
__plugin_version__ = 0.1
|
||||
__plugin_author__ = "HibiKier"
|
||||
__plugin_settings__ = {
|
||||
"level": 5,
|
||||
"default_status": True,
|
||||
"limit_superuser": False,
|
||||
"cmd": ["功能调用统计"],
|
||||
"cmd": ["功能调用统计"],
|
||||
}
|
||||
|
||||
|
||||
statistics = on_command(
|
||||
"功能调用统计",
|
||||
"功能调用统计",
|
||||
aliases={
|
||||
"全局功能调用统计",
|
||||
"全局日功能调用统计",
|
||||
"全局周功能调用统计",
|
||||
"全局月功能调用统计",
|
||||
"日功能调用统计",
|
||||
"周功能调用统计",
|
||||
"月功能调用统计",
|
||||
"我的功能调用统计",
|
||||
"我的日功能调用统计",
|
||||
"我的周功能调用统计",
|
||||
"我的月功能调用统计",
|
||||
"全局功能调用统计",
|
||||
"全局日功能调用统计",
|
||||
"全局周功能调用统计",
|
||||
"全局月功能调用统计",
|
||||
"日功能调用统计",
|
||||
"周功能调用统计",
|
||||
"月功能调用统计",
|
||||
"我的功能调用统计",
|
||||
"我的日功能调用统计",
|
||||
"我的周功能调用统计",
|
||||
"我的月功能调用统计",
|
||||
},
|
||||
priority=5,
|
||||
block=True,
|
||||
@ -91,14 +91,14 @@ statistics_user_file = DATA_PATH / "statistics" / "_prefix_user_count.json"
|
||||
@statistics.handle()
|
||||
async def _(bot: Bot, event: MessageEvent, cmd: Tuple[str, ...] = Command(), arg: Message = CommandArg()):
|
||||
msg = arg.extract_plain_text().strip()
|
||||
if cmd[0][:2] == "全局":
|
||||
if cmd[0][:2] == "全局":
|
||||
if str(event.user_id) in bot.config.superusers:
|
||||
data: dict = json.load(open(statistics_group_file, "r", encoding="utf8"))
|
||||
if cmd[0][2] == '日':
|
||||
if cmd[0][2] == '日':
|
||||
_type = 'day_statistics'
|
||||
elif cmd[0][2] == '周':
|
||||
elif cmd[0][2] == '周':
|
||||
_type = 'week_statistics'
|
||||
elif cmd[0][2] == '月':
|
||||
elif cmd[0][2] == '月':
|
||||
_type = 'month_statistics'
|
||||
else:
|
||||
_type = 'total_statistics'
|
||||
@ -121,26 +121,26 @@ async def _(bot: Bot, event: MessageEvent, cmd: Tuple[str, ...] = Command(), arg
|
||||
await asyncio.get_event_loop().run_in_executor(None, bar_graph.gen_graph)
|
||||
await statistics.finish(image(b64=bar_graph.pic2bs4()))
|
||||
return
|
||||
if cmd[0][:2] == "我的":
|
||||
if cmd[0][:2] == "我的":
|
||||
_type = "user"
|
||||
key = str(event.user_id)
|
||||
cmd = list(cmd)
|
||||
cmd[0] = cmd[0][2:]
|
||||
if not statistics_user_file.exists():
|
||||
await statistics.finish("统计文件不存在...", at_sender=True)
|
||||
await statistics.finish("统计文件不存在...", at_sender=True)
|
||||
else:
|
||||
if not isinstance(event, GroupMessageEvent):
|
||||
await statistics.finish("请在群内调用此功能...")
|
||||
await statistics.finish("请在群内调用此功能...")
|
||||
_type = "group"
|
||||
key = str(event.group_id)
|
||||
if not statistics_group_file.exists():
|
||||
await statistics.finish("统计文件不存在...", at_sender=True)
|
||||
await statistics.finish("统计文件不存在...", at_sender=True)
|
||||
plugin = ""
|
||||
if cmd[0][0] == "日":
|
||||
if cmd[0][0] == "日":
|
||||
arg = "day_statistics"
|
||||
elif cmd[0][0] == "周":
|
||||
elif cmd[0][0] == "周":
|
||||
arg = "week_statistics"
|
||||
elif cmd[0][0] == "月":
|
||||
elif cmd[0][0] == "月":
|
||||
arg = "month_statistics"
|
||||
else:
|
||||
arg = "total_statistics"
|
||||
@ -148,15 +148,15 @@ async def _(bot: Bot, event: MessageEvent, cmd: Tuple[str, ...] = Command(), arg
|
||||
plugin = plugins2settings_manager.get_plugin_module(msg)
|
||||
if not plugin:
|
||||
if arg not in ["day_statistics", "total_statistics"]:
|
||||
await statistics.finish("未找到此功能的调用...", at_sender=True)
|
||||
await statistics.finish("未找到此功能的调用...", at_sender=True)
|
||||
if _type == "group":
|
||||
data: dict = json.load(open(statistics_group_file, "r", encoding="utf8"))
|
||||
if not data[arg].get(str(event.group_id)):
|
||||
await statistics.finish("该群统计数据不存在...", at_sender=True)
|
||||
await statistics.finish("该群统计数据不存在...", at_sender=True)
|
||||
else:
|
||||
data: dict = json.load(open(statistics_user_file, "r", encoding="utf8"))
|
||||
if not data[arg].get(str(event.user_id)):
|
||||
await statistics.finish("该用户统计数据不存在...", at_sender=True)
|
||||
await statistics.finish("该用户统计数据不存在...", at_sender=True)
|
||||
day_index = data["day_index"]
|
||||
data = data[arg][key]
|
||||
if _type == "group":
|
||||
@ -177,7 +177,7 @@ async def generate_statistics_img(
|
||||
pass
|
||||
bar_graph = None
|
||||
if arg == "day_statistics":
|
||||
bar_graph = await init_bar_graph(data, f"{name} 日功能调用统计")
|
||||
bar_graph = await init_bar_graph(data, f"{name} 日功能调用统计")
|
||||
elif arg == "week_statistics":
|
||||
if plugin:
|
||||
current_week = day_index % 7
|
||||
@ -205,7 +205,7 @@ async def generate_statistics_img(
|
||||
bar_graph = BuildMat(
|
||||
y=count,
|
||||
mat_type="line",
|
||||
title=f"{name} 周 {plugin} 功能调用统计【为7天统计】",
|
||||
title=f"{name} 周 {plugin} 功能调用统计【为7天统计】",
|
||||
x_index=week_lst,
|
||||
display_num=True,
|
||||
background=[
|
||||
@ -215,7 +215,7 @@ async def generate_statistics_img(
|
||||
bar_color=["*"],
|
||||
)
|
||||
else:
|
||||
bar_graph = await init_bar_graph(update_data(data), f"{name} 周功能调用统计【为7天统计】")
|
||||
bar_graph = await init_bar_graph(update_data(data), f"{name} 周功能调用统计【为7天统计】")
|
||||
elif arg == "month_statistics":
|
||||
if plugin:
|
||||
day_index = day_index % 30
|
||||
@ -229,7 +229,7 @@ async def generate_statistics_img(
|
||||
bar_graph = BuildMat(
|
||||
y=count,
|
||||
mat_type="line",
|
||||
title=f"{name} 月 {plugin} 功能调用统计【为30天统计】",
|
||||
title=f"{name} 月 {plugin} 功能调用统计【为30天统计】",
|
||||
x_index=day_lst,
|
||||
display_num=True,
|
||||
background=[
|
||||
@ -239,9 +239,9 @@ async def generate_statistics_img(
|
||||
bar_color=["*"],
|
||||
)
|
||||
else:
|
||||
bar_graph = await init_bar_graph(update_data(data), f"{name} 月功能调用统计【为30天统计】")
|
||||
bar_graph = await init_bar_graph(update_data(data), f"{name} 月功能调用统计【为30天统计】")
|
||||
elif arg == "total_statistics":
|
||||
bar_graph = await init_bar_graph(data, f"{name} 功能调用统计")
|
||||
bar_graph = await init_bar_graph(data, f"{name} 功能调用统计")
|
||||
await asyncio.get_event_loop().run_in_executor(None, bar_graph.gen_graph)
|
||||
return bar_graph.pic2bs4()
|
||||
|
||||
@ -270,10 +270,10 @@ def update_data(data: dict):
|
||||
tmp_dict = {}
|
||||
for day in data.keys():
|
||||
for plugin_name in data[day].keys():
|
||||
# print(f'{day}：{plugin_name} = {data[day][plugin_name]}')
|
||||
# print(f'{day}：{plugin_name} = {data[day][plugin_name]}')
|
||||
if data[day][plugin_name] is not None:
|
||||
if tmp_dict.get(plugin_name) is None:
|
||||
tmp_dict[plugin_name] = 1
|
||||
else:
|
||||
tmp_dict[plugin_name] += data[day][plugin_name]
|
||||
return tmp_dict
|
||||
return tmp_dict
|
||||
|
||||
Loading…
Reference in New Issue
Block a user