Merge pull request #174 from crimsobot/aeroplane-cover

In the Aeroplane Over the Sea cover maker; GIF processing in DMs

crimsobot/cogs/image.py

@@ -9,7 +9,7 @@
 from discord.ext import commands
 
 from crimsobot.bot import CrimsoBOT
-from crimsobot.data.img import AENIMA, CAPTION_RULES, CURRENTS, IMAGE_RULES, LATERALUS, URL_CONTAINS
+from crimsobot.data.img import AENIMA, AEROPLANE, CAPTION_RULES, CURRENTS, IMAGE_RULES, LATERALUS, URL_CONTAINS
 from crimsobot.exceptions import BadCaption, NoImageFound
 from crimsobot.utils import image as imagetools, tools as c
 
@@ -174,6 +174,18 @@ async def aenima(self, ctx: commands.Context, image: Optional[str] = None) -> No
 
         await self.get_image_and_embed(ctx, image, effect, None, title)
 
+    @commands.command()
+    async def aeroplane(self, ctx: commands.Context, image: Optional[str] = None) -> None:
+        """Make a new cover for ITAOTS."""
+
+        effect = 'aeroplane'
+        title = random.choice(AEROPLANE)
+
+        if image is None:
+            image = await self.get_previous_image(ctx)  # will be a URL
+
+        await self.get_image_and_embed(ctx, image, effect, None, title)
+
     @commands.command(hidden=True)
     @commands.cooldown(1, 4 * 60 * 60, commands.BucketType.user)
     @shared_max_concurrency(eface_bucket)

crimsobot/data/img/__init__.py

@@ -22,6 +22,7 @@
 EIMG_WIDTH = _EIMG_RULES['width']
 
 AENIMA = _ruleset['aenima']
+AEROPLANE = _ruleset['aeroplane']
 CURRENTS = _ruleset['currents']
 LATERALUS = _ruleset['lateralus']
 

crimsobot/data/img/rules.yaml

@@ -87,6 +87,34 @@ aenima:
   - Dreaming of that fix again
   - So good to see you once again
   - Blue as our new second sun
+aeroplane:
+  - When you were young you were the king of carrot flowers
+  - And from above you how I sank into your soul
+  - Each one a little more than he would dare to try
+  - I love you, Jesus Christ
+  - I will float until I learn how to swim
+  - Mouths open wide and spitting snow
+  - What a beautiful face I have found in this place
+  - Soft and sweet
+  - Count every beautiful thing we can see
+  - Two-headed boy all floating in glass
+  - I am listening to hear where you are
+  - Catching signals that sound in the dark
+  - Playing pianos filled with flames
+  - The Earth looks better from a star
+  - He didn't mean to make you cry
+  - Cars careening from the clouds
+  - Sweet communist, the communist daughter
+  - Wanting something warm and moving
+  - Oh, comely
+  - With some pretty, bright and bubbly terrible scene
+  - In your memory, you're drunk on your awe to me
+  - She was born in a bottle-rocket in 1929
+  - All drenched in milk and holy water pouring from the sky
+  - I know you live within me, feel as you fly
+  - Blister, please, with those wings in your spine
+  - Secret songs that you keep wrapped in boxes so tight
+  - Your brains fell out through your teeth
 currents:
   - All this running around
   - Just let it happen

crimsobot/utils/image.py

@@ -415,7 +415,7 @@ def make_aenima_img(img: Image.Image, arg: None) -> Image.Image:
     width, height = img.size
 
     # 2. paste over white bg
-    bg = Image.new('RGBA', (500, 500), (255, 255, 255, 255))
+    bg = Image.new('RGBA', (500, 500), (240, 230, 190, 255))
     position = int(250 - height/2)
     bg.paste(img, (163, position), img)
 
@@ -426,6 +426,35 @@ def make_aenima_img(img: Image.Image, arg: None) -> Image.Image:
     return bg
 
 
+def make_aeroplane_img(img: Image.Image, arg: None) -> Image.Image:
+    # 1. determine user image size, resize to fit in its place
+    width, height = img.size
+
+    # resize to hole opening (141 x 160)
+    if height > width:
+        ratio = width / 141
+    else:
+        ratio = height / 160
+
+    img = img.resize((int(width / ratio), int(height / ratio)), resample=Image.BICUBIC)
+
+    # get new size
+    width, height = img.size
+
+    # 2. calculate position of image, centered over hole
+    corner = 406 - width // 2, 147 - height // 2
+
+    # 3. paste over cream white bg
+    bg = Image.new('RGBA', (600, 600), (240, 230, 190, 255))
+    bg.paste(img, corner, img)
+
+    # 4. paste cover over result
+    with Image.open(c.clib_path_join('img', 'aeroplane_cover.png')) as cover:
+        bg.alpha_composite(cover, (0, 0))
+
+    return bg
+
+
 def make_captioned_img(img: Image.Image, caption_list: List[str]) -> Image.Image:
     """Captions an image!"""
     # 1. determine image size, resize to standardize text addition
@@ -690,6 +719,7 @@ def process_lower_level(img: Image.Image, effect: str, arg: int) -> BytesIO:
         function_dict: Mapping[str, Callable] = {
             'acid': make_acid_img,
             'aenima': make_aenima_img,
+            'aeroplane': make_aeroplane_img,
             'caption': make_captioned_img,
             'currents': make_currents_img,
             'lateralus': make_lateralus_img,
@@ -752,9 +782,8 @@ async def process_image(ctx: Context, image: Optional[str], effect: str, arg: Op
                 return None, None
 
             else:
-                # debit the user, credit the bot
+                # debit the user
                 await crimsogames.win(ctx.author, -cost)
-                await crimsogames.win(ctx.guild.me, cost)
                 new_bal = await crimsogames.check_balance(ctx.author)
 
             # this embed will keep user updated on processing status; will be edited below as it progresses