Genshin-bots/GenshinUID
1
0
Fork 0
mirror of https://github.com/KimigaiiWuyi/GenshinUID.git synced 2025-05-06 19:53:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 尝试修复原神公告 (#604)

Browse Source
This commit is contained in:
KimigaiiWuyi 2023-11-12 01:10:51 +08:00
parent 4f17dc1b3d
commit 23de41181b
1 changed files with 28 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
GenshinUID/utils/image/image_tools.py
View File

@ -31,6 +31,12 @@ else:
bg_path = NM_BG_PATH bg_path = NM_BG_PATH
async def _get(url: str):
async with httpx.AsyncClient(timeout=None) as client:
resp = await client.get(url=url)
return resp
async def get_pic(url, size: Optional[Tuple[int, int]] = None) -> Image.Image: async def get_pic(url, size: Optional[Tuple[int, int]] = None) -> Image.Image:
""" """
从网络获取图片, 格式化为RGBA格式的指定尺寸 从网络获取图片, 格式化为RGBA格式的指定尺寸
@ -48,6 +54,16 @@ async def get_pic(url, size: Optional[Tuple[int, int]] = None) -> Image.Image:
return pic return pic
def get_size(font: ImageFont.FreeTypeFont, character: str):
if hasattr(font, 'getsize'):
w, h = font.getsize(character) # type: ignore
else:
bbox = font.getbbox(character)
w, h = bbox[2] - bbox[0], bbox[3] - bbox[1]
return w, h
def draw_text_by_line( def draw_text_by_line(
img: Image.Image, img: Image.Image,
pos: Tuple[int, int], pos: Tuple[int, int],
@ -64,7 +80,7 @@ def draw_text_by_line(
line_space 行间距, 单位像素, 默认是字体高度的0.3倍 line_space 行间距, 单位像素, 默认是字体高度的0.3倍
""" """
x, y = pos x, y = pos
_, h = font.getsize('X') # type: ignore _, h = get_size(font, 'X')
if line_space is None: if line_space is None:
y_add = math.ceil(1.3 * h) y_add = math.ceil(1.3 * h)
else: else:
@ -74,23 +90,23 @@ def draw_text_by_line(
length = 0 # 记录本行长度 length = 0 # 记录本行长度
for character in text: for character in text:
# 获取当前字符的宽度 # 获取当前字符的宽度
w, h = font.getsize(character) # type: ignore w, h = get_size(font, character)
if length + w * 2 <= max_length: if length + w * 2 <= max_length:
row += character row += character
length += w length += w
else: else:
row += character row += character
if center: if center:
font_size = font.getsize(row) # type: ignore fw, _ = get_size(font, row)
x = math.ceil((img.size[0] - font_size[0]) / 2) x = math.ceil((img.size[0] - fw) / 2)
draw.text((x, y), row, font=font, fill=fill) draw.text((x, y), row, font=font, fill=fill)
row = "" row = ""
length = 0 length = 0
y += y_add y += y_add
if row != "": if row != "":
if center: if center:
font_size = font.getsize(row) # type: ignore fw, _ = get_size(font, row)
x = math.ceil((img.size[0] - font_size[0]) / 2) x = math.ceil((img.size[0] - fw) / 2)
draw.text((x, y), row, font=font, fill=fill) draw.text((x, y), row, font=font, fill=fill)
@ -203,7 +219,9 @@ async def get_qq_avatar(
avatar_url = f'http://q1.qlogo.cn/g?b=qq&nk={qid}&s=640' avatar_url = f'http://q1.qlogo.cn/g?b=qq&nk={qid}&s=640'
elif avatar_url is None: elif avatar_url is None:
avatar_url = 'https://q1.qlogo.cn/g?b=qq&nk=3399214199&s=640' avatar_url = 'https://q1.qlogo.cn/g?b=qq&nk=3399214199&s=640'
char_pic = Image.open(BytesIO(get(avatar_url).content)).convert('RGBA') char_pic = Image.open(BytesIO((await _get(avatar_url)).content)).convert(
'RGBA'
)
return char_pic return char_pic
@ -294,7 +312,9 @@ async def get_simple_bg(
) -> Image.Image: ) -> Image.Image:
if image: if image:
if isinstance(image, str): if isinstance(image, str):
edit_bg = Image.open(BytesIO(get(image).content)).convert('RGBA') edit_bg = Image.open(BytesIO((await _get(image)).content)).convert(
'RGBA'
)
elif isinstance(image, Image.Image): elif isinstance(image, Image.Image):
edit_bg = image.convert('RGBA') edit_bg = image.convert('RGBA')
else: else: