mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
218 lines
7.9 KiB
Python
Executable File
218 lines
7.9 KiB
Python
Executable File
from configs.path_config import DATA_PATH
|
|
from nonebot.matcher import Matcher
|
|
from nonebot.message import run_postprocessor
|
|
from nonebot.typing import T_State
|
|
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent
|
|
from datetime import datetime
|
|
from utils.manager import plugins2settings_manager
|
|
from utils.utils import scheduler
|
|
from nonebot.typing import Optional
|
|
from pathlib import Path
|
|
|
|
try:
|
|
import ujson as json
|
|
except ModuleNotFoundError:
|
|
import json
|
|
|
|
|
|
__zx_plugin_name__ = "功能调用统计 [Hidden]"
|
|
__plugin_version__ = 0.1
|
|
__plugin_author__ = "HibiKier"
|
|
|
|
statistics_group_file = DATA_PATH / "statistics" / "_prefix_count.json"
|
|
statistics_user_file = DATA_PATH / "statistics" / "_prefix_user_count.json"
|
|
|
|
try:
|
|
with open(statistics_group_file, "r", encoding="utf8") as f:
|
|
_prefix_count_dict = json.load(f)
|
|
except (FileNotFoundError, ValueError):
|
|
_prefix_count_dict = {
|
|
"total_statistics": {
|
|
"total": {},
|
|
},
|
|
"day_statistics": {
|
|
"total": {},
|
|
},
|
|
"week_statistics": {
|
|
"total": {},
|
|
},
|
|
"month_statistics": {
|
|
"total": {},
|
|
},
|
|
"start_time": str(datetime.now().date()),
|
|
"day_index": 0,
|
|
}
|
|
|
|
try:
|
|
with open(statistics_user_file, "r", encoding="utf8") as f:
|
|
_prefix_user_count_dict = json.load(f)
|
|
except (FileNotFoundError, ValueError):
|
|
_prefix_user_count_dict = {
|
|
"total_statistics": {
|
|
"total": {},
|
|
},
|
|
"day_statistics": {
|
|
"total": {},
|
|
},
|
|
"week_statistics": {
|
|
"total": {},
|
|
},
|
|
"month_statistics": {
|
|
"total": {},
|
|
},
|
|
"start_time": str(datetime.now().date()),
|
|
"day_index": 0,
|
|
}
|
|
|
|
|
|
# 以前版本转换
|
|
if _prefix_count_dict.get("day_index") is None:
|
|
tmp = _prefix_count_dict.copy()
|
|
_prefix_count_dict = {
|
|
"total_statistics": tmp["total_statistics"],
|
|
"day_statistics": {
|
|
"total": {},
|
|
},
|
|
"week_statistics": {
|
|
"total": {},
|
|
},
|
|
"month_statistics": {
|
|
"total": {},
|
|
},
|
|
"start_time": tmp["start_time"],
|
|
"day_index": 0,
|
|
}
|
|
|
|
|
|
# 添加命令次数
|
|
@run_postprocessor
|
|
async def _(
|
|
matcher: Matcher,
|
|
exception: Optional[Exception],
|
|
bot: Bot,
|
|
event: GroupMessageEvent,
|
|
state: T_State,
|
|
):
|
|
global _prefix_count_dict
|
|
if (
|
|
matcher.type == "message"
|
|
and matcher.priority not in [1, 9]
|
|
and matcher.plugin_name not in ["update_info", "statistics_handle"]
|
|
):
|
|
module = matcher.plugin_name
|
|
day_index = _prefix_count_dict["day_index"]
|
|
try:
|
|
group_id = str(event.group_id)
|
|
except AttributeError:
|
|
group_id = "total"
|
|
user_id = str(event.user_id)
|
|
plugin_name = plugins2settings_manager.get_plugin_data(module)
|
|
if plugin_name and plugin_name.get('cmd'):
|
|
plugin_name = plugin_name.get('cmd')[0]
|
|
check_exists_key(group_id, user_id, plugin_name)
|
|
for data in [_prefix_count_dict, _prefix_user_count_dict]:
|
|
data["total_statistics"]["total"][plugin_name] += 1
|
|
data["day_statistics"]["total"][plugin_name] += 1
|
|
data["week_statistics"]["total"][plugin_name] += 1
|
|
data["month_statistics"]["total"][plugin_name] += 1
|
|
# print(_prefix_count_dict)
|
|
if group_id != "total":
|
|
for data in [_prefix_count_dict, _prefix_user_count_dict]:
|
|
if data == _prefix_count_dict:
|
|
key = group_id
|
|
else:
|
|
key = user_id
|
|
data["total_statistics"][key][plugin_name] += 1
|
|
data["day_statistics"][key][plugin_name] += 1
|
|
data["week_statistics"][key][str(day_index % 7)][
|
|
plugin_name
|
|
] += 1
|
|
data["month_statistics"][key][str(day_index % 30)][
|
|
plugin_name
|
|
] += 1
|
|
with open(statistics_group_file, "w", encoding="utf8") as f:
|
|
json.dump(_prefix_count_dict, f, indent=4, ensure_ascii=False)
|
|
with open(
|
|
statistics_user_file, "w", encoding="utf8"
|
|
) as f:
|
|
json.dump(_prefix_user_count_dict, f, ensure_ascii=False, indent=4)
|
|
|
|
|
|
def check_exists_key(group_id: str, user_id: str, plugin_name: str):
|
|
global _prefix_count_dict, _prefix_user_count_dict
|
|
for data in [_prefix_count_dict, _prefix_user_count_dict]:
|
|
if data == _prefix_count_dict:
|
|
key = group_id
|
|
else:
|
|
key = user_id
|
|
if not data["total_statistics"]["total"].get(plugin_name):
|
|
data["total_statistics"]["total"][plugin_name] = 0
|
|
if not data["day_statistics"]["total"].get(plugin_name):
|
|
data["day_statistics"]["total"][plugin_name] = 0
|
|
if not data["week_statistics"]["total"].get(plugin_name):
|
|
data["week_statistics"]["total"][plugin_name] = 0
|
|
if not data["month_statistics"]["total"].get(plugin_name):
|
|
data["month_statistics"]["total"][plugin_name] = 0
|
|
|
|
if not data["total_statistics"].get(key):
|
|
data["total_statistics"][key] = {}
|
|
if not data["total_statistics"][key].get(plugin_name):
|
|
data["total_statistics"][key][plugin_name] = 0
|
|
if not data["day_statistics"].get(key):
|
|
data["day_statistics"][key] = {}
|
|
if not data["day_statistics"][key].get(plugin_name):
|
|
data["day_statistics"][key][plugin_name] = 0
|
|
|
|
if key != 'total':
|
|
if not data["week_statistics"].get(key):
|
|
data["week_statistics"][key] = {}
|
|
if data["week_statistics"][key].get("0") is None:
|
|
for i in range(7):
|
|
data["week_statistics"][key][str(i)] = {}
|
|
if data["week_statistics"][key]["0"].get(plugin_name) is None:
|
|
for i in range(7):
|
|
data["week_statistics"][key][str(i)][plugin_name] = 0
|
|
|
|
if not data["month_statistics"].get(key):
|
|
data["month_statistics"][key] = {}
|
|
if data["month_statistics"][key].get("0") is None:
|
|
for i in range(30):
|
|
data["month_statistics"][key][str(i)] = {}
|
|
if data["month_statistics"][key]["0"].get(plugin_name) is None:
|
|
for i in range(30):
|
|
data["month_statistics"][key][str(i)][plugin_name] = 0
|
|
|
|
|
|
# 天
|
|
@scheduler.scheduled_job(
|
|
"cron",
|
|
hour=0,
|
|
minute=1,
|
|
)
|
|
async def _():
|
|
for data in [_prefix_count_dict, _prefix_user_count_dict]:
|
|
data["day_index"] += 1
|
|
for x in data["day_statistics"].keys():
|
|
for key in data["day_statistics"][x].keys():
|
|
try:
|
|
data["day_statistics"][x][key] = 0
|
|
except KeyError:
|
|
pass
|
|
for type_ in ["week_statistics", "month_statistics"]:
|
|
index = str(
|
|
data["day_index"] % 7
|
|
if type_ == "week_statistics"
|
|
else data["day_index"] % 30
|
|
)
|
|
for x in data[type_].keys():
|
|
try:
|
|
for key in data[type_][x][index].keys():
|
|
data[type_][x][index][key] = 0
|
|
except KeyError:
|
|
pass
|
|
with open(statistics_group_file, "w", encoding="utf8") as f:
|
|
json.dump(_prefix_count_dict, f, indent=4, ensure_ascii=False)
|
|
with open(statistics_user_file, "w", encoding="utf8") as f:
|
|
json.dump(_prefix_user_count_dict, f, indent=4, ensure_ascii=False)
|
|
|