From 586020bc6ca4fa0319976d1f477a3776afa5edd8 Mon Sep 17 00:00:00 2001
From: PhenoM4n4n <61065078+phenom4n4n@users.noreply.github.com>
Date: Fri, 9 Oct 2020 16:11:22 -0700
Subject: [PATCH] chatchart blocking (#169)

---
 chatchart/chatchart.py | 21 +++++++++++++++++----
 1 file changed, 17 insertions(+), 4 deletions(-)

diff --git a/chatchart/chatchart.py b/chatchart/chatchart.py
index d023394..8266a76 100644
--- a/chatchart/chatchart.py
+++ b/chatchart/chatchart.py
@@ -3,10 +3,13 @@
 #  Big thanks to Redjumpman for changing the beta version from
 #  Imagemagick/cairosvg to matplotlib.
 #  Thanks to violetnyte for suggesting this cog.
-import discord
+import asyncio
+import functools
 import heapq
 from io import BytesIO
 from typing import Optional
+
+import discord
 import matplotlib
 
 matplotlib.use("agg")
@@ -137,6 +140,16 @@ class Chatchart(commands.Cog):
             key=lambda x: x[1],
         )
         others = 100 - sum(x[1] for x in top_ten)
-        img = self.create_chart(top_ten, others, channel)
-        await em.delete()
-        await ctx.message.channel.send(file=discord.File(img, "chart.png"))
+        task = functools.partial(self.create_chart, top_ten, others, channel)
+        task = self.bot.loop.run_in_executor(None, task)
+        try:
+            chart = await asyncio.wait_for(task, timeout=60)
+        except asyncio.TimeoutError:
+            return await ctx.send(
+                "An error occurred while generating this image. Try again later."
+            )
+        try:
+            await em.delete()
+        except discord.NotFound:
+            pass
+        await ctx.message.channel.send(file=discord.File(chart, "chart.png"))