diff --git a/plugins/statistics/statistics_handle.py b/plugins/statistics/statistics_handle.py index e84e7720..c0623d44 100644 --- a/plugins/statistics/statistics_handle.py +++ b/plugins/statistics/statistics_handle.py @@ -103,8 +103,18 @@ async def _(bot: Bot, event: MessageEvent, state: T_State): itype = 'month_statistics' else: itype = 'total_statistics' - data = data[itype]["total"] - bar_graph = await init_bar_graph(data, state["_prefix"]["raw_command"]) + tmp_dict = {} + data = data[itype] + for group in data.keys(): + if group != 'total': + for day in data[group].keys(): + for plugin_name in data[group][day].keys(): + if data[group][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[group][day][plugin_name] + bar_graph = await init_bar_graph(tmp_dict, state["_prefix"]["raw_command"]) await asyncio.get_event_loop().run_in_executor(None, bar_graph.gen_graph) await statistics.finish(image(b64=bar_graph.pic2bs4())) return