From 1033d585c7da03faf53ba0c2a4ba418b4533622a Mon Sep 17 00:00:00 2001
From: HibiKier <45528451+HibiKier@users.noreply.github.com>
Date: Fri, 8 Oct 2021 16:28:04 +0800
Subject: [PATCH] Update statistics_handle.py

---
 plugins/statistics/statistics_handle.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

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