diff --git a/Shitpost.py b/Shitpost.py index b963f18..d0e6842 100644 --- a/Shitpost.py +++ b/Shitpost.py @@ -245,6 +245,13 @@ class Shitpost(commands.Cog): if 'nee' in answer: reactie += 'Nee helaas..' yield from ctx.channel.send(ctx.author.mention + ' ' + reactie) + + def get_inspiration(self): + inspiration = requests.get(url='http://inspirobot.me/api?generate=true') + if inspiration: + return inspiration.text + return None + @commands.command(pass_context=True, hidden=False) @asyncio.coroutine @@ -253,11 +260,10 @@ class Shitpost(commands.Cog): Bijvoorbeeld: !inspire """ - r = requests.get(url='http://inspirobot.me/api?generate=true') - if r.status_code == 200: - url = r.text + inspo = self.get_inspiration() + if inspo: em = discord.Embed(title='Uw wijze woorden:', type='photo') - em.set_image(url=url) + em.set_image(url=inspo) try: yield from ctx.message.delete() except: diff --git a/tests/shitpost.py b/tests/shitpost.py index dd63da6..aa53cd0 100644 --- a/tests/shitpost.py +++ b/tests/shitpost.py @@ -151,3 +151,8 @@ class TestShitpost(unittest.TestCase): tegeltje = self.cog.get_tegeltje("Man who put penis in peanut butter is fucking nuts", "Confucius") self.assertIsInstance(eriku, discord.Embed) self.assertIsInstance(tegeltje, discord.Embed) + + def test_inspire(self): + inspo = self.cog.get_inspiration() + self.assertIn("https://", inspo) + self.assertIn(".jpg", inspo) \ No newline at end of file