CameraMachine/python/custom_plugins/plugins_mode/detail_generate_base.py

import asyncio
import settings
import os

try:
    is_test_plugins = settings.is_test_plugins
except:
    is_test_plugins = False

if is_test_plugins:
    from custom_plugins.plugins_mode.pic_deal import PictureProcessing
else:
    from plugins_mode.pic_deal import PictureProcessing

from PIL import Image
import shutil
from service.base import get_images, check_path, get_image_mask
from natsort import ns, natsorted

# import math
from PIL import ImageFont
import settings
from settings import sendSocketMessage


def sendMessageAsync(code=0, msg="开始处理详情", data=None, msg_type="detail_progress"):
    """异步发送消息"""
    loop = asyncio.get_event_loop()
    loop.create_task(
        sendSocketMessage(
            code=code,
            msg=msg,
            data=data,
            msg_type=msg_type,
        )
    )


class DetailBase(object):
    def __init__(
        self,
        goods_no,
        goods_no_value: dict,
        out_put_dir,
        windows=None,
        excel_data=None,
        assigned_page_list=None,
        output_queue=None,
    ):
        self.goods_no = goods_no
        self.output_queue = output_queue
        self.out_put_dir = out_put_dir
        self.deal_pic_func_list = []
        self.goods_no_value = goods_no_value
        self.goods_art_nos = [
            item.get("货号", "") for item in self.goods_no_value.get("货号资料", [])
        ]
        self.root = ""
        self.windows = windows
        self.template_name = None

        print(goods_no_value)
        # 重新解析为新的数据结构
        self.data = {}
        self.detailed_images = []
        self.assigned_page_list = assigned_page_list
        self.overlay_pic_dict = {}
        self.init()
        # for goods_art_no_dict in self.goods_no_value["货号资料"]:
        #     print(goods_art_no_dict)
        #
        # raise 1

        if excel_data:
            ig_keys = ["模板名称"]
            for k, v in excel_data.items():
                if k not in ig_keys:
                    self.goods_no_value[k] = v

    def check_shoe_is_right_by_pixel(self, im=None, image_path=None):
        if im is None:
            im = Image.open(image_path)
        # 注意，只支持透明图
        # 打开图像文件
        im = im.crop(im.getbbox())
        # image.show()
        # 获取图像第一行的像素数据
        pixel_data = im.load()
        pix_list = []
        h = int(im.height / 20)
        for i in range(im.width):
            _r, _g, _b, _a = pixel_data[i, h]
            if _a > 10:
                pix_list.append(i)

        left_f_num = 0
        middle_w = int(im.width / 2)
        for i in pix_list:
            if i < middle_w:
                left_f_num += 1
            else:
                left_f_num -= 1
        if left_f_num > 0:
            return True
        else:
            return False

    def del_detail_folder(self):
        out_path = "{out_put_dir}/{goods_no}".format(
            out_put_dir=self.out_put_dir, goods_no=self.goods_no
        )
        if not os.path.exists(out_path):
            return
        try:
            shutil.rmtree(out_path)
        except BaseException as e:
            print("删除文件夹失败", e)

    def run_all(self):
        if self.template_name:
            self.out_put_dir = "{}/详情模板{}".format(
                self.out_put_dir, self.template_name
            )

        print("===================detailed_images=================")
        # 如果没有指定页面，则删除指定目录下的对应的详情文件夹
        if not self.assigned_page_list:
            self.del_detail_folder()
        detailed_images = self.deal_details()
        self.create_folder(self.out_put_dir)
        detail_path = "{out_put_dir}/{goods_no}/详情页切片".format(
            out_put_dir=self.out_put_dir, goods_no=self.goods_no
        )
        self.create_folder(detail_path)
        self.save_to_png(detailed_images=detailed_images, detail_path=detail_path)
        # 生成拼接图
        self.generate_spliced_picture()

        # ------------移动其他图片---------------------
        # 获取主图模板列表
        main_pic_path_list = DetailBase.get_temp_pic_info(root=self.root)[
            "main_pic_path_list"
        ]
        if not main_pic_path_list:
            self.move_other_pic(move_main_pic=True)
        else:
            self.move_other_pic(move_main_pic=True)
            if not self.assigned_page_list:
                self.deal_all_main_pic()
            else:
                if "主图" in self.assigned_page_list:
                    self.deal_all_main_pic()
        # ----------如果是红蜻蜓则创建同颜色下的其他货号颜色文件夹---------------
        if settings.PROJECT == "红蜻蜓":
            if "data_all_goods_art_info" in self.goods_no_value:
                # 数据格式：[{'number': '14250232', 'goods_art_no': 'AC52001173', 'color': '杏色'}, ]
                for pic_data in self.goods_no_value["货号资料"]:
                    if "颜色名称" not in pic_data:
                        continue
                    color_name = pic_data["颜色名称"]

                    color_file_path = "{out_put_dir}/{goods_no}/{goods_number}".format(
                        out_put_dir=self.out_put_dir,
                        goods_no=self.goods_no,
                        goods_number=pic_data["编号"],
                    )
                    for i in self.goods_no_value["data_all_goods_art_info"]:
                        if color_name in i["color"]:
                            new_path = "{out_put_dir}/{goods_no}/{goods_number}".format(
                                out_put_dir=self.out_put_dir,
                                goods_no=self.goods_no,
                                goods_number="NUM{}".format(i["number"]),
                            )
                            if not os.path.exists(new_path):
                                # 创建文件夹
                                os.makedirs(new_path)
                                self.move_one_pic(
                                    color_file_path,
                                    new_path,
                                    "NUM{}".format(i["number"]),
                                )
        sendMessageAsync(
            code=0,
            msg="详情页生成完成",
            msg_type="detail_progress",
            data={
                "goods_no": self.goods_no,
                "temp_name": self.template_name,
                "status": "已完成",
                "goods_art_nos": self.goods_art_nos,
            },
        )
        return True

    # 移动一张图片到新的文件夹
    def move_one_pic(self, old_path, new_path, new_name):
        image_file = os.listdir(old_path)[0]
        old_image_path = "{}/{}".format(old_path, image_file)
        image_e = os.path.splitext(image_file)[1]
        new_image_path = "{}/{}{}".format(new_path, new_name, image_e)
        shutil.copy(old_image_path, new_image_path)

    # 生成各个详情图切片
    def deal_details(self):
        detailed_images = []
        sendMessageAsync(
            code=0,
            msg="正在生成详情页切片",
            msg_type="detail_progress",
            data={
                "goods_no": self.goods_no,
                "temp_name": self.template_name,
                "status": "进行中",
                "goods_art_nos": self.goods_art_nos,
            },
        )
        for index, func in enumerate(self.deal_pic_func_list):
            image_pp = func()
            if not self.assigned_page_list:
                self.image_list_append(detailed_images, image_pp)
            else:
                index = "{}".format(index + 1)
                if index in self.assigned_page_list:
                    self.image_list_append(detailed_images, image_pp)
                else:
                    self.image_list_append(detailed_images, {"mes": "不生成"})

        return [x for x in detailed_images if x]

    # 生成拼接的图片
    def generate_spliced_picture(self):
        sendMessageAsync(
            code=0,
            msg="正在生成详情拼接图",
            msg_type="detail_progress",
            data={
                "goods_no": self.goods_no,
                "temp_name": self.template_name,
                "status": "进行中",
                "goods_art_nos": self.goods_art_nos,
            },
        )
        detail_path = "{out_put_dir}/{goods_no}/详情页切片".format(
            out_put_dir=self.out_put_dir, goods_no=self.goods_no
        )
        if not os.path.exists(detail_path):
            return
        detailed_images = []
        for image_data in get_images(detail_path):
            detailed_images.append(PictureProcessing(image_data["file_path"]))
        # 生成拼接图
        img = self.add_pic(detailed_images)
        join_path = "{out_put_dir}/{goods_no}/详情页".format(
            out_put_dir=self.out_put_dir, goods_no=self.goods_no
        )
        # self.create_folder(join_path)
        img.save("{}.jpg".format(join_path), format="JPEG")

    def image_list_append(self, image_list: list, data):
        self.check_state_end()
        if isinstance(data, list):
            image_list.extend(data)
        else:
            image_list.append(data)

    def save_to_png(self, detailed_images, detail_path):
        self.check_state_end()
        for index, pp in enumerate(detailed_images):
            if isinstance(pp, dict):
                continue
            pp.im.save(
                "{}/{}（{}）.png".format(
                    detail_path, self.goods_no, str(index + 11).zfill(2)
                )
            )

    def check_state_end(self):
        if self.windows is not None:
            if self.windows.state == 99:
                raise "用户主动取消"

    @classmethod
    def get_temp_pic_info(cls, root):
        """
        获取详情页模板中的信息
        """
        main_pic_list = []
        mask_pic_list = []
        if os.path.exists(r"{}\main_image".format(root)):
            for _name in os.listdir(r"{}\main_image".format(root)):
                _path = r"{}\main_image\{}".format(root, _name)
                if os.path.isdir(_path):
                    main_pic_list.append([x["file_path"] for x in get_images(_path)])
                    mask_pic_list.append(
                        [x["file_path"] for x in get_image_mask(_path)]
                    )
        _l = get_images(r"{}\show".format(root))
        temp_pic_path = _l[0]["file_path"] if _l else None

        other_pic_list = [x["file_path"] for x in get_images(r"{}".format(root))]
        return {
            "main_pic_path_list": main_pic_list,
            "temp_pic_path": temp_pic_path,
            "mask_pic_list": mask_pic_list,
            "other_pic_path_list": other_pic_list,
        }

    def init(self):
        for goods_art_no_value in self.goods_no_value["货号资料"]:
            self.data[goods_art_no_value["货号"]] = {
                "pics": goods_art_no_value["pics"],
                "pic_is_deal": {},
            }

    def get_text_value(self, key, subsection_len=0):
        text = ""
        if key in self.goods_no_value:
            if self.goods_no_value[key]:
                text = str(self.goods_no_value[key])
                text = text.replace(r"\n", "\n")

        # if key in ["跟高", "鞋宽", "帮高", "脚掌围", "鞋长"]:
        #     if text:
        #         text = text.split(".")[0]

        if subsection_len != 0:
            text = text.split("\n")
            text = [x for x in text if x]
            if len(text) == 2:
                text_1 = text[0]
                text_2 = text[1]
                return text_1, text_2
            else:
                if text:
                    text_1 = text[0]
                else:
                    text_1 = ""
                text_2 = ""
                return text_1, text_2

        return text

    def create_folder(self, path):
        if not os.path.exists(path):
            os.makedirs(path)

    def get_all_process_pics(self):
        """
        获取所有颜色的过程图片
        data = [
            {"货号": "",
             "素材": [{
                 "名称": "俯视",
                 "抠图": "路径1",
                 "阴影": "路径2"
             }, ]},
        ]
        """
        return_data = []
        for goods_art_no in self.data:
            goods_art_no_dict = {
                "货号": goods_art_no,
                "素材": [],
            }

            # 图片数据重新排序
            pic_data = []
            for pic_name, pic_path in self.data[goods_art_no]["pics"].items():
                root_path, file_name = os.path.split(pic_path)
                pic_data.append(file_name)

            pic_data = natsorted(pic_data, alg=ns.PATH)

            for file_name in pic_data:
                if "阴影" in file_name:
                    _, action_name, _ = file_name.split("_")
                    pic_path = self.data[goods_art_no]["pics"][
                        "{}-阴影".format(action_name)
                    ]
                    pic_cutout_path = self.data[goods_art_no]["pics"][
                        "{}-抠图".format(action_name)
                    ]
                    if os.path.exists(pic_path) and os.path.exists(pic_cutout_path):
                        goods_art_no_dict["素材"].append(
                            {
                                "名称": action_name,
                                "抠图": pic_cutout_path,
                                "阴影": pic_path,
                            }
                        )
            return_data.append(goods_art_no_dict)

        return return_data

    def get_overlay_pic_from_dict(
        self, goods_art_no, color_name, bg_color
    ) -> PictureProcessing:
        self.check_state_end()
        # 增加逻辑，获取任意货号下的组合图
        if "组合" in color_name:
            goods_art_no, color_name = self.get_all_scene_list(goods_art_no, color_name)

        key = "{}-{}-{}".format(goods_art_no, color_name, bg_color)
        if key in self.overlay_pic_dict:
            return self.overlay_pic_dict[key]

        if goods_art_no in self.data:
            for pic_name, pic_path in self.data[goods_art_no]["pics"].items():
                if "阴影" in pic_name:
                    action_name = pic_name.replace("-阴影", "")
                    if action_name == color_name:
                        pp1 = PictureProcessing(pic_path)
                        pp2 = PictureProcessing(
                            self.data[goods_art_no]["pics"][
                                "{}-抠图".format(action_name)
                            ]
                        )
                        pp1 = pp1.get_overlay_pic(top_img=pp2, color=bg_color).resize(
                            mode="pixel", base="width", value=1600
                        )
                        self.overlay_pic_dict[key] = pp1

        if key in self.overlay_pic_dict:
            return self.overlay_pic_dict[key]

    def image_init(self, bg_color=(246, 246, 246)):
        # 制作一批素材图，添加背景色，并保留阴影，以及处理成最小尺寸
        for goods_art_no in self.data:
            for pic_name, pic_path in self.data[goods_art_no]["pics"].items():
                if "阴影" in pic_name:
                    action_name = pic_name.replace("-阴影", "")
                    pp1 = PictureProcessing(pic_path)
                    pp2 = PictureProcessing(
                        self.data[goods_art_no]["pics"]["{}-抠图".format(action_name)]
                    )
                    pp1 = pp1.get_overlay_pic(top_img=pp2, color=bg_color).resize(
                        mode="pixel", base="width", value=1600
                    )
                    self.data[goods_art_no]["pic_is_deal"][action_name] = pp1

    # 获取任意货号的场景图,优先取指定货号；
    # 调整，按顺序从货号列表中提取所有组合图
    def get_all_scene_info(self, goods_art_no):
        data = []
        # 收集所有组合图
        # 找任意一个有组合图的货号
        for goods_art_no_dict in self.goods_no_value["货号资料"]:
            _goods_art_no = goods_art_no_dict["货号"]
            _view_name_list = set([x.split("-")[0] for x in goods_art_no_dict["pics"]])
            for _view_name in _view_name_list:
                if "组合" not in _view_name:
                    continue
                return _goods_art_no
        return goods_art_no

    def get_all_scene_list(self, goods_art_no, view_name: str):
        if "组合" == view_name:
            view_name = "组合1"
        try:
            view_index = int(view_name.replace("组合", "")) - 1
        except:
            return goods_art_no, "无法匹配"

        data = []
        # 收集所有组合图
        # 找任意一个有组合图的货号
        for goods_art_no_dict in self.goods_no_value["货号资料"]:
            _goods_art_no = goods_art_no_dict["货号"]
            if _goods_art_no != goods_art_no:
                continue
            _view_name_list = set([x.split("-")[0] for x in goods_art_no_dict["pics"]])
            for _view_name in _view_name_list:
                if "组合" not in _view_name:
                    continue
                data.append(
                    {
                        "goods_art_no": _goods_art_no,
                        "view_name": _view_name,
                        "real_view_name": (
                            "组合1" if _view_name == "组合" else _view_name
                        ),
                    }
                )
        if len(data) <= view_index:
            return goods_art_no, "无法匹配"
        else:
            data.sort(key=lambda x: x["real_view_name"], reverse=False)
            return data[view_index]["goods_art_no"], data[view_index]["view_name"]

    def image_one_pic(self, goods_art_no, name, bg_color=None, return_orign=None):
        # 增加逻辑，获取任意货号下的组合图
        if "组合" in name:
            print("324==== goods_art_no, name", goods_art_no, name)
            goods_art_no, name = self.get_all_scene_list(goods_art_no, name)
            print("324 goods_art_no, name", goods_art_no, name)

        # 制作一批素材图，添加背景色，并保留阴影，以及处理成最小尺寸
        for pic_name, pic_path in self.data[goods_art_no]["pics"].items():
            if "阴影" in pic_name:
                action_name = pic_name.replace("-阴影", "")
                if name != action_name:
                    continue
                pp1 = PictureProcessing(pic_path)
                pp2 = PictureProcessing(
                    self.data[goods_art_no]["pics"]["{}-抠图".format(action_name)]
                )
                if not return_orign:
                    pp1 = pp1.get_overlay_pic(top_img=pp2, color=bg_color).resize(
                        mode="pixel", base="width", value=1600
                    )
                    return pp1
                else:
                    return pp1, pp2

        if not return_orign:
            return None
        else:
            return None, None

    def move_other_pic(self, move_main_pic=True):
        # ------------------------------移动其他图片------------------------------
        goods_no_main_pic_number = 0
        for goods_art_no_dict in self.goods_no_value["货号资料"]:
            if "800x800" not in goods_art_no_dict:
                continue
            if not goods_art_no_dict["800x800"]:
                continue

            goods_art_no = ""
            if "编号" in goods_art_no_dict:
                if goods_art_no_dict["编号"]:
                    goods_art_no = goods_art_no_dict["编号"]
            if not goods_art_no:
                goods_art_no = goods_art_no_dict["货号"]
            # print("goods_art_no:", goods_art_no)
            # 移动颜色图=====================
            goods_art_no_f = "{}/{}".format(self.out_put_dir, self.goods_no)
            self.create_folder(goods_art_no_f)
            # 放入一张主图
            old_pic_path_1 = goods_art_no_dict["800x800"][0]
            shutil.copy(
                old_pic_path_1,
                "{}/颜色图{}{}".format(
                    goods_art_no_f, goods_art_no, os.path.splitext(old_pic_path_1)[1]
                ),
            )

            # 把其他主图放入作为款号图=====================
            if move_main_pic:
                for pic_path in goods_art_no_dict["800x800"]:
                    goods_no_main_pic_number += 1
                    e = os.path.splitext(pic_path)[1]
                    shutil.copy(
                        pic_path,
                        "{out_put_dir}/{goods_no}/主图{goods_no}（{goods_no_main_pic_number}）{e}".format(
                            out_put_dir=self.out_put_dir,
                            goods_no=self.goods_no,
                            goods_no_main_pic_number=str(
                                goods_no_main_pic_number + 10
                            ).zfill(2),
                            e=e,
                        ),
                    )

    def deal_all_main_pic(self):
        """
        处理主图模板，如存在出图模板则进行对应处理
        """
        # 获取主图模板列表
        all_main_pic_path_list = DetailBase.get_temp_pic_info(root=self.root)[
            "main_pic_path_list"
        ]
        if not all_main_pic_path_list:
            return
        mask_pic_list = DetailBase.get_temp_pic_info(root=self.root)["mask_pic_list"]
        data = self.get_all_process_pics()
        print("========deal_all_main_pic=========主图相关素材：")

        view_list = [
            "组合",
            "组合2",
            "组合3",
            "组合4",
            "组合5",
            "组合6",
            "俯视",
            "侧视",
            "后跟",
            "鞋底",
            "内里",
        ]

        for _index, main_pic_path_list in enumerate(all_main_pic_path_list):
            self.check_state_end()
            out_path_root = "{out_put_dir}/{goods_no}/main_image_{_index}".format(
                out_put_dir=self.out_put_dir, goods_no=self.goods_no, _index=_index
            )
            check_path(out_path_root)
            if mask_pic_list[_index]:
                mask_pic = mask_pic_list[_index][0]
            else:
                mask_pic = None
            goods_no_main_pic_number = 10
            # g_index 为第几个颜色货号
            for g_index, goods_art_no_dict in enumerate(data):
                goods_art_no = goods_art_no_dict["货号"]
                # =====================重新指定=================================
                _material_sort_dict = {}
                for index, material_dict in enumerate(goods_art_no_dict["素材"]):
                    name = material_dict["名称"]
                    _material_sort_dict[name] = material_dict

                # ======================================================
                file_name_index = -1
                for view_name in view_list:
                    # 组合图比较特殊，为全局获取
                    if g_index != 0:
                        if "组合" in view_name:
                            continue

                    if view_name not in _material_sort_dict:
                        continue
                    self.check_state_end()
                    pp_jpg, pp_png = self.image_one_pic(
                        goods_art_no, view_name, bg_color=None, return_orign=True
                    )
                    if not pp_jpg:
                        continue

                    file_name_index += 1
                    # 获取对应主图模板
                    if len(main_pic_path_list) < file_name_index + 1:
                        main_pic_path = main_pic_path_list[-1]
                    else:
                        main_pic_path = main_pic_path_list[file_name_index]
                    pp_bg = PictureProcessing(main_pic_path)
                    original_width = pp_bg.width

                    if original_width != 1600:
                        pp_bg = pp_bg.resize(value=1600)

                    if mask_pic:
                        mask_bg = PictureProcessing(mask_pic)
                        mask_bg = mask_bg.resize(value=1600)
                        mask_box_im = mask_bg.get_im()
                        box_size = mask_box_im.getbbox()
                        result_image = mask_box_im.crop(box_size)
                        mask_width, mask_height = result_image.size
                        mask_x, mask_y = box_size[0], box_size[1]
                    else:
                        mask_width, mask_height = pp_bg.size
                        mask_width, mask_height = int(mask_width * 12 / 16), int(
                            mask_height * 12 / 16
                        )
                        mask_x, mask_y = int((pp_bg.size[0] - mask_width) / 2), int(
                            (pp_bg.size[1] - mask_height) / 2
                        )

                    if view_name != "后跟":
                        pp_jpg = pp_jpg.resize(base_by_box=(mask_width, mask_height))
                        pp_png = pp_png.resize(base_by_box=(mask_width, mask_height))

                        # 计算粘贴的位置 mask的位置+图片在mask中的位置
                        p_x = mask_x + int((mask_width - pp_jpg.width) / 2)
                        p_y = mask_y + int((mask_height - pp_jpg.height) / 2)
                        pp_bg = pp_bg.to_overlay_pic_advance(
                            mode="pixel",
                            top_img=pp_jpg,
                            base="nw",
                            value=(p_x, p_y),
                            top_png_img=pp_png,
                        )
                    else:
                        new_mask_width, new_mask_height = int(mask_width / 1.6), int(
                            mask_height / 1.6
                        )

                        pp_jpg = pp_jpg.resize(
                            base_by_box=(new_mask_width, new_mask_height)
                        )
                        pp_png = pp_png.resize(
                            base_by_box=(new_mask_width, new_mask_height)
                        )
                        new_mask_x = int((mask_width - new_mask_width) / 2 + mask_x)
                        new_mask_y = int((mask_height - new_mask_height) / 2 + mask_y)

                        # 计算粘贴的位置 mask的位置+图片在mask中的位置
                        p_x = new_mask_x + int((new_mask_width - pp_jpg.width) / 2)
                        p_y = new_mask_y + int((new_mask_height - pp_jpg.height) / 2)

                        pp_bg = pp_bg.to_overlay_pic_advance(
                            mode="pixel",
                            top_img=pp_jpg,
                            base="nw",
                            value=(p_x, p_y),
                            top_png_img=pp_png,
                        )
                    out_pci_mode = "." + settings.getSysConfigs(
                        "basic_configs", "image_out_format", "png"
                    )
                    goods_no_main_pic_number += 1
                    out_pic_path = "{out_path_root}/{goods_no}（{goods_no_main_pic_number}）{pic_mode}".format(
                        out_path_root=out_path_root,
                        goods_no=self.goods_no,
                        goods_no_main_pic_number=goods_no_main_pic_number,
                        pic_mode=out_pci_mode,
                    )
                    out_pci_factor = float(
                        1
                        if settings.getSysConfigs(
                            "basic_configs", "image_sharpening", "1"
                        )
                        == ""
                        else settings.getSysConfigs(
                            "basic_configs", "image_sharpening", "1"
                        )
                    )
                    if out_pci_factor > 1.0:
                        print("图片锐化处理")
                        pp_bg = pp_bg.sharpen_image(factor=out_pci_factor)

                    if original_width < 1600:
                        pp_bg = pp_bg.resize(value=original_width)

                    print("392  out_pic_path", out_pic_path)
                    if out_pci_mode == ".jpg":
                        pp_bg.save_as_rgb(out_pic_path)
                    elif out_pci_mode == ".png":
                        pp_bg.save_as_png(out_pic_path)
                    else:
                        pp_bg.save_as_other(out_pic_path, out_pci_mode.split(".")[-1])

    def add_pic(self, detailed_images):
        self.check_state_end()
        todo_detailed_images = []
        detailed_images = [x for x in detailed_images if x]
        if not detailed_images:
            return
        for i in detailed_images:
            if isinstance(i, list):
                for n in i:
                    todo_detailed_images.append(n)
            else:
                todo_detailed_images.append(i)

        page_len = 0
        for index, pp in enumerate(todo_detailed_images):
            page_len += pp.height
        bg_im = Image.new("RGB", (pp.width, page_len), (255, 255, 255))
        n = 0
        for index, pp in enumerate(todo_detailed_images):
            bg_im.paste(pp.im, (0, n))
            n += pp.height
        return bg_im

    # 通用方法，用于写文字
    def add_text_list(self, text_list, spacing=5, base="wn", base_width=1600):
        text_list = [x for x in text_list if x["text"]]
        # print(text_list)
        # spacing 行间距
        text_image_list = []
        max_w = 0
        total_h = 0

        for text_data in text_list:
            _pp = PictureProcessing("RGBA", (base_width, 1200), (255, 255, 255, 0))
            if base == "wn" or base == "nw":
                align = "left"
                anchor = None
                value = (0, 250)
            if base == "cn" or base == "nc":
                align = "center"
                anchor = "mm"
                value = (int(base_width / 2), 250)
            if base == "en" or base == "ne":
                align = "right"
                anchor = "rs"
                value = (base_width - 10, 250)

            _pp = _pp.get_text_image_advanced(
                value=value,
                font=text_data["font"],
                text=text_data["text"],
                align=align,
                anchor=anchor,
                spacing=5,
                fill=text_data["fill"],
                return_mode="min_image",
                margins=(0, 0, 0, 0),
            )
            text_image_list.append(_pp)
            if _pp.width > max_w:
                max_w = _pp.width

            total_h += _pp.height
            if "spacing" in text_data:
                total_h += text_data["spacing"]

        if not text_image_list:
            return None
        #
        bg = PictureProcessing("RGBA", (max_w, total_h * 3), (0, 0, 0, 0))
        y = 0
        for text_image, text_data in zip(text_image_list, text_list):
            bg = bg.paste_img(top_img=text_image, value=(0, y), base=base)
            y += spacing + text_image.height
            if "spacing" in text_data:
                y += text_data["spacing"]

        bg = bg.crop(mode="min")
        # _ = bg.paste_img_invert(top_img=PictureProcessing("RGB", (bg.width,bg.height), (255, 255, 255)))
        # _.show()
        return bg

    def generate_font_list_to_pic(self):
        font_path_list = [
            r"resources\ttf\puhui\Bold.ttf",
            r"resources\ttf\puhui\Medium.ttf",
            r"resources\ttf\puhui\Heavy.ttf",
            r"resources\ttf\puhui\Light.ttf",
            r"resources\ttf\puhui\Regular.ttf",
        ]
        text_v_list = [
            "这是一段话Bold",
            "这是一段话Medium",
            "这是一段话Heavy",
            "这是一段话Light",
            "这是一段话Regular",
        ]
        detailed_images = []
        for font_path, text in zip(font_path_list, text_v_list):
            text_list = []
            for size in range(26, 80, 2):
                font = ImageFont.truetype(font_path, size)
                text_list.append(
                    {
                        "text": "{}-字号{}".format(text, size),
                        "font": font,
                        "fill": (110, 110, 110),
                    }
                )
            text_image = self.add_text_list(text_list, spacing=15, base="nw")
            text_image = text_image.crop(mode="min")
            text_image = text_image.paste_img_invert(
                top_img=PictureProcessing("RGB", text_image.size, (255, 255, 255))
            )
            detailed_images.append(text_image)
        return PictureProcessing(im=self.add_pic(detailed_images))

    # 图片分段，每段至少大于N长度
    def pp_pic_subsection(self, pp: PictureProcessing, one_height=3200):
        total_height = pp.height
        now_height = 0
        detailed_images = []

        while 1:
            if now_height + one_height < total_height:
                h1 = now_height
                h2 = now_height + one_height
                bbox = (0, h1, pp.width, h2)
                # print("bbox1", bbox)
                detailed_images.append(pp.crop(bbox=bbox))
                now_height = now_height + one_height
                continue

            if now_height + one_height >= total_height:
                h1 = now_height
                h2 = total_height
                bbox = (0, h1, pp.width, h2)
                # print("bbox2", bbox)
                detailed_images.append(pp.crop(bbox=bbox))
                break

        return detailed_images