Text_Sticker.py - main - codwiz/remodules

Text_Sticker.py
90 строк3 KB
CodWiz
formatter of the code via ruff && new modules globalrestrict && fixed modules && translate en and ru
17 мар 2024, 00:02
17 мар 2024, 00:027ae6967
# ---------------------------------------------------------------------------------
# Name: Text in sticker
# Description: Text in sticker
# Author: @hikka_mods
# Commands:
# .st <hex color> [text]
# ---------------------------------------------------------------------------------

# 🔒    Licensed under the GNU AGPLv3
# 🌐 https://www.gnu.org/licenses/agpl-3.0.html

# meta developer: @hikka_mods
# scope: Text in sticker
# scope: Text in sticker 0.0.1
# requires: requests
# ---------------------------------------------------------------------------------
import io
from textwrap import wrap

import requests
from PIL import Image, ImageColor, ImageDraw
from PIL import ImageFont

from .. import loader, utils


@loader.tds
class TextinstickerMod(loader.Module):
    """Text to sticker"""

    strings = {
        "name": "Text in sticker",
        "error": "white st <color name> [text]",
    }

    strings_ru = {
        "error": "Укажите .st <color name> [text]",
    }

    def __init__(self):
        self.config = loader.ModuleConfig(
            loader.ConfigValue(
                "font",
                "https://github.com/CodWize/ReModules/blob/main/assets/Samson.ttf?raw=true",
                lambda: "add a link to the font you want",
            )
        )

    @loader.owner
    async def stcmd(self, message):
        """<color name> [text]"""
        await message.delete()
        text = utils.get_args_raw(message)
        reply = await message.get_reply_message()
        if not text:
            if not reply:
                text = self.strings("error")
            elif not reply.message:
                text = self.strings("error")
            else:
                text = reply.raw_text
        color_name = text.split(" ", 1)[0].lower()
        color = None
        if len(text.split(" ", 1)) > 1:
            text = text.split(" ", 1)[1]
        else:
            if reply and reply.message:
                text = reply.raw_text
        try:
            color = ImageColor.getrgb(color_name)
        except ValueError:
            color = (255, 255, 255)
        txt = []
        for line in text.split("\n"):
            txt.append("\n".join(wrap(line, 30)))
        text = "\n".join(txt)
        bytes_font = requests.get(self.config["font"]).content
        font = io.BytesIO(bytes_font)
        font = ImageFont.truetype(font, 100)
        image = Image.new("RGBA", (1, 1), (0, 0, 0, 0))
        draw = ImageDraw.Draw(image)
        w, h = draw.multiline_textsize(text=text, font=font)
        image = Image.new("RGBA", (w + 100, h + 100), (0, 0, 0, 0))
        draw = ImageDraw.Draw(image)
        draw.multiline_text((50, 50), text=text, font=font, fill=color, align="center")
        output = io.BytesIO()
        output.name = f"{color_name}.webp"
        image.save(output, "WEBP")
        output.seek(0)
        await self.client.send_file(message.to_id, output, reply_to=reply)