diff options
Diffstat (limited to 'bot/cogs')
| -rw-r--r-- | bot/cogs/hacktober/candy_collection.py (renamed from bot/cogs/candy_collection.py) | 0 | ||||
| -rw-r--r-- | bot/cogs/hacktober/spookyavatar.py | 52 | 
2 files changed, 52 insertions, 0 deletions
| diff --git a/bot/cogs/candy_collection.py b/bot/cogs/hacktober/candy_collection.py index 59eadd93..59eadd93 100644 --- a/bot/cogs/candy_collection.py +++ b/bot/cogs/hacktober/candy_collection.py diff --git a/bot/cogs/hacktober/spookyavatar.py b/bot/cogs/hacktober/spookyavatar.py new file mode 100644 index 00000000..694eb317 --- /dev/null +++ b/bot/cogs/hacktober/spookyavatar.py @@ -0,0 +1,52 @@ +from io import BytesIO +import os + +import aiohttp +import discord +from discord.ext import commands +from PIL import Image + +from bot.utils import spookifications + + +class SpookyAvatar: + +    """ +    A cog that spookifies an avatar. +    """ + +    def __init__(self, bot): +        self.bot = bot + +    async def get(self, url): +        """ +        Returns the contents of the supplied url. +        """ +        async with aiohttp.ClientSession() as session: +            async with session.get(url) as resp: +                return await resp.read() + +    @commands.command(name='savatar', aliases=['spookyavatar', 'spookify'], +                      brief='Spookify an user\'s avatar.') +    async def spooky_avatar(self, ctx, user: discord.Member = None): +        """ +        A command to print the user's spookified avatar. +        """ +        if user is None: +            user = ctx.message.author + +        embed = discord.Embed(colour=0xFF0000) +        embed.title = "Is this you or am I just really paranoid?" +        embed.set_author(name=str(user.name), icon_url=user.avatar_url) +        resp = await self.get(user.avatar_url) +        im = Image.open(BytesIO(resp)) +        modified_im = spookifications.get_random_effect(im) +        modified_im.save(str(ctx.message.id)+'.png') +        f = discord.File(str(ctx.message.id)+'.png') +        embed.set_image(url='attachment://'+str(ctx.message.id)+'.png') +        await ctx.send(file=f, embed=embed) +        os.remove(str(ctx.message.id)+'.png') + + +def setup(bot): +    bot.add_cog(SpookyAvatar(bot)) | 
