70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
import os
|
|
import aiofiles
|
|
import aiohttp
|
|
from PIL import (
|
|
Image,
|
|
ImageDraw,
|
|
ImageFont,
|
|
)
|
|
|
|
|
|
def changeImageSize(maxWidth, maxHeight, image):
|
|
if image.size[0] == image.size[1]:
|
|
# Does not change the scale of the orientation image and displays it centered.
|
|
# It may look even better
|
|
newImage = image.resize((maxHeight, maxHeight))
|
|
img = Image.new("RGBA", (maxWidth, maxHeight))
|
|
img.paste(newImage, (int((maxWidth - maxHeight) / 2), 0))
|
|
return img
|
|
else:
|
|
widthRatio = maxWidth / image.size[0]
|
|
heightRatio = maxHeight / image.size[1]
|
|
newWidth = int(widthRatio * image.size[0])
|
|
newHeight = int(heightRatio * image.size[1])
|
|
newImage = image.resize((newWidth, newHeight))
|
|
return newImage
|
|
|
|
|
|
async def thumb(thumbnail, title, userid, ctitle, queue=False):
|
|
img_path = f"search/thumb{userid}.png"
|
|
if 'http' in thumbnail:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(thumbnail) as resp:
|
|
if resp.status == 200:
|
|
f = await aiofiles.open(img_path, mode="wb")
|
|
await f.write(await resp.read())
|
|
await f.close()
|
|
else:
|
|
img_path = thumbnail
|
|
image1 = Image.open(img_path)
|
|
image2 = Image.open("driver/source/LightGreen.png")
|
|
image3 = changeImageSize(1280, 720, image1)
|
|
image4 = changeImageSize(1280, 720, image2)
|
|
image5 = image3.convert("RGBA")
|
|
image6 = image4.convert("RGBA")
|
|
Image.alpha_composite(image5, image6).save(f"search/temp{userid}.png")
|
|
img = Image.open(f"search/temp{userid}.png")
|
|
draw = ImageDraw.Draw(img)
|
|
font = ImageFont.truetype("driver/source/SourceHanSansCN-Regular-2.otf", 50)
|
|
font2 = ImageFont.truetype("driver/source/SourceHanSansCN-Medium-2.otf", 72)
|
|
draw.text(
|
|
(25, 615),
|
|
f"{title[:20]}...",
|
|
fill="black",
|
|
font=font2,
|
|
)
|
|
Play_Msg = "正在播放"
|
|
if queue:
|
|
Play_Msg = "即将播放"
|
|
draw.text(
|
|
(27, 543),
|
|
f"{ctitle[:12]} {Play_Msg}",
|
|
fill="black",
|
|
font=font,
|
|
)
|
|
img.save(f"search/final{userid}.png")
|
|
os.remove(f"search/temp{userid}.png")
|
|
os.remove(img_path)
|
|
final = f"search/final{userid}.png"
|
|
return final
|