CameraMachine/python/custom_plugins/plugins/detail_template/hongqingting/detail_hongqingting1.py

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

# 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
is_test_plugins = False
try:
    is_test_plugins = settings.is_test_plugins
except:
    is_test_plugins = False

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

plugins_name = "详情模板"
company_name_list = ["红蜻蜓",]
template_name = "hongqingt-1"


class DetailPicGet(DetailBase):
    need_view = ["俯视", "侧视", "后跟", "鞋底", "内里"]
    root = r"{}\resources\detail_temp\hongqingting\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):

        super().__init__(goods_no, goods_no_value, out_put_dir, windows=windows, excel_data=excel_data,
                         assigned_page_list=assigned_page_list)
        self.template_name = template_name
        self.root = r"{}\resources\detail_temp\hongqingting\1".format(os.getcwd())
        print("run hongqingt-1 ")
        self.base_bg_color = (228, 196, 147)
        bg_color = (246, 246, 246)
        self.image_init(bg_color)

        self.deal_pic_func_list = [
            self.deal_pic_1,
            self.deal_pic_2,
            self.deal_pic_3,
            self.deal_pic_4,
            self.deal_pic_5,
        ]
        self.run_all()

    def deal_pic_1(self):
        bg_color = (246, 246, 246)
        """   制作主图  """
        goods_art_no_list = list(self.data.keys())

        pp_1 = self.data[goods_art_no_list[0]]["pic_is_deal"]["侧视"]
        pp_image = pp_1.resize(mode="relative",
                               base="by_im",
                               value=0,
                               base_im=PictureProcessing("RGB", (550, 340), (248, 248, 248)),
                               percentage=0)

        pp_image = PictureProcessing("RGB", (550, 340), bg_color).paste_img(
            mode="pixel", top_img=pp_image, base="center"
        )

        bg_img = PictureProcessing(r"{}\image (1).jpg".format(self.root)).paste_img(
            mode="pixel", top_img=pp_image, base="nw", value=(26, 263))
        # bg_img.show()

        #   ---------- 第二张图粘贴
        pp_2 = self.data[goods_art_no_list[0]]["pic_is_deal"]["俯视"]
        pp_image = pp_2.resize(mode="relative",
                               base="by_im",
                               value=0,
                               base_im=PictureProcessing("RGB", (550, 340), (248, 248, 248)),
                               percentage=0)
        pp_image = PictureProcessing("RGB", (550, 340), bg_color).paste_img(
            mode="pixel", top_img=pp_image, base="center"
        )
        bg_img = bg_img.paste_img(
            mode="pixel", top_img=pp_image, base="nw", value=(26, 626))

        # -------粘贴文字-------
        font = ImageFont.truetype(r'resources\ttf\simhei.ttf', 30)
        text_list = [("商品款号", self.goods_no_value["款号"], 700, 297),
                     ("商品面料", self.goods_no_value["商品面料"], 700, 380),
                     ("商品内里", self.goods_no_value["商品内里"], 700, 469),
                     ("商品鞋底", self.goods_no_value["商品鞋底"], 700, 549),
                     ("后帮高", self.goods_no_value["后帮高"], 700, 667),
                     ("前掌宽", self.goods_no_value["前掌宽"], 700, 751),
                     ("鞋跟高", self.goods_no_value["鞋跟高"], 700, 828), ]
        for i in text_list:
            bg_img = bg_img.add_text(mode="pixel",
                                     value=(i[2], i[3]),
                                     font=font,
                                     text="{}:{}".format(i[0], i[1]),
                                     align="center",
                                     spacing=10,
                                     fill=(17, 16, 16))
        return bg_img

    def deal_pic_2(self):
        # 设计师说展示图
        # 设计师说
        if "FAB说明" not in self.goods_no_value:
            return
        text = self.goods_no_value["FAB说明"]  # FAB说明
        if not text:
            return
        pp_1 = PictureProcessing(r"{}\image (2).jpg".format(self.root))

        text = text.replace(r"\n", "\n")
        _ = text.split("\n")
        text_img = PictureProcessing("RGB", (1200, len(_) * 100 + 30), (255, 255, 255))
        font = ImageFont.truetype(r'resources\ttf\simhei.ttf', 30)

        text_img = text_img.add_text(mode="pixel",
                                     value=(600, 50),
                                     font=font,
                                     text=text,
                                     anchor="ma",
                                     align="center",
                                     spacing=50,
                                     fill=(83, 83, 83))
        new_image = PictureProcessing("RGB", (1200, pp_1.height + text_img.height), (255, 255, 255))
        new_image = new_image.paste_img(mode="pixel", top_img=pp_1, value=(0, 0))
        new_image = new_image.paste_img(mode="pixel", top_img=text_img, value=(0, pp_1.height))
        return new_image

    def deal_pic_3(self):
        # 制作角度展示图
        bg_color = (246, 246, 246)
        detailed_images = []
        pp_1 = PictureProcessing(r"{}\image (3).jpg".format(self.root))
        detailed_images.append(pp_1)
        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=649 if _name == "后跟" else 1058, )
                bg_img = PictureProcessing("RGBA", (1200, int(_pp.height + 50)), bg_color)
                bg_img = bg_img.paste_img(mode="pixel", top_img=_pp, base="center")

                detailed_images.append(bg_img)
                # 添加分割线
                detailed_images.append(PictureProcessing("RGB", (1200, 50), (255, 255, 255)))

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

    def deal_pic_4(self):
        # 制作详细图
        bg_color = (246, 246, 246)
        detailed_images = []
        pp_1 = PictureProcessing(r"{}\image (4).jpg".format(self.root))
        detailed_images.append(pp_1)
        name_list = ["俯视", "内里", "后跟", "鞋底"]
        goods_art_no = list(self.data.keys())[0]

        view_dict = {"俯视": (0, 0, 1100, 1200),
                     "内里": (-1, -100, 1100, 1200),
                     "后跟": (0, -100, 1100, 1200),
                     "鞋底": (0, -100, 1100, 1200),
                     }

        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=1300 if _name == "后跟" else 2200, )
            _pp = _pp.crop_img(mode="pixel", base="sw", value=view_dict[_name], color_fill=bg_color)
            _pp = _pp.radius(mode="relative", circular_pos=(1, 0, 0, 1), value=100)
            bg_pp = PictureProcessing("RGB", (1200, 1316), (255, 255, 255))
            bg_pp.paste_img(mode="pixel", top_img=_pp, base="ec", value=(0, 0))
            detailed_images.append(bg_pp)

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

    def deal_pic_5(self):
        # 其他图片添加
        pp_1 = PictureProcessing(r"{}\image (5).jpg".format(self.root))
        return pp_1