✨ reworking a bunch of stuff

gwendolyn_old/funcs/other/nerd_shit.py

@@ -0,0 +1,85 @@
+import os
+
+import requests
+import discord
+import wolframalpha
+from PIL import Image, ImageDraw, ImageFont
+
+class NerdShit():
+    def __init__(self, bot):
+        """Runs misc commands."""
+        self.bot = bot
+
+    async def wolf_search(self,ctx,content):
+        await self.bot.defer(ctx)
+        font = ImageFont.truetype('gwendolyn/resources/fonts/times-new-roman.ttf', 20)
+        self.bot.log("Requesting data")
+        bot = wolframalpha.Client(self.bot.credentials["wolfram_alpha_key"])
+        res = bot.query(content)
+
+        self.bot.log("Processing data")
+        titles = []
+        pods = []
+        if int(res.numpods) > 0:
+            for pod in res.pods:
+                titles += [pod.title]
+                for i, sub in enumerate(pod.subpods):
+                    pods += [sub]
+                    if i > 0:
+                        titles += [""]
+
+            pod_chunks = [pods[x:x+2] for x in range(0, len(pods), 2)]
+            title_chunks = [titles[x:x+2] for x in range(0, len(titles), 2)]
+            await ctx.send(f"Response for \"{content}\"")
+
+            for i, chunk in enumerate(pod_chunks):
+                width = 0
+                for title in title_chunks[i]:
+                    width = max(width,font.getsize(title)[0])
+                height = 5
+                heights = []
+                for count, pod in enumerate(chunk):
+                    heights += [height]
+                    width = max(width,int(pod.img['@width']))
+                    if title_chunks[i][count] == "":
+                        place_for_text = 0
+                    else:
+                        place_for_text = 30
+                    height += int(pod.img["@height"]) + 10 + place_for_text
+
+                width += 10
+                height += 5
+                wolf_image = Image.new("RGB",(width,height),color=(255,255,255))
+
+                for count, pod in enumerate(chunk):
+                    response = requests.get(pod.img["@src"])
+                    file = open("gwendolyn/resources/wolfTemp.png", "wb")
+                    file.write(response.content)
+                    file.close()
+                    old_image = Image.open("gwendolyn/resources/wolfTemp.png")
+                    old_size = old_image.size
+                    if title_chunks[i][count] == "":
+                        place_for_text = 0
+                    else:
+                        place_for_text = 30
+                    new_size = (width,int(old_size[1]+10+place_for_text))
+                    new_image = Image.new("RGB",new_size,color=(255,255,255))
+                    new_image.paste(old_image, (int((int(old_size[0]+10)-old_size[0])/2),int(((new_size[1]-place_for_text)-old_size[1])/2)+place_for_text))
+                    if title_chunks[i][count] != "":
+                        drawer = ImageDraw.Draw(new_image,"RGB")
+                        drawer.text((5,7),title_chunks[i][count],font=font,fill=(150,150,150))
+
+                    wolf_image.paste(new_image,(0,heights[count]))
+                    new_image.close()
+                    old_image.close()
+                    count += 1
+
+                wolf_image.save("gwendolyn/resources/wolf.png")
+                wolf_image.close()
+                await ctx.channel.send(file = discord.File("gwendolyn/resources/wolf.png"))
+
+                os.remove("gwendolyn/resources/wolf.png")
+                os.remove("gwendolyn/resources/wolfTemp.png")
+        else:
+            self.bot.log("No returned data")
+            await ctx.send("Could not find anything relating to your search")