Class MaterialThread

pyside/threads/material_thread.py:17–155

素材生产发布线程

class MaterialThread(QThread):
    """素材生产发布线程"""

    log_signal = Signal(str)

    def __init__(self, materials, production_window, material_service):
        super().__init__()
        self.materials = materials
        self.production_window = production_window
        self.material_service = material_service
        self._running = True

    def run(self):
        """运行线程"""
        try:
            selected_model, api_key, config = user_opt()
            if False in [selected_model, api_key]:
                self.log_signal.emit("请先配置模型和api_key")
                time.sleep(10)
                return

            is_publish, is_not_full, api_key, selected_model, prompt = check_user_token()
            print(is_publish, is_not_full, api_key, selected_model, prompt)
            if not is_publish:
                self.log_signal.emit("请先在模型中心配置")
                return

            for material in self.materials:
                if not self._running:
                    break

                try:
                    self.log_signal.emit(f"开始处理素材: {material['title']}")

                    # 1. 下载图片
                    self.log_signal.emit("开始下载图片")
                    img_dir = os.path.join("temp", "img_temp", f"zdy_{material['id']}")
                    os.makedirs(img_dir, exist_ok=True)

                    img_paths = []
                    headers = {
                        "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36",
                    }
                    for i, img_url in enumerate(material["image_list"]):
                        print(img_url)
                        try:
                            response = requests.get(img_url, headers=headers, timeout=30)
                            if response.status_code == 200:
                                img_path = os.path.join(img_dir, f"{i}.jpg")
                                with open(img_path, "wb") as f:
                                    f.write(response.content)
                                img_paths.append(img_path)
                                self.log_signal.emit(f"下载图片 {i + 1}/3 成功")
                            else:
                                self.log_signal.emit(
                                    f"下载图片 {i + 1}/3 失败: HTTP {response.status_code}"
                                )
                        except Exception as e: