diff options
| -rw-r--r-- | bot/cogs/spookyavatar.py | 2 | ||||
| -rw-r--r-- | bot/resources/bat-clipart.png | bin | 0 -> 12313 bytes | |||
| -rw-r--r-- | bot/resources/bloody-pentagram.png | bin | 0 -> 7006 bytes | |||
| -rw-r--r-- | bot/resources/spookifications.py | 42 |
4 files changed, 43 insertions, 1 deletions
diff --git a/bot/cogs/spookyavatar.py b/bot/cogs/spookyavatar.py index b11c72ee..9a73a019 100644 --- a/bot/cogs/spookyavatar.py +++ b/bot/cogs/spookyavatar.py @@ -39,7 +39,7 @@ class SpookyAvatar: 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.inversion(im) + 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') diff --git a/bot/resources/bat-clipart.png b/bot/resources/bat-clipart.png Binary files differnew file mode 100644 index 00000000..7df26ba9 --- /dev/null +++ b/bot/resources/bat-clipart.png diff --git a/bot/resources/bloody-pentagram.png b/bot/resources/bloody-pentagram.png Binary files differnew file mode 100644 index 00000000..4e6da07a --- /dev/null +++ b/bot/resources/bloody-pentagram.png diff --git a/bot/resources/spookifications.py b/bot/resources/spookifications.py index 880b24e7..43e8b038 100644 --- a/bot/resources/spookifications.py +++ b/bot/resources/spookifications.py @@ -1,5 +1,11 @@ +import logging +from random import choice, randint + +from PIL import Image from PIL import ImageOps +log = logging.getLogger() + def inversion(im): """Inverts an image. @@ -9,3 +15,39 @@ def inversion(im): im = im.convert('RGB') inv = ImageOps.invert(im) return inv + + +def pentagram(im): + """Adds pentagram to image.""" + im = im.convert('RGB') + wt, ht = im.size + penta = Image.open('bot/resources/bloody-pentagram.png') + penta = penta.resize((wt, ht)) + im.paste(penta, (0, 0), penta) + return im + + +def bat(im): + """Adds a bat silhoutte to the image. + + The bat silhoutte is of a size at least one-fifths that of the original + image and may be rotated upto 90 degrees anti-clockwise.""" + im = im.convert('RGB') + wt, ht = im.size + bat = Image.open('bot/resources/bat-clipart.png') + bat_size = randint(wt//5, wt) + rot = randint(0, 90) + bat = bat.resize((bat_size, bat_size)) + bat = bat.rotate(rot) + x = randint(0, wt-bat_size) + y = randint(0, wt-bat_size) + im.paste(bat, (x, y), bat) + return im + + +def get_random_effect(im): + """Randomly selects and applies an effect.""" + effects = [inversion, pentagram, bat] + effect = choice(effects) + log.info("Spookyavatar's chosen effect:" + str(effect)) + return effect(im) |
