diff --git a/General.py b/General.py index d433f05..9486453 100644 --- a/General.py +++ b/General.py @@ -28,6 +28,14 @@ class General(commands.Cog): def __check(self, ctx): #Todo: Is dit het porno kanaal? return True + def get_emoji_embed(self, tag: str): + emoji = re.search(".*\<\:([A-z]*)\:(\d*)\>.*", tag) + embed = discord.Embed(title=emoji.group(1), type='photo') + hugemoji = 'https://cdn.discordapp.com/emojis/' + emoji.group(2) + '.png' + embed.set_image(url=hugemoji) + return embed + + @commands.command() @asyncio.coroutine def huge(self, ctx, *, arg): @@ -35,10 +43,7 @@ class General(commands.Cog): Bijvoorbeeld: !huge :ancilla: """ - id = re.search("\<\:([A-z]*)\:(\d*)\>", arg) - hugemoji = 'https://cdn.discordapp.com/emojis/' + id.group(2) + '.png' - em = discord.Embed(title=id.group(1), type='photo') - em.set_image(url=hugemoji) + em = self.get_emoji_embed(arg) try: yield from ctx.message.delete() except: diff --git a/tests.py b/tests.py index e57ed51..a958bdb 100644 --- a/tests.py +++ b/tests.py @@ -4,6 +4,7 @@ import random import requests import json import dotenv +import re import discord from discord.ext import commands @@ -42,3 +43,7 @@ class TestGeneral(unittest.TestCase): post = self.cog.get_kopieerpasta() self.assertIsInstance(post.url, str) + def test_huge(self): + embed = self.cog.get_emoji_embed("<:test:123456789>") + self.assertIsInstance(embed, discord.Embed) +