🚸 new example with PIL

.gitignore

@@ -31,4 +31,7 @@ MANIFEST
 .coverage
 
 # IDE Ext's
-.history/
+.history/
+
+# tweet_generator example
+examples/tweet_generator/*.ttf

examples/tweet_generator/tweet_generator.py

@@ -0,0 +1,180 @@
+import re
+import textwrap
+import os
+import sys
+from datetime import datetime
+from PIL import Image, ImageFont, ImageDraw, ImageOps
+from pincer import command, Client
+from pincer.commands import CommandArg, Description
+from pincer.objects import Message, Embed, MessageContext
+
+
+# you need to manually download the font files and put them into the folder
+# ./examples/tweet_generator/ to make the script works using this link:
+# https://fonts.google.com/share?selection.family=Noto%20Sans:wght@400;700
+if not all(font in os.listdir() for font in ["NotoSans-Regular.ttf", "NotoSans-Bold.ttf"]):
+    sys.exit()
+
+
+class Bot(Client):
+    @Client.event
+    async def on_ready(self):
+        print(
+            f"Started client on {self.bot}\n"
+            f"Registered commands: {', '.join(self.chat_commands)}"
+        )
+
+    @command(
+        description="to create fake tweets",
+    )
+    async def twitter(
+        self, ctx: MessageContext, content: CommandArg[str, Description["The content of the message"]]
+    ):
+        await ctx.interaction.ack()
+
+        for text_match, user_id in re.findall(
+            re.compile(r"(<@!(\d+)>)"), content
+        ):
+            content = content.replace(
+                text_match, f"@{await self.get_user(user_id)}"
+            )
+
+        if len(content) > 280:
+            return "A tweet can be at maximum 280 characters long"
+
+        # download the profile picture and convert it into Image object
+        avatar = (await ctx.author.user.get_avatar()).resize((128, 128))
+        avatar = circular_avatar(avatar)
+
+        # create the tweet by pasting the profile picture into a white image
+        tweet = trans_paste(
+            avatar,
+            Image.new("RGBA", (800, 250 + 50 * len(textwrap.wrap(content, 38))), (255, 255, 255)),
+            box=(15, 15),
+        )
+
+        # add the fonts
+        font_normal = ImageFont.truetype("NotoSans-Regular.ttf", 40)
+        font_small = ImageFont.truetype("NotoSans-Regular.ttf", 30)
+        font_bold = ImageFont.truetype("NotoSans-Bold.ttf", 40)
+
+        # write the name and username on the Image
+        draw = ImageDraw.Draw(tweet)
+        draw.text(
+            (180, 20), str(ctx.author.user),
+            fill=(0, 0, 0), font=font_bold
+        )
+        draw.text(
+            (180, 70),
+            f"@{ctx.author.user.username}",
+            fill=(120, 120, 120), font=font_normal,
+        )
+
+        content = add_color_to_mentions(content)
+
+        # write the text
+        tweet = draw_multicolored_text(tweet, content, font_normal)
+
+        # write the footer
+        draw.text(
+            (30, tweet.size[1] - 60),
+            datetime.now().strftime(
+                "%I:%M %p · %d %b. %Y · Twitter for Discord"
+            ),
+            fill=(120, 120, 120),
+            font=font_small,
+        )
+
+        return Message(
+            embeds=[
+                Embed(title="Twitter for Discord", description="").set_image(
+                    url="attachment://image0.png"
+                )
+            ],
+            attachments=[tweet],
+        )
+
+
+def trans_paste(fg_img, bg_img, box=(0, 0)):
+    """
+    https://stackoverflow.com/a/53663233/15485584
+    paste an image into one another
+    """
+    fg_img_trans = Image.new("RGBA", fg_img.size)
+    fg_img_trans = Image.blend(fg_img_trans, fg_img, 1.0)
+    bg_img.paste(fg_img_trans, box, fg_img_trans)
+    return bg_img
+
+
+def circular_avatar(avatar):
+    mask = Image.new("L", (128, 128), 0)
+    draw = ImageDraw.Draw(mask)
+    draw.ellipse((0, 0, 128, 128), fill=255)
+    avatar = ImageOps.fit(avatar, mask.size, centering=(0.5, 0.5))
+    avatar.putalpha(mask)
+    return avatar
+
+
+def add_color_to_mentions(message):
+    """
+    generate a dict to set were the text need to be in different colors.
+    if a word starts with '@' it will be write in blue.
+
+    Parameters
+    ----------
+    message: the text
+
+    Returns
+    -------
+    a list with all colors selected
+    example:
+        [
+            {'color': (0, 0, 0), 'text': 'hello world '},
+            {'color': (0, 154, 234), 'text': '@drawbu'}
+        ]
+
+    """
+    message = textwrap.wrap(message, 38)
+    message = "\n".join(message).split(" ")
+    result = []
+    for word in message:
+        for index, text in enumerate(word.splitlines()):
+
+            text += "\n" if index != len(word.split("\n")) - 1 else " "
+
+            if not result:
+                result.append({"color": (0, 0, 0), "text": text})
+                continue
+
+            if not text.startswith("@"):
+                if result[-1:][0]["color"] == (0, 0, 0):
+                    result[-1:][0]["text"] += text
+                    continue
+
+                result.append({"color": (0, 0, 0), "text": text})
+                continue
+
+            result.append({"color": (0, 154, 234), "text": text})
+    return result
+
+
+def draw_multicolored_text(image, message, font):
+    draw = ImageDraw.Draw(image)
+    x = 30
+    y = 170
+    for text in message:
+        y -= font.getsize(" ")[1]
+        for l_index, line in enumerate(text["text"].splitlines()):
+            if l_index:
+                x = 30
+            y += font.getsize(" ")[1]
+            draw.text((x, y), line, fill=text["color"], font=font)
+            x += font.getsize(line)[0]
+    return image
+
+
+if __name__ == "__main__":
+    # Of course we have to run our client, you can replace the
+    # XXXYOURBOTTOKENHEREXXX with your token, or dynamically get it
+    # through a dotenv/env.
+    Bot("XXXYOURBOTTOKENHEREXXX").run()

pincer/objects/user/user.py

@@ -134,8 +134,9 @@ def premium(self) -> APINullable[PremiumTypes]:
         user their premium type in a usable enum.
         """
         return (
-            MISSING if self.premium_type is MISSING else PremiumTypes(
-                self.premium_type)
+            MISSING
+            if self.premium_type is MISSING
+            else PremiumTypes(self.premium_type)
         )
 
     @property
@@ -179,7 +180,9 @@ async def get_avatar(self, size: int = 512, ext: str = "png") -> Image:
             :class: Image
                 The user's avatar as a Pillow image.
             """
-            async with ClientSession().get(url=self.get_avatar_url()) as resp:
+            async with ClientSession().get(
+                url=self.get_avatar_url(size, ext)
+            ) as resp:
                 avatar = io.BytesIO(await resp.read())
                 return Image.open(avatar).convert("RGBA")
 
@@ -221,9 +224,8 @@ async def get_dm_channel(self) -> channel.Channel:
             construct_client_dict(
                 self._client,
                 await self._http.post(
-                    "/users/@me/channels",
-                    data={"recipient_id": self.id}
-                )
+                    "/users/@me/channels", data={"recipient_id": self.id}
+                ),
             )
         )