mirror of
https://github.com/TeamPGM/PagerMaid_Plugins.git
synced 2024-11-22 01:15:34 +00:00
✨⚡️ pic2sticker 图片转贴纸
This commit is contained in:
parent
9fc4b989e8
commit
04f94ca8e9
14
list.json
14
list.json
@ -332,13 +332,13 @@
|
||||
},
|
||||
{
|
||||
"name": "stickertopic",
|
||||
"version": "1.11",
|
||||
"version": "1.111",
|
||||
"section": "chat",
|
||||
"maintainer": "TNTcraftHIM",
|
||||
"size": "3.5 kb",
|
||||
"supported": true,
|
||||
"des-short": "贴纸转图片",
|
||||
"des": "把别人发的贴纸转换成图片（只支持静态）。命令：pic。"
|
||||
"des": "把别人发的贴纸转换成图片（只支持静态）。命令：stickertopic 。"
|
||||
},
|
||||
{
|
||||
"name": "vip",
|
||||
@ -549,6 +549,16 @@
|
||||
"supported": true,
|
||||
"des-short": "删除烦人的贴纸",
|
||||
"des": "删除最近 50 条消息中的 sticker 。"
|
||||
},
|
||||
{
|
||||
"name": "pic2sticker",
|
||||
"version": "1.0",
|
||||
"section": "chat",
|
||||
"maintainer": "xtaodada",
|
||||
"size": "4.7 kb",
|
||||
"supported": true,
|
||||
"des-short": "图片转贴纸",
|
||||
"des": "把别人发的图片转换成贴纸（只支持静态）。命令：pic2sticker 。"
|
||||
}
|
||||
]
|
||||
}
|
||||
|
||||
151
pic2sticker.py
Normal file
151
pic2sticker.py
Normal file
@ -0,0 +1,151 @@
|
||||
""" PagerMaid module to handle sticker collection. """
|
||||
|
||||
from io import BytesIO
|
||||
from telethon.tl.types import DocumentAttributeFilename, MessageMediaPhoto
|
||||
from PIL import Image, ImageOps
|
||||
from math import floor
|
||||
from pagermaid import bot, redis, redis_status
|
||||
from pagermaid.listener import listener
|
||||
from pagermaid.utils import lang, alias_command
|
||||
|
||||
|
||||
@listener(is_plugin=False, outgoing=True, command=alias_command("pic2sticker"),
|
||||
description='将图片转换为贴纸',
|
||||
parameters="<round>")
|
||||
async def pic2sticker(context):
|
||||
""" Fetches images and send it as sticker. """
|
||||
pic_round = False
|
||||
if len(context.parameter) >= 1:
|
||||
pic_round = True
|
||||
|
||||
if redis_status():
|
||||
if redis.get("sticker.round"):
|
||||
pic_round = True
|
||||
|
||||
message = await context.get_reply_message()
|
||||
if message and message.media:
|
||||
if isinstance(message.media, MessageMediaPhoto):
|
||||
photo = BytesIO()
|
||||
photo = await bot.download_media(message.photo, photo)
|
||||
elif "image" in message.media.document.mime_type.split('/'):
|
||||
photo = BytesIO()
|
||||
try:
|
||||
await context.edit(lang('sticker_downloading'))
|
||||
except:
|
||||
pass
|
||||
await bot.download_file(message.media.document, photo)
|
||||
if (DocumentAttributeFilename(file_name='sticker.webp') in
|
||||
message.media.document.attributes):
|
||||
try:
|
||||
await context.edit(lang('sticker_type_not_support'))
|
||||
except:
|
||||
pass
|
||||
return
|
||||
else:
|
||||
try:
|
||||
await context.edit(lang('sticker_type_not_support'))
|
||||
except:
|
||||
pass
|
||||
return
|
||||
else:
|
||||
try:
|
||||
await context.edit(lang('sticker_reply_not_sticker'))
|
||||
except:
|
||||
pass
|
||||
return
|
||||
|
||||
if photo:
|
||||
file = BytesIO()
|
||||
try:
|
||||
await context.edit(lang('sticker_resizing'))
|
||||
except:
|
||||
pass
|
||||
image = await resize_image(photo)
|
||||
if pic_round:
|
||||
try:
|
||||
await context.edit(lang('us_static_rounding'))
|
||||
except:
|
||||
pass
|
||||
image = await rounded_image(image)
|
||||
file.name = "sticker.webp"
|
||||
image.save(file, "WEBP")
|
||||
file.seek(0)
|
||||
try:
|
||||
await context.edit(lang('us_static_uploading'))
|
||||
except:
|
||||
pass
|
||||
await bot.send_file(context.chat_id, file, force_document=False)
|
||||
try:
|
||||
await context.delete()
|
||||
except:
|
||||
pass
|
||||
|
||||
|
||||
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
|
||||
|
||||
|
||||
async def rounded_image(image):
|
||||
w = image.width
|
||||
h = image.height
|
||||
# 比较长宽
|
||||
if w > h:
|
||||
resize_size = h
|
||||
else:
|
||||
resize_size = w
|
||||
half_size = floor(resize_size / 2)
|
||||
|
||||
# 获取圆角模版，切割成4个角
|
||||
tl = (0, 0, 256, 256)
|
||||
tr = (256, 0, 512, 256)
|
||||
bl = (0, 256, 256, 512)
|
||||
br = (256, 256, 512, 512)
|
||||
border = Image.open('pagermaid/static/images/rounded.png').convert('L')
|
||||
tlp = border.crop(tl)
|
||||
trp = border.crop(tr)
|
||||
blp = border.crop(bl)
|
||||
brp = border.crop(br)
|
||||
|
||||
# 缩放四个圆角
|
||||
tlp = tlp.resize((half_size, half_size))
|
||||
trp = trp.resize((half_size, half_size))
|
||||
blp = blp.resize((half_size, half_size))
|
||||
brp = brp.resize((half_size, half_size))
|
||||
|
||||
# 扩展四个角大小到目标图大小
|
||||
# tlp = ImageOps.expand(tlp, (0, 0, w - tlp.width, h - tlp.height))
|
||||
# trp = ImageOps.expand(trp, (w - trp.width, 0, 0, h - trp.height))
|
||||
# blp = ImageOps.expand(blp, (0, h - blp.height, w - blp.width, 0))
|
||||
# brp = ImageOps.expand(brp, (w - brp.width, h - brp.height, 0, 0))
|
||||
|
||||
# 四个角合并到一张新图上
|
||||
ni = Image.new('RGB', (w, h), (0, 0, 0)).convert('L')
|
||||
ni.paste(tlp, (0, 0))
|
||||
ni.paste(trp, (w - trp.width, 0))
|
||||
ni.paste(blp, (0, h - blp.height))
|
||||
ni.paste(brp, (w - brp.width, h - brp.height))
|
||||
|
||||
# 合并圆角和原图
|
||||
image.putalpha(ImageOps.invert(ni))
|
||||
|
||||
return image
|
||||
@ -9,21 +9,21 @@ from pagermaid.utils import alias_command
|
||||
from random import random
|
||||
|
||||
|
||||
@listener(outgoing=True, command=alias_command("pic"),
|
||||
@listener(outgoing=True, command=alias_command("stickertopic"),
|
||||
description="将你回复的静态贴纸转换为图片", parameters="<y/n>（是否发送原图，默认为n）")
|
||||
async def stickertopic(context):
|
||||
try:
|
||||
if len(context.parameter) >= 1:
|
||||
if context.parameter[0][0].lower()=="y":
|
||||
as_file=True
|
||||
elif context.parameter[0][0].lower()=="n":
|
||||
as_file=False
|
||||
if context.parameter[0][0].lower() == "y":
|
||||
as_file = True
|
||||
elif context.parameter[0][0].lower() == "n":
|
||||
as_file = False
|
||||
elif not context.parameter[0]:
|
||||
as_file=False
|
||||
as_file = False
|
||||
else:
|
||||
raise IndexError
|
||||
else:
|
||||
as_file=False
|
||||
as_file = False
|
||||
except:
|
||||
await context.edit("出错了呜呜呜 ~ 无效的参数。")
|
||||
return
|
||||
@ -78,7 +78,7 @@ async def stickertopic(context):
|
||||
if not animated:
|
||||
await context.edit("正在转换...\n███████70%")
|
||||
image = Image.open(photo)
|
||||
filename = "sticker"+str(random())[2:]+".png"
|
||||
filename = "sticker" + str(random())[2:] + ".png"
|
||||
image.save(filename, "PNG")
|
||||
else:
|
||||
await context.edit("出错了呜呜呜 ~ 目标不是**静态**贴纸 。")
|
||||
@ -86,7 +86,7 @@ async def stickertopic(context):
|
||||
await context.delete()
|
||||
return
|
||||
await context.edit("正在上传...\n██████████99%")
|
||||
await bot.send_file(context.chat_id,filename, force_document=as_file)
|
||||
await bot.send_file(context.chat_id, filename, force_document=as_file)
|
||||
try:
|
||||
await context.delete()
|
||||
except:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user