CameraMachine/python/service/generate_goods_no_detail_pic/detail_hlm_3.py

"""
步骤：
1、整理需要处理的款号图-输出款号图文件夹
2、整理所有相关的图片作为素材图
3、按要求进行拼接
"""
import os
from PIL import ImageFont
from module.view_control.generate_goods_no_detail_pic.detail_generate_base import DetailBase
from module.view_control.generate_goods_no_detail_pic.pic_deal import PictureProcessing


class DetailPicGetHLM3(DetailBase):
    need_view = ["俯视", "侧视", "后跟", "鞋底", "内里"]
    root = r"{}\resources\detail_temp\huilima\1".format(os.getcwd())

    def __init__(self, goods_no, goods_no_value: dict, out_put_dir, windows=None, test=False, excel_data=None,
                 assigned_page_list=None,**kwargs):
        super().__init__(goods_no, goods_no_value, out_put_dir,windows=windows)
        self.root = r"{}\resources\detail_temp\huilima\1".format(os.getcwd())
        self.run()

    def run(self):
        detailed_images = self.deal_details()

        self.create_folder(self.out_put_dir)
        detail_path = "{out_put_dir}/{goods_no}/details".format(out_put_dir=self.out_put_dir, goods_no=self.goods_no)
        self.create_folder(detail_path)
        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)

        for index, pp in enumerate(detailed_images):
            pp.im.save("{}/{}（{}）.jpg".format(detail_path, self.goods_no, str(index + 11).zfill(2)), format="JPEG")

        img = self.add_pic(detailed_images)
        # img.show()
        img.save("{}/1.jpg".format(join_path), format="JPEG")

        # ------------移动其他图片---------------------
        self.move_other_pic()
        return True

    def deal_details(self):
        bg_color = (246, 246, 246)
        self.image_init(bg_color)
        detailed_images = []
        detailed_images.append(self.deal_pic_1())
        # detailed_images.append(self.deal_pic_2())
        # detailed_images.append(self.deal_pic_3())
        # detailed_images.append(self.deal_pic_4())
        # detailed_images.append(self.deal_pic_5())
        return [x for x in detailed_images if x]

    def deal_pic_1(self):
        bg_color = (255, 255, 255)
        goods_art_no_list = list(self.data.keys())
        page_width = 1200
        baseWidth = page_width / 1.4
        pp_list_1 = []
        for index, goods_art_no in enumerate(goods_art_no_list):
            pp_jpg1, pp_png1 = self.image_one_pic(
                goods_art_no=goods_art_no, name="俯视", return_orign=True
            )
            pp_jpg1 = pp_jpg1.resize(value=baseWidth)
            pp_png1 = pp_png1.resize(value=baseWidth)
            maxHeight = pp_jpg1.height
            overlay_bg1 = PictureProcessing(
                "RGB",
                (
                    pp_jpg1.width,
                    maxHeight,
                ),
                bg_color,
            )
            overlay_bg1 = overlay_bg1.to_overlay_pic_advance(
                mode="pixel",
                top_img=pp_jpg1,
                base="cc",
                value=(0, 0),
                top_png_img=pp_png1,
            )
            pp_list_1.append(overlay_bg1)
        _pp_jpg = PictureProcessing()
        rows = 1
        _pp_jpg = _pp_jpg.horizontal_distribution(
            pp_list=pp_list_1,
            bg_width=page_width,
            margins=(0, 0, 0, 0),
            line_spacing=100,
            number_per_row=rows,
        )
        overlay_top = PictureProcessing(
            "RGB",
            (
                page_width,
                100,
            ),
            (255, 255, 255),
        )
        overlay_bottom = PictureProcessing(
            "RGB",
            (
                page_width,
                100,
            ),
            (255, 255, 255),
        )
        detailed_images = []
        # 其他图片
        detailed_images.append(overlay_top)
        detailed_images.append(_pp_jpg)
        detailed_images.append(overlay_bottom)
        return PictureProcessing(im=self.add_pic(detailed_images))

    def deal_pic_2(self):
        # 尺码表
        pp_1 = PictureProcessing(r"{}\image (2).jpg".format(self.root))
        return pp_1

    def deal_pic_3(self):
        # 细节图展示
        bg_color = (246, 246, 246)

        detailed_images = []
        goods_art_no_list = list(self.data.keys())
        view_dict = {"俯视": {"crop_img_value": (-77, 401, 1086, 752),
                            "crop_img_base": "sw",
                            "paste_img_value": (0, 0),
                            "paste_img_base": "center",
                            },
                     "侧视": {"crop_img_value": (0, 0, 1086, 752),
                            "crop_img_base": "sw",
                            "paste_img_value": (0, 0),
                            "paste_img_base": "center",
                            },
                     }
        for index, goods_art_no in enumerate(goods_art_no_list):
            if index == 0:
                name_list = ["俯视", "侧视", "鞋底", "内里"]
            else:
                name_list = ["俯视", "侧视", ]

            # 第一个颜色
            if "俯视" in name_list:
                _pp = self.image_one_pic(goods_art_no, "俯视", bg_color=bg_color)
                if _pp is None:
                    continue

                bg_img = PictureProcessing("RGBA", (1200, 848), (255, 255, 255))
                _pp = (_pp.resize(value=1900)
                       .crop_img(mode="pixel", base="sw", value=(-77, 0, 1086, 752), color_fill=bg_color)
                       .radius(circular_pos=(1, 1, 1, 1), mode="relative", value=100)
                       .paste_img_invert(mode="pixel", top_img=bg_img, base="center")
                       )
                detailed_images.append(_pp)

            if "侧视" in name_list:
                _pp = self.image_one_pic(goods_art_no, "侧视", bg_color=bg_color)
                if _pp is None:
                    continue

                bg_img = PictureProcessing("RGBA", (1200, 848), (255, 255, 255))
                _pp = (_pp.resize(value=1733)
                       .crop_img(mode="pixel", base="wc", value=(715, 0, 1086, 752), color_fill=bg_color)
                       .radius(circular_pos=(1, 1, 1, 1), mode="relative", value=100)
                       .paste_img_invert(mode="pixel", top_img=bg_img, base="center")
                       )
                detailed_images.append(_pp)

            if "鞋底" in name_list:
                _pp = self.image_one_pic(goods_art_no, "鞋底", bg_color=bg_color)
                if _pp is None:
                    continue

                bg_img = PictureProcessing("RGBA", (1200, 848), (255, 255, 255))
                _pp = (_pp.resize(value=1733)
                       .crop_img(mode="pixel", base="wc", value=(715, 0, 1086, 752), color_fill=bg_color)
                       .radius(circular_pos=(1, 1, 1, 1), mode="relative", value=100)
                       .paste_img_invert(mode="pixel", top_img=bg_img, base="center")
                       )
                detailed_images.append(_pp)

            if "内里" in name_list:
                _pp = self.image_one_pic(goods_art_no, "内里", bg_color=bg_color)
                if _pp is None:
                    continue

                bg_img = PictureProcessing("RGBA", (1200, 848), (255, 255, 255))
                _pp = (_pp.resize(value=1663)
                       .crop_img(mode="pixel", base="wc", value=(-40, 0, 1086, 752), color_fill=bg_color)
                       .radius(circular_pos=(1, 1, 1, 1), mode="relative", value=100)
                       .paste_img_invert(mode="pixel", top_img=bg_img, base="center")
                       )
                detailed_images.append(_pp)

        return PictureProcessing(im=self.add_pic(detailed_images))

    def deal_pic_4(self):
        # 制作角度展示图
        bg_color = (246, 246, 246)

        detailed_images = []

        goods_art_no_list = list(self.data.keys())
        for index, goods_art_no in enumerate(goods_art_no_list):
            if index == 0:
                name_list = ["俯视", "侧视", "后跟", "鞋底"]
            else:
                name_list = ["俯视", ]

            for _name in name_list:
                # 处理图片，需要粘贴到背景等处理
                if _name not in self.data[goods_art_no]["pic_is_deal"]:
                    continue
                _pp = self.data[goods_art_no]["pic_is_deal"][_name]
                _pp = _pp.resize(mode="pixel", base="width", value=600 if _name == "后跟" else 1058, )

                bg_pp = PictureProcessing("RGB", (1200, int(_pp.height + 50)), bg_color)
                bg_pp = bg_pp.paste_img(mode="pixel", top_img=_pp, base="cc", value=(0, 0))
                detailed_images.append(bg_pp)
                # 分割线
                _p = PictureProcessing("RGB", (1200, 50), (255, 255, 255))
                detailed_images.append(_p)

        return PictureProcessing(im=self.add_pic(detailed_images))

    def deal_pic_5(self):
        return PictureProcessing(r"{}\image (3).jpg".format(self.root))