mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
106 lines
3.4 KiB
Python
106 lines
3.4 KiB
Python
|
from typing import List, Tuple
|
||
|
|
from configs.path_config import IMAGE_PATH
|
||
|
|
from utils.image_utils import BuildImage
|
||
|
|
import os
|
||
|
|
import random
|
||
|
|
|
||
|
|
|
||
|
|
background_path = IMAGE_PATH / "background" / "help" / "simple_help"
|
||
|
|
|
||
|
|
|
||
|
|
async def build_help_image(image_group: List[List[BuildImage]], h: int):
|
||
|
|
bk = None
|
||
|
|
random_bk = os.listdir(background_path)
|
||
|
|
if random_bk:
|
||
|
|
bk = random.choice(random_bk)
|
||
|
|
A = BuildImage(
|
||
|
|
h,
|
||
|
|
h,
|
||
|
|
font_size=24,
|
||
|
|
color="#FFEFD5",
|
||
|
|
background=(background_path / bk) if bk else None,
|
||
|
|
)
|
||
|
|
curr_w = 50
|
||
|
|
for ig in image_group:
|
||
|
|
curr_h = 180
|
||
|
|
for img in ig:
|
||
|
|
await A.apaste(img, (curr_w, curr_h), True)
|
||
|
|
curr_h += img.h + 10
|
||
|
|
curr_w += max([x.w for x in ig]) + 30
|
||
|
|
return A
|
||
|
|
|
||
|
|
|
||
|
|
def group_image(image_list: List[BuildImage]) -> Tuple[List[List[BuildImage]], int]:
|
||
|
|
"""
|
||
|
|
说明:
|
||
|
|
根据图片大小进行分组
|
||
|
|
参数:
|
||
|
|
:param image_list: 排序图片列表
|
||
|
|
"""
|
||
|
|
image_list.sort(key=lambda x: x.h, reverse=True)
|
||
|
|
max_image = max(image_list, key=lambda x: x.h)
|
||
|
|
|
||
|
|
image_list.remove(max_image)
|
||
|
|
max_h = max_image.h
|
||
|
|
total_w = 0
|
||
|
|
|
||
|
|
# 图片分组
|
||
|
|
image_group = [[max_image]]
|
||
|
|
is_use = []
|
||
|
|
surplus_list = image_list[:]
|
||
|
|
|
||
|
|
for image in image_list:
|
||
|
|
if image.uid not in is_use:
|
||
|
|
group = [image]
|
||
|
|
is_use.append(image.uid)
|
||
|
|
curr_h = image.h
|
||
|
|
while True:
|
||
|
|
surplus_list = [x for x in surplus_list if x.uid not in is_use]
|
||
|
|
for tmp in surplus_list:
|
||
|
|
temp_h = curr_h + tmp.h + 10
|
||
|
|
if temp_h < max_h or abs(max_h - temp_h) < 100:
|
||
|
|
curr_h += tmp.h + 15
|
||
|
|
is_use.append(tmp.uid)
|
||
|
|
group.append(tmp)
|
||
|
|
break
|
||
|
|
else:
|
||
|
|
break
|
||
|
|
total_w += max([x.w for x in group]) + 15
|
||
|
|
image_group.append(group)
|
||
|
|
while surplus_list:
|
||
|
|
surplus_list = [x for x in surplus_list if x.uid not in is_use]
|
||
|
|
if not surplus_list:
|
||
|
|
break
|
||
|
|
surplus_list.sort(key=lambda x: x.h, reverse=True)
|
||
|
|
for img in surplus_list:
|
||
|
|
if img.uid not in is_use:
|
||
|
|
_w = 0
|
||
|
|
index = -1
|
||
|
|
for i, ig in enumerate(image_group):
|
||
|
|
if s := sum([x.h for x in ig]) > _w:
|
||
|
|
_w = s
|
||
|
|
index = i
|
||
|
|
if index != -1:
|
||
|
|
image_group[index].append(img)
|
||
|
|
is_use.append(img.uid)
|
||
|
|
max_h = 0
|
||
|
|
max_w = 0
|
||
|
|
for i, ig in enumerate(image_group):
|
||
|
|
if (_h := sum([x.h + 15 for x in ig])) > max_h:
|
||
|
|
max_h = _h
|
||
|
|
max_w += max([x.w for x in ig]) + 30
|
||
|
|
is_use.clear()
|
||
|
|
while abs(max_h - max_w) > 200 and len(image_group) - 1 >= len(image_group[-1]):
|
||
|
|
for img in image_group[-1]:
|
||
|
|
_min_h = 0
|
||
|
|
_min_index = -1
|
||
|
|
for i, ig in enumerate(image_group):
|
||
|
|
if i not in is_use and (_h := sum([x.h for x in ig]) + img.h) > _min_h:
|
||
|
|
_min_h = _h
|
||
|
|
_min_index = i
|
||
|
|
is_use.append(_min_index)
|
||
|
|
image_group[_min_index].append(img)
|
||
|
|
max_w -= max([x.w for x in image_group[-1]])
|
||
|
|
image_group.pop(-1)
|
||
|
|
return image_group, max(max_h + 250, max_w + 70)
|