tweet_generator.Bot.twitter() - Code Metrics - Inspection of ":children_crossing: new example with PIL" - Pincer-org/Pincer - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Pull Request — main (#241)

unknown

created 2021-11-28 15:55 UTC

tweet_generator.Bot.twitter() B

↳ Parent: tweet_generator

Complexity

Conditions

Size

Total Lines	68
Code Lines	42

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	42
dl	0
loc	68
rs	8.872
c	0
b	0
f	0
cc	3
nop	3

How to fix Long Method

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]["color"] == (0, 0, 0):
                    result[-1]["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()


1			import re
2			import textwrap
3			import os
4			import sys
5			from datetime import datetime
6			from PIL import Image, ImageFont, ImageDraw, ImageOps
7			from pincer import command, Client
8			from pincer.commands import CommandArg, Description
9			from pincer.objects import Message, Embed, MessageContext
10
11
12			# you need to manually download the font files and put them into the folder
13			# ./examples/tweet_generator/ to make the script works using this link:
14			# https://fonts.google.com/share?selection.family=Noto%20Sans:wght@400;700
15			if not all(font in os.listdir() for font in ["NotoSans-Regular.ttf", "NotoSans-Bold.ttf"]):
16			sys.exit()
17
18
19			class Bot(Client):
20			@Client.event
21			async def on_ready(self):
22			print(
23			f"Started client on {self.bot}\n"
24			f"Registered commands: {', '.join(self.chat_commands)}"
25			)
26
27			@command(description="to create fake tweets")
28			async def twitter(
29			self,
30			ctx: MessageContext,
31			content: CommandArg[str, Description["The content of the message"]]
32			):
33			await ctx.interaction.ack()
34
35			for text_match, user_id in re.findall(
36			re.compile(r"(<@!(\d+)>)"), content
37			):
38			content = content.replace(
39			text_match, f"@{await self.get_user(user_id)}"
40			)
41
42			if len(content) > 280:
43			return "A tweet can be at maximum 280 characters long"
44
45			# download the profile picture and convert it into Image object
46			avatar = (await ctx.author.user.get_avatar()).resize((128, 128))
47			avatar = circular_avatar(avatar)
48
49			# create the tweet by pasting the profile picture into a white image
50			tweet = trans_paste(
51			avatar,
52			Image.new("RGBA", (800, 250 + 50 * len(textwrap.wrap(content, 38))), (255, 255, 255)),
53			box=(15, 15),
54			)
55
56			# add the fonts
57			font_normal = ImageFont.truetype("NotoSans-Regular.ttf", 40)
58			font_small = ImageFont.truetype("NotoSans-Regular.ttf", 30)
59			font_bold = ImageFont.truetype("NotoSans-Bold.ttf", 40)
60
61			# write the name and username on the Image
62			draw = ImageDraw.Draw(tweet)
63			draw.text(
64			(180, 20), str(ctx.author.user),
65			fill=(0, 0, 0), font=font_bold
66			)
67			draw.text(
68			(180, 70),
69			f"@{ctx.author.user.username}",
70			fill=(120, 120, 120), font=font_normal,
71			)
72
73			content = add_color_to_mentions(content)
74
75			# write the text
76			tweet = draw_multicolored_text(tweet, content, font_normal)
77
78			# write the footer
79			draw.text(
80			(30, tweet.size[1] - 60),
81			datetime.now().strftime(
82			"%I:%M %p · %d %b. %Y · Twitter for Discord"
83			),
84			fill=(120, 120, 120),
85			font=font_small,
86			)
87
88			return Message(
89			embeds=[
90			Embed(title="Twitter for Discord", description="").set_image(
91			url="attachment://image0.png"
92			)
93			],
94			attachments=[tweet],
95			)
96
97
98			def trans_paste(fg_img, bg_img, box=(0, 0)):
99			"""
100			https://stackoverflow.com/a/53663233/15485584
101			paste an image into one another
102			"""
103			fg_img_trans = Image.new("RGBA", fg_img.size)
104			fg_img_trans = Image.blend(fg_img_trans, fg_img, 1.0)
105			bg_img.paste(fg_img_trans, box, fg_img_trans)
106			return bg_img
107
108
109			def circular_avatar(avatar):
110			mask = Image.new("L", (128, 128), 0)
111			draw = ImageDraw.Draw(mask)
112			draw.ellipse((0, 0, 128, 128), fill=255)
113			avatar = ImageOps.fit(avatar, mask.size, centering=(0.5, 0.5))
114			avatar.putalpha(mask)
115			return avatar
116
117
118			def add_color_to_mentions(message):
119			"""
120			generate a dict to set were the text need to be in different colors.
121			if a word starts with '@' it will be write in blue.
122
123			Parameters
124			----------
125			message: the text
126
127			Returns
128			-------
129			a list with all colors selected
130			example:
131			[
132			{'color': (0, 0, 0), 'text': 'hello world '},
133			{'color': (0, 154, 234), 'text': '@drawbu'}
134			]
135
136			"""
137			message = textwrap.wrap(message, 38)
138			message = "\n".join(message).split(" ")
139			result = []
140			for word in message:
141			for index, text in enumerate(word.splitlines()):
142
143			text += "\n" if index != len(word.split("\n")) - 1 else " "
144
145			if not result:
146			result.append({"color": (0, 0, 0), "text": text})
147			continue
148
149			if not text.startswith("@"):
150			if result[-1]["color"] == (0, 0, 0):
151			result[-1]["text"] += text
152			continue
153
154			result.append({"color": (0, 0, 0), "text": text})
155			continue
156
157			result.append({"color": (0, 154, 234), "text": text})
158			return result
159
160
161			def draw_multicolored_text(image, message, font):
162			draw = ImageDraw.Draw(image)
163			x = 30
164			y = 170
165			for text in message:
166			y -= font.getsize(" ")[1]
167			for l_index, line in enumerate(text["text"].splitlines()):
168			if l_index:
169			x = 30
170			y += font.getsize(" ")[1]
171			draw.text((x, y), line, fill=text["color"], font=font)
172			x += font.getsize(line)[0]
173			return image
174
175
176			if __name__ == "__main__":
177			# Of course we have to run our client, you can replace the
178			# XXXYOURBOTTOKENHEREXXX with your token, or dynamically get it
179			# through a dotenv/env.
180			Bot("XXXYOURBOTTOKENHEREXXX").run()
181

Pincer-org / Pincer

Pull Request — main (#241)

tweet_generator.Bot.twitter() B

Complexity

Size

Duplication

Importance

How to fix Long Method

Long Method

Duplication Side-by-Side

Filter issues like