|
|
@@ -0,0 +1,1004 @@
|
|
|
+import settings
|
|
|
+from module.view_control.generate_goods_no_detail_pic import detail_func
|
|
|
+import json
|
|
|
+from base import *
|
|
|
+from import_qt_mode import *
|
|
|
+from module.view_control.match_and_cutout_mode_control.base_deal_image_v2 import BaseDealImage
|
|
|
+from module.view_control.MineQWidget import DialogShow, WorkerOneThread
|
|
|
+import threading
|
|
|
+from concurrent.futures import ThreadPoolExecutor
|
|
|
+import concurrent.futures
|
|
|
+from module.view_control.generate_goods_no_detail_pic.data import DataModeGenerateDetail, DataModeUploadPic
|
|
|
+from module.view_control.generate_goods_no_detail_pic.detail_func import create_folder
|
|
|
+import time
|
|
|
+from PIL import Image
|
|
|
+from io import BytesIO
|
|
|
+import os, re
|
|
|
+from functools import partial
|
|
|
+# from multiprocessing import Process, Queue
|
|
|
+import pickle
|
|
|
+from base_deal import BaseDealImage
|
|
|
+
|
|
|
+
|
|
|
+class RunMain(QThread):
|
|
|
+ run_end_sign = Signal(dict)
|
|
|
+ show_dialog_sign = Signal(dict)
|
|
|
+ show_progress_detail_sign = Signal(str)
|
|
|
+
|
|
|
+ # 定义一个信号用于请求显示对话框
|
|
|
+ show_dialog_signal = Signal()
|
|
|
+ # 定义一个信号用于返回对话框的结果
|
|
|
+ dialog_result_signal = Signal(str)
|
|
|
+
|
|
|
+ # dialog_result_signal = Signal(str)
|
|
|
+
|
|
|
+ def __init__(self, windows):
|
|
|
+ super().__init__()
|
|
|
+ self.windows = windows
|
|
|
+ self.dialog_result = ""
|
|
|
+ self.data_mode_generate_detail = DataModeGenerateDetail()
|
|
|
+ self.dialog_result_signal.connect(self.on_dialog_result)
|
|
|
+ self.event = threading.Event()
|
|
|
+
|
|
|
+ def set_state(self, state_value: int):
|
|
|
+ # 0禁用 1进行中 2已结束
|
|
|
+ if state_value not in [0, 1, 2, 99]:
|
|
|
+ return
|
|
|
+ # self.windows.change_state_sign.emit(state_value)
|
|
|
+ # self.windows.set_state(state_value)
|
|
|
+
|
|
|
+ # 抠图前先做数据规整处理；类似详情图生成逻辑
|
|
|
+ def check_before_cutout(self, config_data):
|
|
|
+ return_data = {
|
|
|
+ "code": 99,
|
|
|
+ "message": "",
|
|
|
+ "data": {
|
|
|
+ "all_goods_art_no_folder_data": [],
|
|
|
+ "config_data": config_data,
|
|
|
+ },
|
|
|
+ }
|
|
|
+
|
|
|
+ image_dir = config_data["image_dir"]
|
|
|
+ image_order = config_data["image_order"]
|
|
|
+ is_check_number = config_data["is_check_number"]
|
|
|
+ is_filter = config_data["cutout_is_pass"]
|
|
|
+ resize_image_view = config_data["resize_image_view"]
|
|
|
+ logo_path = config_data["logo_path"]
|
|
|
+ cutout_mode = config_data["cutout_mode"] # 是否精细化抠图，默认为普通抠图
|
|
|
+ special_goods_art_no_folder_line = config_data["special_goods_art_no_folder_line"]
|
|
|
+
|
|
|
+ # 自动处理红蜻蜓货号，进行重命名
|
|
|
+ if settings.PROJECT == "红蜻蜓":
|
|
|
+ # 规整红蜻蜓货号图
|
|
|
+ all_goods_art_no_folder_data = get_all_goods_art_no_folder(path=image_dir)
|
|
|
+ BaseDealImage().rename_folder_for_hqt(all_goods_art_no_folder_data=all_goods_art_no_folder_data)
|
|
|
+
|
|
|
+ # 重新获取文件夹信息
|
|
|
+ all_goods_art_no_folder_data = get_all_goods_art_no_folder(path=image_dir)
|
|
|
+
|
|
|
+ f = True
|
|
|
+ is_do_other = False
|
|
|
+ if is_do_other:
|
|
|
+ for i in all_goods_art_no_folder_data:
|
|
|
+ i["label"] = "不处理"
|
|
|
+ is_filter = False
|
|
|
+ specified_goods_art_no_folder = special_goods_art_no_folder_line
|
|
|
+ specified_goods_art_no_folder = specified_goods_art_no_folder.strip()
|
|
|
+ specified_goods_art_no_folder = specified_goods_art_no_folder.replace("，", ",")
|
|
|
+ specified_goods_art_no_folder_list = specified_goods_art_no_folder.split(",")
|
|
|
+ specified_goods_art_no_folder_list = [x for x in specified_goods_art_no_folder_list if x]
|
|
|
+ if not specified_goods_art_no_folder_list:
|
|
|
+ return_data["message"] += '请手动输入文件夹名称（多选），或关闭指定文件夹模式\n'
|
|
|
+ else:
|
|
|
+ for i in all_goods_art_no_folder_data:
|
|
|
+ if i["folder_path"] in specified_goods_art_no_folder_list:
|
|
|
+ i["label"] = "待处理"
|
|
|
+ # 哪些数据不合规
|
|
|
+ all_folder_name_list = [x["folder_name"] for x in all_goods_art_no_folder_data]
|
|
|
+ for goods_art_no_folder_name in specified_goods_art_no_folder_list:
|
|
|
+ if goods_art_no_folder_name not in all_folder_name_list:
|
|
|
+ return_data["message"] += '文件夹：{}，在您选的目录下不存在\n'.format(goods_art_no_folder_name)
|
|
|
+ f = False
|
|
|
+
|
|
|
+ if not f:
|
|
|
+ self.set_state(state_value=2)
|
|
|
+ return
|
|
|
+
|
|
|
+ # 清空指定文件夹的已抠图文件
|
|
|
+ if is_do_other:
|
|
|
+ for folder_data in all_goods_art_no_folder_data:
|
|
|
+ goods_art_no_folder_path = "{}/原始图_已抠图".format(folder_data["folder_path"])
|
|
|
+ if os.path.exists(goods_art_no_folder_path):
|
|
|
+ remove_all_file(goods_art_no_folder_path)
|
|
|
+
|
|
|
+ return_data["data"]["succeed_folder_list"] = 1
|
|
|
+
|
|
|
+ # ==================检查填写的图片视角是否符合要求
|
|
|
+ res = BaseDealImage().getImageOrder(image_order=image_order, resize_image_view=resize_image_view)
|
|
|
+ if res['code'] != 0:
|
|
|
+ return_data["message"] += "{}\n".format(res['msg'])
|
|
|
+ return return_data
|
|
|
+ else:
|
|
|
+ # 图片命名顺序
|
|
|
+ image_order_list = res['imageOrderList']
|
|
|
+ for goods_art_no_folder_data in all_goods_art_no_folder_data:
|
|
|
+ if goods_art_no_folder_data["label"] != "待处理":
|
|
|
+ continue
|
|
|
+ goods_art_no_folder_data["image_order_list"] = image_order_list
|
|
|
+
|
|
|
+ # ================是否过滤已有生成的文件夹
|
|
|
+ if is_filter:
|
|
|
+ for goods_art_no_folder_data in all_goods_art_no_folder_data:
|
|
|
+ if goods_art_no_folder_data["label"] != "待处理":
|
|
|
+ continue
|
|
|
+ folder_path = goods_art_no_folder_data["folder_path"]
|
|
|
+ _p = "{}/800x800".format(folder_path)
|
|
|
+ if os.path.exists(_p):
|
|
|
+ if len(os.listdir(_p)):
|
|
|
+ goods_art_no_folder_data["label"] = "不处理"
|
|
|
+
|
|
|
+ # ================检查每个货号文件夹图片数量是否符合要求
|
|
|
+ all_goods_art_no_folder_data, message = BaseDealImage().check_folders_image_amount(all_goods_art_no_folder_data,
|
|
|
+ image_order_list)
|
|
|
+ if message:
|
|
|
+ return_data["message"] += "{}\n".format(message)
|
|
|
+ return_data["code"] = 0
|
|
|
+ return_data["data"]["all_goods_art_no_folder_data"] = all_goods_art_no_folder_data
|
|
|
+ return_data["data"]["image_dir"] = image_dir
|
|
|
+ return_data["data"]["resize_image_view"] = resize_image_view
|
|
|
+ return_data["data"]["cutout_mode"] = cutout_mode
|
|
|
+ return_data["data"]["image_order_list"] = image_order_list
|
|
|
+ return_data["data"]["logo_path"] = logo_path
|
|
|
+ return return_data
|
|
|
+
|
|
|
+ # 抠图校验后的回调函数处理
|
|
|
+ def check_for_cutout_image_first_call_back(self, return_data):
|
|
|
+ # return_data = {
|
|
|
+ # "code": 99,
|
|
|
+ # "message": "",
|
|
|
+ # "data": {
|
|
|
+ # "all_goods_art_no_folder_data": [],
|
|
|
+ # },
|
|
|
+ # }
|
|
|
+ code = return_data["code"]
|
|
|
+ config_data = return_data["data"]["config_data"]
|
|
|
+ config_data["sign_text"] = ""
|
|
|
+ # if code != 0:
|
|
|
+ # # self.windows.show_message(return_data["message"])
|
|
|
+ # # self.show_progress_detail(return_data["message"])
|
|
|
+ # _dialog_dict = {"text": return_data["message"],
|
|
|
+ # "windows": self,
|
|
|
+ # }
|
|
|
+ # self.show_dialog_sign.emit(_dialog_dict)
|
|
|
+ # self.event.wait()
|
|
|
+ # return
|
|
|
+
|
|
|
+ do_next = False
|
|
|
+ text = ""
|
|
|
+ all_goods_art_no_folder_data = return_data["data"]["all_goods_art_no_folder_data"]
|
|
|
+ button_1, button_2, button_3 = None, None, None
|
|
|
+ text += return_data["message"]
|
|
|
+ # 存在错误文件夹
|
|
|
+ error_folder = [x for x in all_goods_art_no_folder_data if x["label"] == "错误"]
|
|
|
+ todo_folder = [x for x in all_goods_art_no_folder_data if x["label"] == "待处理"]
|
|
|
+ if error_folder:
|
|
|
+ button_2 = "移除错误文件"
|
|
|
+ if error_folder and todo_folder:
|
|
|
+ button_1 = "移除并继续"
|
|
|
+ button_3 = "继续(忽略其他)"
|
|
|
+ if not error_folder and todo_folder:
|
|
|
+ button_1 = "继续"
|
|
|
+ # do_next = True
|
|
|
+ text += "\n==================\n错误数据：{}个，校验无误数据：{}个".format(len(error_folder), len(todo_folder))
|
|
|
+ self.show_progress_detail(text)
|
|
|
+ if button_1 is None and button_2 is None and button_3 is None:
|
|
|
+ pass
|
|
|
+ elif button_1 == "继续" and button_2 is None and button_3 is None:
|
|
|
+ do_next = True
|
|
|
+ else:
|
|
|
+ # print("runmain 179----------------")
|
|
|
+ # # print(self)
|
|
|
+ # # print(self.windows)
|
|
|
+ # _dialog_dict = {"text": text,
|
|
|
+ # "button_1": button_1,
|
|
|
+ # "button_2": button_2,
|
|
|
+ # "button_3": button_3,
|
|
|
+ # "windows": self,
|
|
|
+ # }
|
|
|
+ # self.show_dialog_sign.emit(_dialog_dict)
|
|
|
+ # # self.exec_()
|
|
|
+ # # 等待事件被设置
|
|
|
+ # self.event.wait()
|
|
|
+ # print("self.dialog_result", self.dialog_result)
|
|
|
+ # #
|
|
|
+ # # my_dialog = DialogShow(self.windows, text=text, button_1=button_1, button_2=button_2,
|
|
|
+ # # button_3=button_3)
|
|
|
+ # # ret = my_dialog.exec()
|
|
|
+ # print("460 ===============my_dialog.flag_name===============")
|
|
|
+ # print(my_dialog.flag_name)
|
|
|
+
|
|
|
+ if "移除" in self.dialog_result:
|
|
|
+ for error_folder_data in [x for x in all_goods_art_no_folder_data if x["label"] == "错误"]:
|
|
|
+ self.move_error_folders(
|
|
|
+ one_path=error_folder_data["folder_path"],
|
|
|
+ target_folder="{}/软件-处理失败".format(config_data["image_dir"]),
|
|
|
+ )
|
|
|
+ if "继续" in self.dialog_result:
|
|
|
+ do_next = True
|
|
|
+
|
|
|
+ if do_next:
|
|
|
+ all_goods_art_no_folder_data = [x for x in all_goods_art_no_folder_data if x["label"] == "待处理"]
|
|
|
+ print("===============all_goods_art_no_folder_data===============")
|
|
|
+ print(all_goods_art_no_folder_data)
|
|
|
+
|
|
|
+ new_func = partial(self.do_run_cutout_image,
|
|
|
+ all_goods_art_no_folder_data=all_goods_art_no_folder_data,
|
|
|
+ callback_func=self.show_progress_detail,
|
|
|
+ image_order_list=return_data["data"]["image_order_list"],
|
|
|
+ cutout_mode=return_data["data"]["cutout_mode"],
|
|
|
+ resize_image_view=return_data["data"]["resize_image_view"],
|
|
|
+ windows=self.windows,
|
|
|
+ logo_path=return_data["data"]["logo_path"],
|
|
|
+ config_data=config_data)
|
|
|
+ self._w_3 = WorkerOneThread(func=new_func, name="_w_3")
|
|
|
+ self._w_3.start()
|
|
|
+
|
|
|
+ # self.t = threading.Thread(target=self.do_run_cutout_image,
|
|
|
+ # kwargs={"all_goods_art_no_folder_data": all_goods_art_no_folder_data,
|
|
|
+ # "callback_func": self.show_progress_detail,
|
|
|
+ # "image_order_list": return_data["data"]["image_order_list"],
|
|
|
+ # "cutout_mode": return_data["data"]["cutout_mode"],
|
|
|
+ # "resize_image_view": return_data["data"]["resize_image_view"],
|
|
|
+ # "windows": self.windows,
|
|
|
+ # "logo_path": return_data["data"]["logo_path"],
|
|
|
+ # "config_data": config_data,
|
|
|
+ # })
|
|
|
+ # self.t.start()
|
|
|
+
|
|
|
+ else:
|
|
|
+ config_data["sign_text"] = "已结束抠图处理"
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def do_run_cutout_image(self,
|
|
|
+ all_goods_art_no_folder_data,
|
|
|
+ callback_func,
|
|
|
+ image_order_list,
|
|
|
+ cutout_mode,
|
|
|
+ resize_image_view,
|
|
|
+ windows,
|
|
|
+ logo_path,
|
|
|
+ config_data):
|
|
|
+
|
|
|
+ BaseDealImage().run_main(all_goods_art_no_folder_data=all_goods_art_no_folder_data,
|
|
|
+ callback_func=callback_func,
|
|
|
+ image_order_list=image_order_list,
|
|
|
+ cutout_mode=cutout_mode,
|
|
|
+ resize_image_view=resize_image_view,
|
|
|
+ windows=windows,
|
|
|
+ logo_path=logo_path,
|
|
|
+ )
|
|
|
+
|
|
|
+ # ==============完成处理==============
|
|
|
+ # self.set_state(state_value=2)
|
|
|
+ callback_func("已结束抠图处理")
|
|
|
+ config_data["sign_text"] = "已结束抠图处理"
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def do_run_cutout_image1111(self, all_goods_art_no_folder_data,
|
|
|
+ callback_func,
|
|
|
+ image_order_list,
|
|
|
+ cutout_mode,
|
|
|
+ resize_image_view,
|
|
|
+ windows,
|
|
|
+ logo_path,
|
|
|
+ config_data):
|
|
|
+
|
|
|
+ max_workers = 1
|
|
|
+ with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
|
|
|
+ futures = []
|
|
|
+ futures.append(executor.submit(
|
|
|
+ BaseDealImage().run_main,
|
|
|
+ all_goods_art_no_folder_data=all_goods_art_no_folder_data,
|
|
|
+ callback_func=callback_func,
|
|
|
+ image_order_list=image_order_list,
|
|
|
+ cutout_mode=cutout_mode,
|
|
|
+ resize_image_view=resize_image_view,
|
|
|
+ windows=windows,
|
|
|
+ logo_path=logo_path,
|
|
|
+ ))
|
|
|
+
|
|
|
+ # 使用 wait 方法等待所有任务完成
|
|
|
+ done, not_done = concurrent.futures.wait(futures, timeout=60)
|
|
|
+ # 处理完成的任务
|
|
|
+ for future in done:
|
|
|
+ if settings.IS_TEST:
|
|
|
+ result = future.result()
|
|
|
+ else:
|
|
|
+ try:
|
|
|
+ result = future.result()
|
|
|
+ except BaseException as e:
|
|
|
+ print("2039 图片处理失败.{}".format(e))
|
|
|
+ callback_func("2039 处理失败.{}".format(e))
|
|
|
+
|
|
|
+ # ==============完成处理==============
|
|
|
+
|
|
|
+ # self.set_state(state_value=2)
|
|
|
+ callback_func("已结束抠图处理")
|
|
|
+ config_data["sign_text"] = "已结束抠图处理"
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def check_before_detail(self, config_data):
|
|
|
+
|
|
|
+ # =============
|
|
|
+ # 整体数据校验，返回错误内容，以及
|
|
|
+ # temp_name:模板名称
|
|
|
+ """
|
|
|
+ 步骤：
|
|
|
+ 1、 整体文件夹检查，并输出数据结构
|
|
|
+ 2、数据进行对应规整（可能有excel，红蜻蜓等）
|
|
|
+ 3、执行单个款数据处理
|
|
|
+ """
|
|
|
+ return_data = {
|
|
|
+ "code": 99,
|
|
|
+ "message": "",
|
|
|
+ "data": {
|
|
|
+ "error_folder_list": [],
|
|
|
+ "goods_no_dict": {},
|
|
|
+ "succeed_folder_list": [],
|
|
|
+ "temp_name": config_data["temp_name"],
|
|
|
+ "temp_name_list": config_data["temp_name_list"],
|
|
|
+ "assigned_page_dict": config_data["assigned_page_dict"],
|
|
|
+ "excel_temp_goods_no_data": {}, # 表格数据可能存在多模板，数据结构为一个款号下的多个模板的数据列表
|
|
|
+ "finally_goods_no_need_temps": {}, # 每个款号需要生成的模板数据
|
|
|
+ "config_data": config_data,
|
|
|
+ },
|
|
|
+ }
|
|
|
+ temp_name = config_data["temp_name"]
|
|
|
+ temp_name_list = config_data["temp_name_list"]
|
|
|
+ assigned_page_dict = config_data["assigned_page_dict"]
|
|
|
+ image_dir = config_data["image_dir"]
|
|
|
+ is_use_excel = config_data["is_use_excel"]
|
|
|
+ excel_path = config_data["excel_path"]
|
|
|
+ temp_class = config_data["temp_class"]
|
|
|
+ is_check_color_is_all = config_data["is_check_color_is_all"]
|
|
|
+ detail_is_pass = config_data["detail_is_pass"]
|
|
|
+
|
|
|
+ error_folder_list = []
|
|
|
+ # 遍历货号获取所有货号--可能为编号
|
|
|
+ folder_name_list = detail_func.get_all_dir_info(image_dir=image_dir)
|
|
|
+ if not folder_name_list:
|
|
|
+ return_data["message"] += "不存在任何货号/编号文件夹\n"
|
|
|
+ print("不存在任何货号/编号文件夹")
|
|
|
+ return return_data
|
|
|
+
|
|
|
+ # =========================组装数据---数据来源多种途径=========================
|
|
|
+ _result = {"code": 99, "message": "无法解析到数据，请检查登录企业"}
|
|
|
+ if not is_use_excel:
|
|
|
+ if settings.PROJECT == "红蜻蜓":
|
|
|
+ # goods_no_dict输出为文件夹下涉及到的所有款为key的字典，后续通过解析字典，进行提取对应文件夹
|
|
|
+ _result = self.data_mode_generate_detail.get_basic_goods_art_data_by_hqt_and_hlm(
|
|
|
+ folder_name_list
|
|
|
+ )
|
|
|
+
|
|
|
+ elif settings.PROJECT == "惠利玛":
|
|
|
+ if settings.Company:
|
|
|
+ if "惠利玛" in settings.Company:
|
|
|
+ _result = self.data_mode_generate_detail.get_basic_goods_art_data_by_hqt_and_hlm(
|
|
|
+ folder_name_list
|
|
|
+ )
|
|
|
+ else:
|
|
|
+ keys = settings.keys
|
|
|
+ _result = (
|
|
|
+ self.data_mode_generate_detail.get_basic_goods_art_data_form_excel(
|
|
|
+ folder_name_list,
|
|
|
+ excel_path,
|
|
|
+ keys,
|
|
|
+ )
|
|
|
+ )
|
|
|
+
|
|
|
+ if _result["code"] == 0:
|
|
|
+ remote_data = _result["data"]
|
|
|
+ else:
|
|
|
+ return_data["message"] += _result["message"] + "\n"
|
|
|
+ return return_data
|
|
|
+ # print(json.dumps(remote_data))
|
|
|
+ # =========================拼接数据组合为款数据=========================
|
|
|
+ """
|
|
|
+ 1、获取所有文件夹基础数据内容
|
|
|
+ 2、结合上述返回结果进行数据组合拼接
|
|
|
+ 3、输出结果为按款为主键信息
|
|
|
+ 4、注意模板ID
|
|
|
+ """
|
|
|
+ # 获取所有文件夹基础数据内容 检查不满足要求的文件不满足要求移动到错误文件夹
|
|
|
+ need_view_list = temp_class[temp_name].need_view
|
|
|
+ _all_dir_info_data = detail_func.get_all_dir_info_and_pic_info(
|
|
|
+ image_dir, folder_name_list, need_view_list
|
|
|
+ )
|
|
|
+ all_dir_info_data = {}
|
|
|
+ for one_folder, value in _all_dir_info_data.items():
|
|
|
+ if "message" in value:
|
|
|
+ if value["message"]:
|
|
|
+ error_folder_list.append(
|
|
|
+ {
|
|
|
+ "folder_name": one_folder,
|
|
|
+ "folder_path": "{}/{}".format(image_dir, one_folder),
|
|
|
+ }
|
|
|
+ )
|
|
|
+ return_data["message"] += "文件夹：{} 结构错误：{}\n".format(
|
|
|
+ one_folder, value["message"]
|
|
|
+ )
|
|
|
+ continue
|
|
|
+
|
|
|
+ # 符合要求的数据处理
|
|
|
+ all_dir_info_data[one_folder] = value
|
|
|
+
|
|
|
+ # 结合上述返回结果进行数据组合拼接
|
|
|
+ # 返回可能存在的情况：1、存在文件夹不匹配数据，2、存在货号数据缺失
|
|
|
+ goods_no_dict, error_folder_name_list = detail_func.merge_local_and_remote_data(
|
|
|
+ all_dir_info_data=all_dir_info_data, remote_data=remote_data
|
|
|
+ )
|
|
|
+ # 文件在系统中没有匹配的结果
|
|
|
+ if error_folder_name_list:
|
|
|
+ for one_folder in error_folder_name_list:
|
|
|
+ error_folder_list.append(
|
|
|
+ {
|
|
|
+ "folder_name": one_folder,
|
|
|
+ "folder_path": "{}/{}".format(image_dir, one_folder),
|
|
|
+ }
|
|
|
+ )
|
|
|
+ return_data["message"] += "文件夹：{} 找不到对应数据\n".format(
|
|
|
+ one_folder
|
|
|
+ )
|
|
|
+
|
|
|
+ print("===============goods_no_dict==================")
|
|
|
+ if settings.IS_TEST:
|
|
|
+ _text = json.dumps(goods_no_dict)
|
|
|
+ print(goods_no_dict)
|
|
|
+ create_folder("qt_test")
|
|
|
+ with open("qt_test/goods_no_dict.txt", 'w', encoding='utf-8') as file:
|
|
|
+ file.write(_text)
|
|
|
+ print("===============goods_no_dict==================")
|
|
|
+
|
|
|
+ # ===================================是否齐色检查=============================================
|
|
|
+ # 如为红蜻蜓企业则还需要检查同款下是否齐全;数据返回结果为款号列表
|
|
|
+ error_data_dict = {}
|
|
|
+ if is_check_color_is_all:
|
|
|
+ if not is_use_excel:
|
|
|
+ if settings.PROJECT == "红蜻蜓":
|
|
|
+ error_data_dict = (
|
|
|
+ self.data_mode_generate_detail.check_goods_is_not_deficiency(
|
|
|
+ goods_no_dict
|
|
|
+ )
|
|
|
+ )
|
|
|
+ else:
|
|
|
+ error_data_dict = self.data_mode_generate_detail.check_goods_is_not_deficiency_form_excel(
|
|
|
+ goods_no_dict, excel_path
|
|
|
+ )
|
|
|
+
|
|
|
+ if error_data_dict:
|
|
|
+ print("----error_data_dict----")
|
|
|
+ print(json.dumps(error_data_dict))
|
|
|
+
|
|
|
+ # 款号反向映射;因为部分key键格式为KUM9999999
|
|
|
+ _x = {}
|
|
|
+ for i, v in goods_no_dict.items():
|
|
|
+ _x[v["款号"]] = i
|
|
|
+
|
|
|
+ for goods_no, value in error_data_dict.items():
|
|
|
+ if goods_no in _x:
|
|
|
+ goods_no = _x[goods_no]
|
|
|
+ if goods_no in goods_no_dict:
|
|
|
+ # =====移动到错误文件夹======
|
|
|
+ for _folder_name in [
|
|
|
+ x["文件夹名称"] for x in goods_no_dict[goods_no]["货号资料"]
|
|
|
+ ]:
|
|
|
+ error_folder_list.append(
|
|
|
+ {
|
|
|
+ "folder_name": _folder_name,
|
|
|
+ "folder_path": "{}\{}".format(
|
|
|
+ image_dir, _folder_name
|
|
|
+ ),
|
|
|
+ }
|
|
|
+ )
|
|
|
+ return_data["message"] += "文件夹：{}；{}\n".format(
|
|
|
+ _folder_name, value["message"]
|
|
|
+ )
|
|
|
+ # 从 正确 款下面进行数据剔除
|
|
|
+ goods_no_dict.pop(goods_no)
|
|
|
+
|
|
|
+ print("-----------------1goods_no_dict---------------")
|
|
|
+ print(json.dumps(goods_no_dict, ensure_ascii=False))
|
|
|
+ print("-----------------1goods_no_dict---------------")
|
|
|
+
|
|
|
+ # 如果没有有效数据则进行退出
|
|
|
+ if not goods_no_dict:
|
|
|
+ return_data["message"] += "没有任何有效数据\n"
|
|
|
+ return return_data
|
|
|
+
|
|
|
+ # 校验无误的文件夹数据 goods_no_dict为最终有效数据
|
|
|
+ for goods_no, value in goods_no_dict.items():
|
|
|
+ return_data["data"]["succeed_folder_list"].extend(
|
|
|
+ [x["文件夹名称"] for x in goods_no_dict[goods_no]["货号资料"]]
|
|
|
+ )
|
|
|
+
|
|
|
+ # 如果是表格数据，则获取表格数据中需要生成的货号
|
|
|
+ # 数据结构
|
|
|
+ # {“款号1”：
|
|
|
+ # {“模板名称1”：data_dict1,
|
|
|
+ # “模板名称2”：data_dict2,
|
|
|
+ # }}
|
|
|
+
|
|
|
+ if is_use_excel:
|
|
|
+ excel_temp_goods_no_data = self.data_mode_generate_detail.get_basic_template_information(
|
|
|
+ _goods_no_dict=goods_no_dict, excel_path=excel_path)
|
|
|
+ else:
|
|
|
+ excel_temp_goods_no_data = {}
|
|
|
+
|
|
|
+ print("731===================excel_temp_goods_no_data,is_use_excel:{}".format(is_use_excel))
|
|
|
+ print(json.dumps(excel_temp_goods_no_data))
|
|
|
+
|
|
|
+ # ===========数据组装，统计每个款需要生成哪些模板==============
|
|
|
+ # 不适用表格指定模板
|
|
|
+ goods_no_need_temps = {}
|
|
|
+ # isUseTemplate 为false 表示使用excel表格数据
|
|
|
+ if not is_use_excel:
|
|
|
+ for i in goods_no_dict:
|
|
|
+ goods_no_need_temps[i] = [temp_name]
|
|
|
+ else:
|
|
|
+ for i in goods_no_dict:
|
|
|
+ # i 为款号
|
|
|
+ if i in excel_temp_goods_no_data:
|
|
|
+ # 获取 某个款号的所有允许生成的模板列表
|
|
|
+ goods_no_need_temps[i] = list(excel_temp_goods_no_data[i].keys())
|
|
|
+
|
|
|
+ # ========开始执行详情图生成===================
|
|
|
+ # _goods_no_dict = goods_no_dict
|
|
|
+ _goods_no_dict = {}
|
|
|
+
|
|
|
+ # assigned_page_dict 如存在对应模板的，则不管是否有过滤都需要生成
|
|
|
+ # 检查是否已存在模板，已存在的需要进行跳过；可能部分模板已存在，部分不存在。
|
|
|
+ finally_goods_no_need_temps = {}
|
|
|
+ for goods_no, value in goods_no_dict.items():
|
|
|
+ if goods_no not in goods_no_need_temps:
|
|
|
+ continue
|
|
|
+ for __temp_name in goods_no_need_temps[goods_no]:
|
|
|
+ _path = "{}/{}/{}/{}".format(image_dir, "软件-详情图生成", __temp_name, goods_no)
|
|
|
+ if not os.path.exists(_path):
|
|
|
+ print("款号详情图不存在", _path)
|
|
|
+ if goods_no not in finally_goods_no_need_temps:
|
|
|
+ finally_goods_no_need_temps[goods_no] = []
|
|
|
+ _goods_no_dict[goods_no] = value # 需要生成的数据
|
|
|
+ finally_goods_no_need_temps[goods_no].append(__temp_name)
|
|
|
+ else:
|
|
|
+ print("款号详情图存在", _path)
|
|
|
+ # 如果在指定模板中存在，则也需要生成
|
|
|
+ if __temp_name in assigned_page_dict:
|
|
|
+ print("指定模板需要更新", _path)
|
|
|
+ if goods_no not in finally_goods_no_need_temps:
|
|
|
+ finally_goods_no_need_temps[goods_no] = []
|
|
|
+ _goods_no_dict[goods_no] = value # 需要生成的数据
|
|
|
+ finally_goods_no_need_temps[goods_no].append(__temp_name)
|
|
|
+ else:
|
|
|
+ if detail_is_pass:
|
|
|
+ return_data["message"] += "\n款号：{}，模板：{} 已存在".format(goods_no, __temp_name)
|
|
|
+ else:
|
|
|
+ if goods_no not in finally_goods_no_need_temps:
|
|
|
+ finally_goods_no_need_temps[goods_no] = []
|
|
|
+ _goods_no_dict[goods_no] = value # 需要生成的数据
|
|
|
+ finally_goods_no_need_temps[goods_no].append(__temp_name)
|
|
|
+
|
|
|
+ pass
|
|
|
+ # _path = "{}/{}".format(self.image_dir, "软件-详情图生成")
|
|
|
+ # if os.path.exists(_path):
|
|
|
+ # _goods_no_dict = {}
|
|
|
+ # # 数据返回为 已有的款数据，为款号列表
|
|
|
+ # is_pass_goods_no = detail_func.get_all_detail_info(_path)
|
|
|
+ # for goods_no, value in goods_no_dict.items():
|
|
|
+ # if "软件" in goods_no:
|
|
|
+ # continue
|
|
|
+ #
|
|
|
+ # if value["模板名称"] in assigned_page_dict:
|
|
|
+ # need_todo = True
|
|
|
+ # else:
|
|
|
+ # if goods_no in is_pass_goods_no:
|
|
|
+ # need_todo = False
|
|
|
+ # else:
|
|
|
+ # need_todo = True
|
|
|
+ # if need_todo:
|
|
|
+ # _goods_no_dict[goods_no] = value
|
|
|
+
|
|
|
+ print("-----------------2goods_no_dict---------------")
|
|
|
+ print(json.dumps(_goods_no_dict, ensure_ascii=False))
|
|
|
+ print("-----------------2goods_no_dict---------------")
|
|
|
+ return_data["data"]["error_folder_list"] = error_folder_list
|
|
|
+
|
|
|
+ if len(_goods_no_dict) == 0:
|
|
|
+ return_data["message"] += "\n没有任何文件夹需要执行"
|
|
|
+
|
|
|
+ return_data["data"]["goods_no_dict"] = _goods_no_dict
|
|
|
+ return_data["data"]["excel_temp_goods_no_data"] = excel_temp_goods_no_data
|
|
|
+ return_data["data"]["finally_goods_no_need_temps"] = finally_goods_no_need_temps
|
|
|
+
|
|
|
+ return_data["code"] = 0
|
|
|
+ return return_data
|
|
|
+
|
|
|
+ def move_error_folders(self, one_path, target_folder, message=""):
|
|
|
+ if os.path.exists(one_path):
|
|
|
+ check_path(target_folder)
|
|
|
+ move_folders(path_list=[one_path], target_folder=target_folder)
|
|
|
+
|
|
|
+ def check_for_detail_first_call_back(self, data):
|
|
|
+ # 首次数据校验的信息返回
|
|
|
+ # self.show_message(text="22222222222222222222222")
|
|
|
+ # QMessageBox.critical(self, "警告", "1111111", QMessageBox.Ok)
|
|
|
+ code = data["code"]
|
|
|
+ config_data = data["data"]["config_data"]
|
|
|
+ target_error_folder = config_data["target_error_folder"]
|
|
|
+ print("635 check_for_detail_first_call_back")
|
|
|
+ print(data)
|
|
|
+ if code != 0:
|
|
|
+ # self.windows.show_message(data["message"])
|
|
|
+ # my_dialog = DialogShow(self.windows.windows, text=data["message"])
|
|
|
+ # ret = my_dialog.exec()
|
|
|
+ self.show_progress_detail(text=data["message"])
|
|
|
+ _dialog_dict = {"text": data["message"],
|
|
|
+ "windows": self,
|
|
|
+ }
|
|
|
+ self.show_dialog_sign.emit(_dialog_dict)
|
|
|
+ self.event.wait()
|
|
|
+
|
|
|
+ # self.windows.set_state(2)
|
|
|
+ config_data["sign_text"] = "已结束详情处理"
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+ return
|
|
|
+
|
|
|
+ do_next = False
|
|
|
+ if data["message"]:
|
|
|
+ button_1, button_2, button_3 = None, None, None
|
|
|
+ text = data["message"]
|
|
|
+ if code == 0:
|
|
|
+ if data["data"]:
|
|
|
+ if data["data"]["error_folder_list"]:
|
|
|
+ print("22----------error_folder_list------------")
|
|
|
+ print(json.dumps(data["data"]["error_folder_list"]))
|
|
|
+ button_2 = "移除错误文件"
|
|
|
+ if data["data"]["goods_no_dict"]:
|
|
|
+ if button_2:
|
|
|
+ button_1 = "移除并继续"
|
|
|
+ button_3 = "继续(忽略其他)"
|
|
|
+ else:
|
|
|
+ button_1 = "继续"
|
|
|
+ if data["data"]["succeed_folder_list"]:
|
|
|
+ text += "\n==================\n校验无误数据：{}个文件夹".format(
|
|
|
+ len(data["data"]["succeed_folder_list"])
|
|
|
+ )
|
|
|
+ self.show_progress_detail(text=data["message"])
|
|
|
+ if button_1 is None and button_2 is None and button_3 is None:
|
|
|
+ pass
|
|
|
+ elif button_1 == "继续" and button_2 is None and button_3 is None:
|
|
|
+ do_next = True
|
|
|
+ else:
|
|
|
+ print("runmain 642----------------")
|
|
|
+ # todo 弹框修改处理等
|
|
|
+ _dialog_dict = {"text": text,
|
|
|
+ "button_1": button_1,
|
|
|
+ "button_2": button_2,
|
|
|
+ "button_3": button_3,
|
|
|
+ "windows": self,
|
|
|
+ }
|
|
|
+ self.show_dialog_sign.emit(_dialog_dict)
|
|
|
+ # 等待事件被设置
|
|
|
+ self.event.wait()
|
|
|
+ print("self.dialog_result", self.dialog_result)
|
|
|
+
|
|
|
+ # my_dialog = DialogShow(
|
|
|
+ # self.windows, text=text, button_1=button_1, button_2=button_2, button_3=button_3
|
|
|
+ # )
|
|
|
+ print("my_dialog.flag_name", self.dialog_result)
|
|
|
+ if "移除" in self.dialog_result:
|
|
|
+ for error_folder_data in data["data"]["error_folder_list"]:
|
|
|
+ self.move_error_folders(
|
|
|
+ one_path=error_folder_data["folder_path"],
|
|
|
+ target_folder=target_error_folder,
|
|
|
+ )
|
|
|
+ if "继续" in self.dialog_result:
|
|
|
+ do_next = True
|
|
|
+
|
|
|
+ if data["data"]["goods_no_dict"] and not data["data"]["error_folder_list"]:
|
|
|
+ do_next = True
|
|
|
+
|
|
|
+ if do_next:
|
|
|
+ # self.set_state(state_value=1)
|
|
|
+ getAllData = data["data"]
|
|
|
+ base_temp_name = getAllData["temp_name"]
|
|
|
+ set_temp_name = getAllData.get("template_name", "")
|
|
|
+ kwargs = {
|
|
|
+ "config_data": config_data,
|
|
|
+ "_goods_no_dict": data["data"]["goods_no_dict"],
|
|
|
+ "temp_name": base_temp_name,
|
|
|
+ "temp_name_list": data["data"]["temp_name_list"],
|
|
|
+ "assigned_page_dict": data["data"]["assigned_page_dict"],
|
|
|
+ "excel_temp_goods_no_data": data["data"]["excel_temp_goods_no_data"],
|
|
|
+ # 表格数据可能存在多模板，数据结构为一个款号下的多个模板的数据列表
|
|
|
+ "finally_goods_no_need_temps": data["data"]["finally_goods_no_need_temps"], # 每个款号需要生成的模板数据
|
|
|
+ }
|
|
|
+ # todo work
|
|
|
+ new_func = partial(self.detail_run_by_thread,
|
|
|
+ config_data=kwargs["config_data"],
|
|
|
+ _goods_no_dict=kwargs["_goods_no_dict"],
|
|
|
+ temp_name=kwargs["temp_name"],
|
|
|
+ temp_name_list=kwargs["temp_name_list"],
|
|
|
+ assigned_page_dict=kwargs["assigned_page_dict"],
|
|
|
+ excel_temp_goods_no_data=kwargs["excel_temp_goods_no_data"],
|
|
|
+ finally_goods_no_need_temps=kwargs["finally_goods_no_need_temps"])
|
|
|
+ self._w_3 = WorkerOneThread(func=new_func, name="_w_3")
|
|
|
+ self._w_3.start()
|
|
|
+ # threading.Thread(target=self.detail_run_by_thread, kwargs=kwargs).start()
|
|
|
+ else:
|
|
|
+ config_data["sign_text"] = "已结束详情处理"
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def detail_run_by_thread(self, config_data, _goods_no_dict, temp_name, temp_name_list, assigned_page_dict,
|
|
|
+ excel_temp_goods_no_data,
|
|
|
+ finally_goods_no_need_temps):
|
|
|
+ """
|
|
|
+ excel_temp_goods_no_data: {}, # 表格数据可能存在多模板，数据结构为一个款号下的多个模板的数据列表
|
|
|
+ finally_goods_no_need_temps: {}, # 每个款号需要生成的模板数据
|
|
|
+ """
|
|
|
+
|
|
|
+ # 开始处理
|
|
|
+ self.n = 0
|
|
|
+ self.total_num = len(_goods_no_dict)
|
|
|
+ self.fail_num = 0
|
|
|
+ is_use_excel = config_data["is_use_excel"]
|
|
|
+ image_dir = config_data["image_dir"]
|
|
|
+
|
|
|
+ # 详情图生成结果文件夹
|
|
|
+ out_put_dir = "{}\软件-详情图生成".format(image_dir)
|
|
|
+ if settings.IS_TEST:
|
|
|
+ print("==============_goods_no_dict 打印=================")
|
|
|
+
|
|
|
+ print(json.dumps(_goods_no_dict))
|
|
|
+
|
|
|
+ print("==============_goods_no_dict 打印-end=================")
|
|
|
+
|
|
|
+ all_detail_path_list = []
|
|
|
+ for goods_no, temp_name_list in finally_goods_no_need_temps.items():
|
|
|
+ try:
|
|
|
+ for _temp_name in temp_name_list:
|
|
|
+ # if _temp_name != "xiaosushuoxie-4":
|
|
|
+ # continue
|
|
|
+ assigned_page_list = []
|
|
|
+ if _temp_name in assigned_page_dict:
|
|
|
+ assigned_page_list = assigned_page_dict[_temp_name]
|
|
|
+ # 如果为使用表格，则获取表格中的数据作为款号的基础数据
|
|
|
+ temp_info_data = copy.copy(_goods_no_dict[goods_no])
|
|
|
+ if is_use_excel:
|
|
|
+ # 将表格中的数据进行替换
|
|
|
+ if goods_no in excel_temp_goods_no_data:
|
|
|
+ if _temp_name in excel_temp_goods_no_data[goods_no]:
|
|
|
+ # 将表格中的特定的模板的行，替换到goods_no的data中，因为不同的模板有数据特殊性
|
|
|
+ for _key, _key_value in excel_temp_goods_no_data[goods_no][_temp_name].items():
|
|
|
+ if _key in temp_info_data:
|
|
|
+ temp_info_data[_key] = _key_value
|
|
|
+ print("temp_info_data")
|
|
|
+ print("goods_no:{},_temp_name:{}".format(goods_no, _temp_name))
|
|
|
+ all_detail_path_list.append("{}/{}/".format(out_put_dir, _temp_name, goods_no))
|
|
|
+ # continue
|
|
|
+ self.detail_deal_one_data(goods_no=goods_no,
|
|
|
+ value=temp_info_data,
|
|
|
+ out_put_dir=out_put_dir,
|
|
|
+ temp_name=_temp_name,
|
|
|
+ assigned_page_list=assigned_page_list)
|
|
|
+ except BaseException as e:
|
|
|
+ self.show_progress_detail(
|
|
|
+ "款：{}生成详情异常：{}".format(goods_no, e))
|
|
|
+ print(e)
|
|
|
+
|
|
|
+ # ==============完成处理==============
|
|
|
+ self.set_state(state_value=2)
|
|
|
+ if self.total_num:
|
|
|
+ if self.fail_num:
|
|
|
+ self.show_progress_detail("处理完成,-----处理失败数据：{}个款".format(self.fail_num))
|
|
|
+ else:
|
|
|
+ self.show_progress_detail("处理完成")
|
|
|
+ else:
|
|
|
+ self.show_progress_detail("没有任何数据")
|
|
|
+
|
|
|
+ config_data["sign_text"] = "已结束详情处理"
|
|
|
+ config_data["all_detail_path_list"] = all_detail_path_list
|
|
|
+ # 打开文件夹
|
|
|
+ os.startfile(out_put_dir)
|
|
|
+
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def detail_run_by_thread11111(self, config_data, _goods_no_dict, temp_name, temp_name_list, assigned_page_dict,
|
|
|
+ excel_temp_goods_no_data,
|
|
|
+ finally_goods_no_need_temps):
|
|
|
+ """
|
|
|
+ excel_temp_goods_no_data: {}, # 表格数据可能存在多模板，数据结构为一个款号下的多个模板的数据列表
|
|
|
+ finally_goods_no_need_temps: {}, # 每个款号需要生成的模板数据
|
|
|
+ """
|
|
|
+
|
|
|
+ # 开始处理
|
|
|
+ self.n = 0
|
|
|
+ self.total_num = len(_goods_no_dict)
|
|
|
+ self.fail_num = 0
|
|
|
+ is_use_excel = config_data["is_use_excel"]
|
|
|
+ image_dir = config_data["image_dir"]
|
|
|
+
|
|
|
+ # 详情图生成结果文件夹
|
|
|
+ out_put_dir = "{}\软件-详情图生成".format(image_dir)
|
|
|
+ if settings.IS_TEST:
|
|
|
+ print("==============_goods_no_dict 打印=================")
|
|
|
+
|
|
|
+ print(json.dumps(_goods_no_dict))
|
|
|
+
|
|
|
+ print("==============_goods_no_dict 打印-end=================")
|
|
|
+
|
|
|
+ if settings.IS_TEST:
|
|
|
+ max_workers = 1
|
|
|
+ else:
|
|
|
+ max_workers = 1
|
|
|
+
|
|
|
+ all_detail_path_list = []
|
|
|
+
|
|
|
+ with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
|
|
|
+ futures = []
|
|
|
+ for goods_no, temp_name_list in finally_goods_no_need_temps.items():
|
|
|
+ for _temp_name in temp_name_list:
|
|
|
+ # if _temp_name != "xiaosushuoxie-4":
|
|
|
+ # continue
|
|
|
+
|
|
|
+ assigned_page_list = []
|
|
|
+ if _temp_name in assigned_page_dict:
|
|
|
+ assigned_page_list = assigned_page_dict[_temp_name]
|
|
|
+ # 如果为使用表格，则获取表格中的数据作为款号的基础数据
|
|
|
+ temp_info_data = copy.copy(_goods_no_dict[goods_no])
|
|
|
+ if is_use_excel:
|
|
|
+ # 将表格中的数据进行替换
|
|
|
+ if goods_no in excel_temp_goods_no_data:
|
|
|
+ if _temp_name in excel_temp_goods_no_data[goods_no]:
|
|
|
+ # 将表格中的特定的模板的行，替换到goods_no的data中，因为不同的模板有数据特殊性
|
|
|
+ for _key, _key_value in excel_temp_goods_no_data[goods_no][_temp_name].items():
|
|
|
+ if _key in temp_info_data:
|
|
|
+ temp_info_data[_key] = _key_value
|
|
|
+
|
|
|
+ print("temp_info_data")
|
|
|
+ print("goods_no:{},_temp_name:{}".format(goods_no, _temp_name))
|
|
|
+ all_detail_path_list.append("{}/{}/".format(out_put_dir, _temp_name, goods_no))
|
|
|
+ # continue
|
|
|
+ futures.append(executor.submit(
|
|
|
+ self.detail_deal_one_data,
|
|
|
+ goods_no=goods_no,
|
|
|
+ value=temp_info_data,
|
|
|
+ out_put_dir=out_put_dir,
|
|
|
+ temp_name=_temp_name,
|
|
|
+ assigned_page_list=assigned_page_list,
|
|
|
+ ))
|
|
|
+
|
|
|
+ # for goods_no, value in _goods_no_dict.items():
|
|
|
+ # _temp_name = temp_name
|
|
|
+ # # 使用自定义的表格数据
|
|
|
+ # if self.isUseTemplate is False:
|
|
|
+ # if "模板名称" in value:
|
|
|
+ # if value["模板名称"] in temp_name_list:
|
|
|
+ # _temp_name = value["模板名称"]
|
|
|
+ # assigned_page_list = []
|
|
|
+ # if _temp_name in assigned_page_dict:
|
|
|
+ # assigned_page_list = assigned_page_dict[_temp_name]
|
|
|
+ #
|
|
|
+ # futures.append(executor.submit(
|
|
|
+ # self.deal_one_data,
|
|
|
+ # goods_no=goods_no,
|
|
|
+ # value=value,
|
|
|
+ # out_put_dir=out_put_dir,
|
|
|
+ # temp_name=_temp_name,
|
|
|
+ # assigned_page_list=assigned_page_list,
|
|
|
+ # ))
|
|
|
+
|
|
|
+ # 使用 wait 方法等待所有任务完成
|
|
|
+ done, not_done = concurrent.futures.wait(futures)
|
|
|
+
|
|
|
+ # 处理完成的任务
|
|
|
+ for future in done:
|
|
|
+ if settings.IS_TEST:
|
|
|
+ result = future.result()
|
|
|
+
|
|
|
+ # ==============完成处理==============
|
|
|
+ self.set_state(state_value=2)
|
|
|
+ if self.total_num:
|
|
|
+ if self.fail_num:
|
|
|
+ self.show_progress_detail(
|
|
|
+ "处理完成,-----------处理失败数据：{}个款".format(self.fail_num)
|
|
|
+ )
|
|
|
+ else:
|
|
|
+ self.show_progress_detail("处理完成")
|
|
|
+ else:
|
|
|
+ self.show_progress_detail("没有任何数据")
|
|
|
+
|
|
|
+ config_data["sign_text"] = "已结束详情处理"
|
|
|
+ config_data["all_detail_path_list"] = all_detail_path_list
|
|
|
+ self.run_end_sign.emit(config_data)
|
|
|
+
|
|
|
+ def show_progress_detail(self, text):
|
|
|
+ # self.show_progress_detail_sign.emit(text)
|
|
|
+ # self.windows.show_progress_detail(text)
|
|
|
+
|
|
|
+ def detail_deal_one_data(self, goods_no, value, out_put_dir, temp_name, assigned_page_list):
|
|
|
+ if self.windows.state == 99:
|
|
|
+ self.show_progress_detail("用户主动取消：{}".format(goods_no))
|
|
|
+ return
|
|
|
+
|
|
|
+ self.show_progress_detail("正在生成：{}，模板：{}".format(goods_no, temp_name))
|
|
|
+ is_deal_success = False
|
|
|
+ print("=================deal_one_data=====================")
|
|
|
+ print("goods_no", goods_no)
|
|
|
+ print("模板：", temp_name)
|
|
|
+ print("value：", value)
|
|
|
+
|
|
|
+ if settings.IS_TEST:
|
|
|
+ d = self.windows.temp_class[temp_name](goods_no, value,
|
|
|
+ out_put_dir=out_put_dir,
|
|
|
+ windows=self.windows,
|
|
|
+ assigned_page_list=assigned_page_list)
|
|
|
+ is_deal_success = True
|
|
|
+ else:
|
|
|
+ try:
|
|
|
+ # # 处理图片详情图生成
|
|
|
+ d = self.windows.temp_class[temp_name](goods_no, value,
|
|
|
+ out_put_dir=out_put_dir,
|
|
|
+ windows=self.windows,
|
|
|
+ assigned_page_list=assigned_page_list)
|
|
|
+ is_deal_success = True
|
|
|
+ except BaseException as e:
|
|
|
+ self.show_progress_detail("{}处理失败".format(goods_no))
|
|
|
+ error_text = "{}".format(e)
|
|
|
+ if "Unable to allocate" in error_text:
|
|
|
+ error_text = "电脑内存不足，生成失败"
|
|
|
+
|
|
|
+ self.show_progress_detail("失败原因：{}".format(error_text))
|
|
|
+ self.fail_num += 1
|
|
|
+
|
|
|
+ self.n += 1
|
|
|
+
|
|
|
+ if not is_deal_success:
|
|
|
+ goods_art_no_list = value["货号资料"]
|
|
|
+ self.show_progress_detail("处理失败")
|
|
|
+ self.show_progress_detail(
|
|
|
+ "相关货号：{}".format([x["货号"] for x in goods_art_no_list])
|
|
|
+ )
|
|
|
+ # 将相关的文件夹统一移动至错误文件夹
|
|
|
+ detail_func.move_folders(
|
|
|
+ path_list=[
|
|
|
+ "{}/{}".format(self.windows.image_dir, x)
|
|
|
+ for x in [x["文件夹名称"] for x in goods_art_no_list]
|
|
|
+ ],
|
|
|
+ target_folder=self.windows.target_error_folder,
|
|
|
+ )
|
|
|
+ pass
|
|
|
+ # 更新进度
|
|
|
+ print(self.n, self.total_num)
|
|
|
+ self.windows.progress_sign.emit(
|
|
|
+ {
|
|
|
+ "type": "详情图生成",
|
|
|
+ "progress_bar_value": int(self.n / self.total_num * 100),
|
|
|
+ }
|
|
|
+ )
|
|
|
+
|
|
|
+ def check_serializable(self, obj): # 检查某个对象其中的属性哪些不能被序列化
|
|
|
+ for attr_name in dir(obj):
|
|
|
+ if not attr_name.startswith('__'):
|
|
|
+ try:
|
|
|
+ attr_value = getattr(obj, attr_name)
|
|
|
+ serialized = pickle.dumps(attr_value)
|
|
|
+ print(f"属性 {attr_name} 是可序列化的。")
|
|
|
+ except (TypeError, pickle.PicklingError):
|
|
|
+ print(f"属性 {attr_name} 不可序列化。")
|
|
|
+
|
|
|
+ def on_dialog_result(self, result):
|
|
|
+ """处理对话框结果"""
|
|
|
+ self.dialog_result = result
|
|
|
+ print("972 处理对话框结果:{}".format(result))
|
|
|
+ # self.quit() # 结束事件循
|
|
|
+ self.event.set()
|
zhangyh