yvlu 支持回复贴纸\图片生成语录

list.json

@@ -702,7 +702,7 @@
         },
         {
             "name": "yvlu",
-            "version": "1.0",
+            "version": "1.1",
             "section": "chat",
             "maintainer": "xtaodada",
             "size": "4.8 kb",

yvlu.py

@@ -1,8 +1,12 @@
+from io import BytesIO
+from math import floor
 from os.path import exists
 from os import makedirs, remove
 from PIL import Image, ImageFont, ImageDraw
 from requests import get
 from telethon.tl.functions.users import GetFullUserRequest
+from telethon.tl.types import MessageMediaPhoto
+from pagermaid import bot
 from pagermaid.listener import listener
 from pagermaid.utils import alias_command
 
@@ -16,9 +20,9 @@ def cut(obj, sec):
 
 
 def yv_lu_process_image(name, text, photo, path):
-    if len(name) > 27:
-        name = name[:27] + '...'
-    text = cut(text, 22)
+    if len(name) > 16:
+        name = name[:16] + '...'
+    text = cut(text, 17)
     # 用户不存在头像时
     if not photo:
         photo = Image.open(f'{path}4.png')
@@ -36,8 +40,8 @@ def yv_lu_process_image(name, text, photo, path):
     else:
         photo = Image.open(f'{path}{photo}')
     # 读取图片
-    img1, img2, img3, mask = Image.open(f'{path}1.png'), Image.open(f'{path}2.png'), \
-                             Image.open(f'{path}3.png'), Image.open(f'{path}mask.png')
+    img1, img2, img3, mask = Image.open(f'{path}p1.png'), Image.open(f'{path}p2.png'), \
+                             Image.open(f'{path}p3.png'), Image.open(f'{path}mask.png')
     size1, size2, size3 = img1.size, img2.size, img3.size
     photo_size = photo.size
     mask_size = mask.size
@@ -55,7 +59,7 @@ def yv_lu_process_image(name, text, photo, path):
     draw = ImageDraw.Draw(img1)
     draw.text((60, 10), name, (255, 110, 164), font(path, 18))
     for i in range(len(text)):
-        temp = Image.open(f'{path}2.png')
+        temp = Image.open(f'{path}p2.png')
         draw = ImageDraw.Draw(temp)
         draw.text((60, 0), text[i], (255, 255, 255), font(path, 18))
         result.paste(temp, (0, size1[1] + size2[1] * i))
@@ -69,6 +73,43 @@ def yv_lu_process_image(name, text, photo, path):
     result.save(f'{path}result.png')
 
 
+def yv_lu_process_sticker(name, photo, sticker, path):
+    # 用户不存在头像时
+    if not photo:
+        photo = Image.open(f'{path}4.png')
+        # 对图片写字
+        draw = ImageDraw.Draw(photo)
+        # 计算使用该字体占据的空间
+        # 返回一个 tuple (width, height)
+        # 分别代表这行字占据的宽和高
+        text_width = font(path, 60).getsize(name[0])
+        if name[0].isalpha():
+            text_coordinate = int((photo.size[0] - text_width[0]) / 2), int((photo.size[1] - text_width[1]) / 2) - 10
+        else:
+            text_coordinate = int((photo.size[0] - text_width[0]) / 2), int((photo.size[1] - text_width[1]) / 2)
+        draw.text(text_coordinate, name[0], (255, 110, 164), font(path, 60))
+    else:
+        photo = Image.open(f'{path}{photo}')
+    # 读取图像
+    sticker = Image.open(f'{sticker}')
+    sticker = sticker.resize((400, 400))
+    # 创建空图片
+    result = Image.new('RGBA', (512, 512))
+    # 处理头像
+    mask = Image.open(f'{path}mask1.png')
+    photo_size = photo.size
+    mask_size = mask.size
+    scale = photo_size[1] / mask_size[1]
+    photo = photo.resize((int(photo_size[0] / scale), int(photo_size[1] / scale)), Image.LANCZOS)
+    mask1 = Image.new('RGBA', mask_size)
+    mask1.paste(photo, mask=mask)
+    # 粘贴图像
+    result.paste(mask1, (10, 350))
+    result.paste(sticker, (130, 50))
+    # 保存图像
+    result.save(f'{path}result.png')
+
+
 @listener(is_plugin=True, outgoing=True, command=alias_command("yvlu"),
           description="将回复的消息转换成语录")
 async def yv_lu(context):
@@ -78,24 +119,34 @@ async def yv_lu(context):
     if not exists('plugins/yvlu/'):
         makedirs('plugins/yvlu/')
     await context.edit('处理中。。。')
+    # 下载资源文件
     for num in range(1, 5):
-        if not exists('plugins/yvlu/' + str(num) + '.png'):
-            re = get('https://raw.githubusercontent.com/Xtao-Labs/PagerMaid_Plugins/master/yvlu/' + str(num) + '.png')
-            with open('plugins/yvlu/' + str(num) + '.png', 'wb') as bg:
+        if not exists('plugins/yvlu/p' + str(num) + '.png'):
+            re = get('https://raw.githubusercontent.com/Xtao-Labs/PagerMaid_Plugins/master/yvlu/p' + str(num) + '.png')
+            with open('plugins/yvlu/p' + str(num) + '.png', 'wb') as bg:
                 bg.write(re.content)
     if not exists('plugins/yvlu/mask.png'):
         re = get('https://raw.githubusercontent.com/Xtao-Labs/PagerMaid_Plugins/master/yvlu/mask.png')
         with open('plugins/yvlu/mask.png', 'wb') as bg:
             bg.write(re.content)
+    if not exists('plugins/yvlu/mask1.png'):
+        re = get('https://raw.githubusercontent.com/Xtao-Labs/PagerMaid_Plugins/master/yvlu/mask1.png')
+        with open('plugins/yvlu/mask1.png', 'wb') as bg:
+            bg.write(re.content)
     if not exists('plugins/yvlu/ZhuZiAWan-2.ttc'):
         await context.edit('下载字体中。。。')
         re = get('https://raw.githubusercontent.com/Xtao-Labs/Telegram_PaimonBot/master/assets/fonts/ZhuZiAWan-2.ttc')
         with open('plugins/yvlu/ZhuZiAWan-2.ttc', 'wb') as bg:
             bg.write(re.content)
+    # 获取回复信息
     reply_message = await context.get_reply_message()
+    if not reply_message:
+        await context.edit('你需要回复一条消息。')
+        return
     user_id = reply_message.sender_id
     target_user = await context.client(GetFullUserRequest(user_id))
-    await context.client.download_profile_photo(
+    # 下载头像
+    await bot.download_profile_photo(
         target_user.user.id,
         "plugins/yvlu/" + str(target_user.user.id) + ".jpg",
         download_big=True
@@ -103,18 +154,65 @@ async def yv_lu(context):
     name = target_user.user.first_name
     if target_user.user.last_name:
         name += f' {target_user.user.last_name}'
+    # 判断是否为贴纸
+    file_name = None
+    if reply_message and reply_message.media:
+        if isinstance(reply_message.media, MessageMediaPhoto):
+            file_name = 'plugins/yvlu/sticker.jpg'
+            await bot.download_media(reply_message.photo, file_name)
+        elif "image" in reply_message.media.document.mime_type.split('/'):
+            file_name = 'plugins/yvlu/sticker.jpg'
+            await bot.download_file(reply_message.media.document, file_name)
+        else:
+            await context.edit('不支持的文件类型。')
+            return
     if exists("plugins/yvlu/" + str(target_user.user.id) + ".jpg"):
-        yv_lu_process_image(name, reply_message.text, f"{target_user.user.id}.jpg", 'plugins/yvlu/')
+        if file_name:
+            yv_lu_process_sticker(name, f"{target_user.user.id}.jpg", file_name, 'plugins/yvlu/')
+            remove(file_name)
+        else:
+            yv_lu_process_image(name, reply_message.message, f"{target_user.user.id}.jpg", 'plugins/yvlu/')
         remove("plugins/yvlu/" + str(target_user.user.id) + ".jpg")
     else:
-        yv_lu_process_image(name, reply_message.text, None, 'plugins/yvlu/')
-    force_document = True
-    if context.arguments:
-        force_document = False
+        if file_name:
+            yv_lu_process_sticker(name, None, file_name, 'plugins/yvlu/')
+            remove(file_name)
+        else:
+            yv_lu_process_image(name, reply_message.message, None, 'plugins/yvlu/')
+    # 转换为贴纸
+    image = await resize_image('plugins/yvlu/result.png')
+    file = BytesIO()
+    file.name = "sticker.webp"
+    image.save(file, "WEBP")
+    file.seek(0)
     await context.client.send_file(
         context.chat_id,
-        'plugins/yvlu/result.png',
-        force_document=force_document,
+        file,
+        force_document=False,
         reply_to=context.message.reply_to_msg_id
     )
     await context.delete()
+
+
+async def resize_image(photo):
+    image = Image.open(photo)
+    maxsize = (512, 512)
+    if (image.width and image.height) < 512:
+        size1 = image.width
+        size2 = image.height
+        if image.width > image.height:
+            scale = 512 / size1
+            size1new = 512
+            size2new = size2 * scale
+        else:
+            scale = 512 / size2
+            size1new = size1 * scale
+            size2new = 512
+        size1new = floor(size1new)
+        size2new = floor(size2new)
+        size_new = (size1new, size2new)
+        image = image.resize(size_new)
+    else:
+        image.thumbnail(maxsize)
+
+    return image