mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-16 08:59:00 +00:00
yvlu 修复问题 fix #61
This commit is contained in:
parent
f8489222de
commit
507450607e
GPG Key ID:
4CBB3F4FA8C85659
253
yvlu/main.py
253
yvlu/main.py
@ -1,118 +1,15 @@
|
|||||||
import contextlib
|
import contextlib
|
||||||
from asyncio import sleep
|
from asyncio import sleep
|
||||||
from doctest import OutputChecker
|
|
||||||
from io import BytesIO
|
from pyrogram.errors import Flood
|
||||||
from math import floor
|
|
||||||
from os.path import exists
|
|
||||||
from os import makedirs, remove
|
|
||||||
from PIL import Image, ImageFont, ImageDraw
|
|
||||||
from asyncio import TimeoutError
|
|
||||||
from pagermaid import bot
|
|
||||||
from pagermaid.listener import listener
|
from pagermaid.listener import listener
|
||||||
from pagermaid.single_utils import Message
|
from pagermaid.enums import Client, Message
|
||||||
|
|
||||||
|
|
||||||
def font(path, size):
|
@listener(command="yvlu",
|
||||||
return ImageFont.truetype(f'{path}ZhuZiAWan-2.ttc', size=size, encoding="utf-8")
|
|
||||||
|
|
||||||
|
|
||||||
def cut(obj, sec):
|
|
||||||
return [obj[i:i + sec] for i in range(0, len(obj), sec)]
|
|
||||||
|
|
||||||
|
|
||||||
def yv_lu_process_image(name, text, photo, path):
|
|
||||||
if len(name) > 16:
|
|
||||||
name = f'{name[:16]}...'
|
|
||||||
text = cut(text, 17)
|
|
||||||
# 用户不存在头像时
|
|
||||||
if not photo:
|
|
||||||
photo = Image.open(f'{path}p4.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}')
|
|
||||||
# 读取图片
|
|
||||||
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
|
|
||||||
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 = Image.new(img1.mode, (size1[0], size1[1] + size2[1] * len(text) + size3[1]))
|
|
||||||
|
|
||||||
# 读取粘贴位置
|
|
||||||
loc1, loc3, loc4 = (0, 0), (0, size1[1] + size2[1] * len(text)), (6, size1[1] + size2[1] * len(text) - 23)
|
|
||||||
|
|
||||||
# 对图片写字
|
|
||||||
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}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))
|
|
||||||
|
|
||||||
# 粘贴图片
|
|
||||||
result.paste(img1, loc1)
|
|
||||||
result.paste(img3, loc3)
|
|
||||||
result.paste(mask1, loc4)
|
|
||||||
|
|
||||||
# 保存图片
|
|
||||||
result.save(f'{path}result.png')
|
|
||||||
|
|
||||||
|
|
||||||
async def yv_lu_process_sticker(name, photo, sticker, path):
|
|
||||||
# 用户不存在头像时
|
|
||||||
if not photo:
|
|
||||||
photo = Image.open(f'{path}p4.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 = await resize_image(sticker, 400)
|
|
||||||
# 创建空图片
|
|
||||||
result = Image.new('RGBA', (512, 400))
|
|
||||||
# 处理头像
|
|
||||||
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, (0, 300))
|
|
||||||
result.paste(sticker, (112, 400 - sticker.size[1]))
|
|
||||||
# 保存图像
|
|
||||||
result.save(f'{path}result.png')
|
|
||||||
|
|
||||||
|
|
||||||
@listener(is_plugin=True, outgoing=True, command="yvlu",
|
|
||||||
description="将回复的消息或者输入的字符串转换成语录")
|
description="将回复的消息或者输入的字符串转换成语录")
|
||||||
async def yv_lu(message: Message):
|
async def yv_lu(bot: Client, message: Message):
|
||||||
bot_username = "PagerMaid_QuotLy_bot"
|
bot_username = "PagerMaid_QuotLy_bot"
|
||||||
if message.reply_to_message:
|
if message.reply_to_message:
|
||||||
reply = message.reply_to_message
|
reply = message.reply_to_message
|
||||||
@ -121,128 +18,22 @@ async def yv_lu(message: Message):
|
|||||||
else:
|
else:
|
||||||
return await message.edit('你需要回复一条消息或者输入一串字符。')
|
return await message.edit('你需要回复一条消息或者输入一串字符。')
|
||||||
with contextlib.suppress(Exception):
|
with contextlib.suppress(Exception):
|
||||||
await message.bot.unblock_user(bot_username)
|
await bot.unblock_user(bot_username)
|
||||||
async with message.bot.conversation(bot_username) as conv:
|
async with bot.conversation(bot_username) as conv:
|
||||||
await reply.forward(bot_username)
|
await reply.forward(bot_username)
|
||||||
|
await sleep(.1)
|
||||||
chat_response = await conv.get_response()
|
chat_response = await conv.get_response()
|
||||||
await conv.mark_as_read()
|
await conv.mark_as_read()
|
||||||
await chat_response.copy(
|
try:
|
||||||
message.chat.id,
|
await chat_response.copy(
|
||||||
reply_to_message_id=message.reply_to_message_id if message.reply_to_message else None)
|
message.chat.id,
|
||||||
await message.delete()
|
reply_to_message_id=message.reply_to_message_id)
|
||||||
|
except Flood as e:
|
||||||
|
await sleep(e.value + 1)
|
||||||
# @listener(is_plugin=True, outgoing=True, command=alias_command("yvlu_"),
|
with contextlib.suppress(Exception):
|
||||||
# description="将回复的消息转换成语录")
|
await chat_response.copy(
|
||||||
# async def yv_lu_(context):
|
message.chat.id,
|
||||||
# if not context.reply_to_msg_id:
|
reply_to_message_id=message.reply_to_message_id)
|
||||||
# await context.edit('你需要回复一条消息。')
|
except Exception:
|
||||||
# return
|
pass
|
||||||
# if not exists('plugins/yvlu/'):
|
await message.safe_delete()
|
||||||
# makedirs('plugins/yvlu/')
|
|
||||||
# await context.edit('处理中。。。')
|
|
||||||
# # 下载资源文件
|
|
||||||
# for num in range(1, 5):
|
|
||||||
# if not exists('plugins/yvlu/p' + str(num) + '.png'):
|
|
||||||
# re = get('https://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/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://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/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://gitlab.com/Xtao-Labs/PagerMaid_Plugins/-/raw/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://gitlab.com/Xtao-Labs/Telegram_PaimonBot/-/raw/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 bot.download_profile_photo(
|
|
||||||
# target_user.user.id,
|
|
||||||
# "plugins/yvlu/" + str(target_user.user.id) + ".jpg",
|
|
||||||
# download_big=True
|
|
||||||
# )
|
|
||||||
# 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 isinstance(reply_message.media, MessageMediaWebPage):
|
|
||||||
# return await context.edit('不支持的文件类型。')
|
|
||||||
# elif isinstance(reply_message.media, MessageMediaUnsupported):
|
|
||||||
# return await context.edit('不支持的文件类型。')
|
|
||||||
# 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"):
|
|
||||||
# if file_name:
|
|
||||||
# await 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:
|
|
||||||
# if file_name:
|
|
||||||
# await 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', 512)
|
|
||||||
# file = BytesIO()
|
|
||||||
# file.name = "sticker.webp"
|
|
||||||
# try:
|
|
||||||
# image.save(file, "WEBP")
|
|
||||||
# except KeyError:
|
|
||||||
# await context.delete()
|
|
||||||
# return
|
|
||||||
# file.seek(0)
|
|
||||||
# await context.client.send_file(
|
|
||||||
# context.chat_id,
|
|
||||||
# file,
|
|
||||||
# force_document=False,
|
|
||||||
# reply_to=context.message.reply_to_msg_id
|
|
||||||
# )
|
|
||||||
# await context.delete()
|
|
||||||
|
|
||||||
|
|
||||||
async def resize_image(photo, num):
|
|
||||||
image = Image.open(photo)
|
|
||||||
maxsize = (num, num)
|
|
||||||
if (image.width and image.height) < num:
|
|
||||||
size1 = image.width
|
|
||||||
size2 = image.height
|
|
||||||
if image.width > image.height:
|
|
||||||
scale = num / size1
|
|
||||||
size1new = num
|
|
||||||
size2new = size2 * scale
|
|
||||||
else:
|
|
||||||
scale = num / size2
|
|
||||||
size1new = size1 * scale
|
|
||||||
size2new = num
|
|
||||||
size1new = floor(size1new)
|
|
||||||
size2new = floor(size2new)
|
|
||||||
size_new = (size1new, size2new)
|
|
||||||
image = image.resize(size_new)
|
|
||||||
else:
|
|
||||||
image.thumbnail(maxsize)
|
|
||||||
|
|
||||||
return image
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user