"""Concrete PySide6 workers used by GUI tabs.""" from __future__ import annotations from .widgets import * class GenerateWorker(BaseWorker): """Generate titles and covers for eligible collected or failed generation tasks.""" def __init__( self, tasks, prompt_values, db_path=None, config=None, diagnostic_log_dir=None, ): super().__init__() self.tasks = list(tasks) self.prompt_values = dict(prompt_values or {}) self.db_path = db_path self.config = config self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None self._account_by_alias = {} self._task_positions = {} self._eligible_total = 0 self._last_progress_payload = {} self._cmhub_points_balance = None self._billing_error = None self._billing_stop_requested = False def execute(self): account_rows = accounts.list_accounts(path=self.db_path, config=self.config) account_by_alias = { str(account.alias).strip(): account for account in account_rows if str(account.alias).strip() } self._account_by_alias = account_by_alias ai_cfg = appconfig.ai_config(self.config) generate_cover = bool(ai_cfg.get("generate_cover", False)) eligible = [ task for task in self.tasks if ai.is_generatable_task(task, generate_cover=generate_cover) ] component_totals = ai.generation_component_totals( eligible, generate_cover=generate_cover, ) self._eligible_total = len(eligible) self._task_positions = { getattr(task, "id", None): index for index, task in enumerate(eligible, start=1) } batch_ids = self._batch_ids(eligible) self._run_id = self._create_run_log(eligible, batch_ids) if generate_cover: start_message = "[开始] 本轮生成 {total} 条：标题{title_total}，图片{cover_total}；标题并发{title_concurrency}，图片并发{image_concurrency}".format( total=len(eligible), title_total=component_totals["title_total"], cover_total=component_totals["cover_total"], title_concurrency=ai_cfg.get("title_concurrency", 1), image_concurrency=ai_cfg.get("image_concurrency", 1), ) else: start_message = "[开始] 本轮生成 {total} 条：本轮仅生成标题，不生成图片；标题{title_total}；标题并发{title_concurrency}".format( total=len(eligible), title_total=component_totals["title_total"], title_concurrency=ai_cfg.get("title_concurrency", 1), ) self._log_run_event(start_message) try: summary = ai.generate_batch( self.tasks, self.prompt_values, ai_cfg={ "config": self.config, "db_path": self.db_path, "image_dir": appconfig.image_dir(self.config), "account_by_alias": account_by_alias, "on_task_update": self._emit_row_update, "on_event": self._on_generation_event, "on_error": self._on_generation_error, "generate_cover": generate_cover, }, on_progress=self._emit_generate_progress, should_stop=self._should_stop_generation, ) except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) summary = { "ok": False, "error": error, "total": len(eligible), "title_total": component_totals["title_total"], "title_done": 0, "cover_done": 0, "cover_total": component_totals["cover_total"] if generate_cover else 0, "generated_done": 0, "failed": len(eligible), "cancelled": self.should_cancel(), "generate_cover": generate_cover, } self._log_run_event( f"[失败] AI 生成运行失败：{error}", level="error", ) self._write_diagnostic_log( "AI生成运行失败", level="ERROR", step="execute", payload={"error": error}, exc=exc, ) if self._cmhub_points_balance is not None: summary["points_balance"] = self._cmhub_points_balance if self._billing_error is not None: summary["billing_error"] = dict(self._billing_error) summary["ok"] = False summary["cancelled"] = True summary["run_id"] = self._run_id summary["batch_ids"] = batch_ids status = "failed" if summary.get("billing_error") or summary.get("error") else ("cancelled" if summary.get("cancelled") else "done") level = "error" if summary.get("billing_error") or summary.get("error") else ("warning" if summary.get("cancelled") else "info") self._log_run_event(self._format_generate_completion(summary), level=level) self._finish_run_log(status, summary) return summary def _emit_generate_progress(self, payload): progress = dict(payload or {}) if self._cmhub_points_balance is not None: progress["points_balance"] = self._cmhub_points_balance if self._billing_error is not None: progress["billing_error"] = dict(self._billing_error) self._last_progress_payload = dict(progress) self.progress.emit(progress) def _should_stop_generation(self): return self.should_cancel() or self._billing_stop_requested def _emit_row_update(self, task_id, fields): self.row_updated.emit(int(task_id), dict(fields or {})) def _on_generation_event(self, payload): task = payload.get("task") self._remember_cmhub_metadata(payload) message = self._format_generation_event(payload) if not message: return self._log_run_event(message, task=task, level=payload.get("level") or "info") def _remember_cmhub_metadata(self, payload): metadata = payload.get("metadata") if not isinstance(metadata, dict): return if metadata.get("points_balance") is not None: self._cmhub_points_balance = metadata.get("points_balance") self._emit_generate_progress(self._last_progress_payload) def _format_generation_event(self, payload): task = payload.get("task") phase = payload.get("phase") or "generate" step = payload.get("step") or "unknown" result = payload.get("result") or "start" detail = self._short_detail(payload.get("detail")) if isinstance(payload.get("metadata"), dict): return self._format_cmhub_billing_event(task, phase, payload.get("metadata")) if phase == "title": if result == "start" and step == "title_submit": return f"[标题] {self._task_progress_label(task)} 开始生成" if result == "skipped": return f"[标题] {self._task_progress_label(task)} 已有标题，跳过生文" if result == "success" and step == "title_done": return f"[标题] {self._task_progress_label(task)} 成功" if result == "success" and step == "db_write": suffix = f"，{detail}" if detail else "" return f"[标题] {self._task_progress_label(task)} 已保存{suffix}" if result == "retry": return self._retry_message("标题", task, payload, detail) if result == "failed": return f"[失败] {self._task_plain_label(task)} 标题生成失败：{detail or '未知错误'}" if result == "cancelled": return f"[停止] {self._task_plain_label(task)} 标题生成已取消" return None if phase == "cover": if result == "start" and step == "cover_submit": return f"[图片] {self._task_progress_label(task)} 开始生成" if result == "success" and step == "db_write": suffix = f"，已保存 {detail}" if detail else "" return f"[图片] {self._task_progress_label(task)} 成功{suffix}" if result == "retry": return self._retry_message("图片", task, payload, detail) if result == "failed": return f"[失败] {self._task_plain_label(task)} 图片生成失败：{detail or '未知错误'}" if result == "cancelled": return f"[停止] {self._task_plain_label(task)} 图片生成已取消" return None return None def _format_cmhub_billing_event(self, task, phase, metadata): label = "标题" if phase == "title" else ("图片" if phase == "cover" else "AI") parts = [] alias = metadata.get("alias") or metadata.get("model_used") if alias: parts.append(f"别名 {alias}") if metadata.get("points_cost") is not None: parts.append(f"扣点 {metadata.get('points_cost')}") if metadata.get("points_balance") is not None: parts.append(f"余额 {metadata.get('points_balance')}") if metadata.get("call_id"): parts.append(f"call_id={metadata.get('call_id')}") if not parts: return None return f"[计费] {self._task_plain_label(task)} {label}生成：" + "，".join(str(part) for part in parts) def _retry_message(self, label, task, payload, detail): attempt = int(payload.get("attempt", 0) or 0) attempts = int(payload.get("attempts", 0) or 0) max_retries = max(0, attempts - 1) retry_text = f"准备重试 {attempt}/{max_retries}" if max_retries else "准备重试" reason = f"：{detail}" if detail else "" return f"[{label}] {self._task_progress_label(task)} 调用失败，{retry_text}{reason}" def _task_progress_label(self, task): index = self._task_positions.get(getattr(task, "id", None), 0) total = self._eligible_total or 0 item_id = getattr(task, "item_id", "") or "未知商品" shop = self._task_shop_label(task) shop_text = f"（{shop}）" if shop else "" return f"{index}/{total} 商品 {item_id}{shop_text}" def _task_plain_label(self, task): item_id = getattr(task, "item_id", "") or "未知商品" shop = self._task_shop_label(task) return f"商品 {item_id}（{shop}）" if shop else f"商品 {item_id}" def _task_shop_label(self, task): alias = str(getattr(task, "alias", "") or "").strip() account = self._account_by_alias.get(alias) if account is not None: return getattr(account, "account_name", None) or getattr(account, "alias", None) or alias return getattr(task, "account_name", None) or alias def _short_detail(self, detail): if detail is None: return "" text = diagnostics.redact_log_text(str(detail)).replace("\r", " ").replace("\n", " ").strip() if len(text) > 180: return text[:177] + "..." return text def _format_generate_completion(self, summary): progress = self._summary_text(summary) billing_error = summary.get("billing_error") or {} if billing_error: return f"[失败] AI 生成已中止：{billing_error.get('message') or '点数不足，请先充值'}，{progress}" if summary.get("cancelled"): return f"[停止] AI 生成已停止：{progress}" if summary.get("error"): return f"[失败] AI 生成失败：{summary.get('error')}，{progress}" return f"[完成] AI 生成完成：{progress}" def _summary_text(self, summary): title_total = summary.get("title_total", summary.get("total", 0)) cover_total = summary.get("cover_total", summary.get("total", 0)) return "标题{title}/{total}，图片{cover}/{cover_total}，失败{failed}".format( title=summary.get("title_done", 0), cover=summary.get("cover_done", 0), cover_total=cover_total, total=title_total, failed=summary.get("failed", 0), ) def _on_generation_error(self, payload): task = payload.get("task") phase = payload.get("phase") or "generate" step = payload.get("step") or "unknown" exception = payload.get("exception") code = payload.get("code") or getattr(exception, "code", None) status = payload.get("status") or getattr(exception, "status", None) error = diagnostics.redact_log_text(payload.get("error") or "未知错误") if str(code or "") == "insufficient_points": self._billing_stop_requested = True message = "点数不足，请先充值。本轮未开始任务将停止。" self._billing_error = { "code": "insufficient_points", "message": message, "phase": phase, "task_id": getattr(task, "id", None), "item_id": getattr(task, "item_id", None), } if status is not None: self._billing_error["status"] = status self._log_run_event( f"[计费] {self._task_plain_label(task)} 点数不足，请先充值；本轮未开始任务将停止", task=task, level="error", ) self._emit_generate_progress(self._last_progress_payload) diagnostic_payload = {"phase": phase, "error": error} if code is not None: diagnostic_payload["code"] = str(code) if status is not None: diagnostic_payload["status"] = status self._write_diagnostic_log( "AI生成任务失败", level="ERROR", step=step, task=task, payload=diagnostic_payload, exc=exception, ) def _batch_ids(self, tasks): batch_ids = [] for task in tasks: batch_id = getattr(task, "batch_id", None) if batch_id and batch_id not in batch_ids: batch_ids.append(batch_id) return batch_ids def _create_run_log(self, eligible, batch_ids): try: ai_cfg = appconfig.ai_config(self.config) return db.create_run_log( "generate", dry_run=False, total=len(eligible), options={ "batch_ids": batch_ids, "default_text_model": ai_cfg.get("default_text_model"), "default_image_model": ai_cfg.get("default_image_model"), "resolution": ai_cfg.get("resolution"), "title_concurrency": ai_cfg.get("title_concurrency"), "image_concurrency": ai_cfg.get("image_concurrency"), "generate_cover": ai_cfg.get("generate_cover", False), "backend": ai_cfg.get("backend", "direct"), }, path=self.db_path, ) except Exception: return None def _finish_run_log(self, status, summary): if self._run_id is None: return try: generated_done = summary.get("generated_done") if generated_done is None: generated_done = summary.get("cover_done", 0) if not summary.get("generate_cover", True) and not generated_done: generated_done = summary.get("title_done", 0) done = int(generated_done or 0) + int(summary.get("failed", 0) or 0) db.finish_run_log( self._run_id, status=status, done=done, success_count=generated_done, skipped_count=0, failed_count=summary.get("failed", 0), summary_json=summary, path=self.db_path, ) except Exception: return def _log_run_event(self, message, task=None, level="info"): safe_message = diagnostics.redact_log_text(message) self.log.emit(str(safe_message)) if self._run_id is None: return try: db.add_run_log_event( self._run_id, safe_message, task_id=getattr(task, "id", None), alias=getattr(task, "alias", None), item_id=getattr(task, "item_id", None), level=level, path=self.db_path, ) except Exception: return def _write_diagnostic_log( self, message, level="INFO", step=None, task=None, payload=None, exc=None, ): try: diagnostics.write_diagnostic_log( message, level=level, step=step, task_id=getattr(task, "id", None), alias=getattr(task, "alias", None), item_id=getattr(task, "item_id", None), payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) except Exception: return class ApplyWorker(BaseWorker): """Apply generated title/cover changes, optionally previewing or grouping by account.""" def __init__( self, tasks, db_path=None, config=None, preflight=True, close_success_tab=False, dry_run=False, parallel_accounts=False, max_parallel_accounts=1, batch_size=None, diagnostic_log_dir=None, ): super().__init__() self.tasks = list(tasks) self.db_path = db_path self.config = config self.preflight = preflight self.close_success_tab = close_success_tab self.dry_run = bool(dry_run) self.parallel_accounts = bool(parallel_accounts) self.max_parallel_accounts = max(1, int(max_parallel_accounts or 1)) self.batch_size = None if batch_size is None else max(1, int(batch_size or 1)) self._current_batch_size = None self._batch_count = 0 self._progress_lock = threading.Lock() self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): account_rows = accounts.list_accounts(path=self.db_path, config=self.config) account_by_alias = { str(account.alias).strip(): account for account in account_rows if str(account.alias).strip() } eligible = [task for task in self.tasks if self._is_actionable_task(task)] batch_ids = self._batch_ids(eligible) total = len(eligible) batch_size = self._effective_batch_size(total) batches = self._task_batches(eligible, batch_size) self._current_batch_size = batch_size self._batch_count = len(batches) counters = { "done": 0, "applied": 0, "skipped": 0, "failed": 0, } self._run_id = self._create_run_log(eligible, batch_ids) self._log_run_event( "step=start result=start detail=运行开始：{mode}，任务{total}，每批最多{batch_size}，批次{batch_count}，{parallel}".format( mode="检查本轮更新" if self.dry_run else "真实更新", total=total, batch_size=batch_size, batch_count=len(batches), parallel=( f"多账号并行最多{self.max_parallel_accounts}" if self.parallel_accounts else "串行" ), ) ) if self.preflight and not self.dry_run: self._log_run_event("step=preflight result=start detail=账号就绪检查") blocked = self._preflight_block(eligible, account_rows, account_by_alias) if blocked: self._log_preflight_blocked(blocked) summary = self._summary( ok=False, total=total, counters=counters, batch_ids=batch_ids, blocked=True, extra=blocked, ) self._finish_run_log("blocked", summary) return summary self._log_run_event("step=preflight result=success detail=账号检查通过") elif not self.preflight: self._log_run_event( "step=preflight result=skipped detail=测试模式跳过更新前检查", level="warning", ) for batch_index, batch_tasks in enumerate(batches, start=1): if self.should_cancel(): break self._log_batch_start(batch_index, len(batches), batch_tasks, counters, total) if self.dry_run: for task in batch_tasks: if self.should_cancel(): break outcome = self._preview_task(task, account_by_alias) self._record_outcome(counters, total, outcome) elif self.parallel_accounts and self.max_parallel_accounts > 1: self._run_parallel_by_account(batch_tasks, account_by_alias, counters, total) else: for task in batch_tasks: if self.should_cancel(): break outcome = self._apply_one_task(task, account_by_alias) self._record_outcome(counters, total, outcome) summary = self._summary( ok=counters["failed"] == 0, total=total, counters=counters, batch_ids=batch_ids, ) self._finish_run_log("cancelled" if self.should_cancel() else "done", summary) return summary def _is_actionable_task(self, task): return ( getattr(task, "stage", None) == "generated" and getattr(task, "status", None) in {"success", "pending", "failed"} and bool(getattr(task, "new_title", None) or getattr(task, "new_cover_path", None)) ) def _preflight_block(self, eligible, account_rows, account_by_alias): if not account_rows: return { "reason": "NO_ACCOUNTS", "no_accounts": True, } duplicate_ports = self._duplicate_debug_ports(account_rows, eligible, account_by_alias) if duplicate_ports: return { "reason": "DUPLICATE_DEBUG_PORT", "duplicate_ports": duplicate_ports, } required_accounts = [] seen_aliases = set() for task in eligible: alias = str(task.alias).strip() account = account_by_alias.get(alias) if account is not None and alias not in seen_aliases: required_accounts.append(account) seen_aliases.add(alias) not_running = [] logged_out = [] for account in required_accounts: self._log_run_event( f"step=check_chrome result=start detail=账号 {account.alias} debug_port={account.debug_port}", level="info", ) if not chrome.is_running(account.debug_port): self._log_run_event( f"step=check_chrome result=blocked detail=账号 {account.alias} CDP 端口未响应 debug_port={account.debug_port}", level="warning", ) not_running.append(self._account_payload(account, "CDP 端口未响应")) continue self._log_run_event( f"step=check_chrome result=success detail=账号 {account.alias} debug_port={account.debug_port}", level="info", ) self._log_run_event( f"step=login_check result=start detail=账号 {account.alias}", level="info", ) status = self._login_status(account) if not status.get("logged_in"): reason = self._login_skip_reason(status) self._log_run_event( f"step=login_check result=blocked detail=账号 {account.alias} {reason}", level="warning", ) logged_out.append( self._account_payload(account, reason) ) else: self._log_run_event( f"step=login_check result=success detail=账号 {account.alias}", level="info", ) if not_running or logged_out: return { "reason": "ACCOUNT_NOT_READY", "not_running": not_running, "logged_out": logged_out, } return None def _duplicate_debug_ports(self, account_rows, eligible, account_by_alias): required_aliases = { str(task.alias).strip() for task in eligible if account_by_alias.get(str(task.alias).strip()) is not None } by_port = {} for account in account_rows: if account.alias not in required_aliases: continue by_port.setdefault(int(account.debug_port), []).append(account) duplicates = [] for port, rows in by_port.items(): if len(rows) > 1: duplicates.append( { "debug_port": port, "aliases": [row.alias for row in rows], } ) return duplicates def _effective_batch_size(self, total): if self.batch_size is None: return max(1, int(total or 1)) return self.batch_size def _task_batches(self, tasks, batch_size): if not tasks: return [] return [ tasks[index:index + batch_size] for index in range(0, len(tasks), batch_size) ] def _log_batch_start(self, batch_index, batch_count, batch_tasks, counters, total): first = counters["done"] + 1 last = min(first + len(batch_tasks) - 1, total) label = "检查批次" if self.dry_run else "更新批次" self._log_run_event( f"step=batch result=start detail={label} {batch_index}/{batch_count} 开始：任务 {first}-{last}/{total}" ) def _run_parallel_by_account(self, eligible, account_by_alias, counters, total): groups = self._group_tasks_by_alias(eligible) max_workers = min(self.max_parallel_accounts, len(groups)) if max_workers <= 1: for group_tasks in groups: self._run_task_group(group_tasks, account_by_alias, counters, total) return with ThreadPoolExecutor(max_workers=max_workers) as executor: futures = [ executor.submit( self._run_task_group, group_tasks, account_by_alias, counters, total, ) for group_tasks in groups ] for future in as_completed(futures): future.result() def _group_tasks_by_alias(self, tasks): groups = [] index_by_alias = {} for task in tasks: alias = str(task.alias).strip() if alias not in index_by_alias: index_by_alias[alias] = len(groups) groups.append([]) groups[index_by_alias[alias]].append(task) return groups def _run_task_group(self, tasks, account_by_alias, counters, total): for task in tasks: if self.should_cancel(): break outcome = self._apply_one_task(task, account_by_alias) self._record_outcome(counters, total, outcome) def _preview_task(self, task, account_by_alias): account = account_by_alias.get(str(task.alias).strip()) if account is None: reason = "别名未匹配账号" self._log_run_event( f"step=preview result=skipped detail=检查：任务 {task.id} 商品 {task.item_id} 将略过：{reason}", task=task, level="warning", ) return "skipped" action_parts = [] if getattr(task, "new_title", None): action_parts.append("标题") if getattr(task, "new_cover_path", None): action_parts.append("封面") action_text = "+".join(action_parts) or "无变更" self._log_run_event( "step=preview result=success detail=检查：任务 {task_id} 商品 {item_id} 账号 {alias} 将更新 {action}".format( task_id=task.id, item_id=task.item_id, alias=account.alias, action=action_text, ), task=task, ) return "applied" def _apply_one_task(self, task, account_by_alias): account = account_by_alias.get(str(task.alias).strip()) if account is None: reason = "别名未匹配账号" db.mark_skipped(task.id, reason, path=self.db_path) self.row_updated.emit(task.id, {"status": "skipped", "last_error": reason}) self._log_run_event( f"step=preflight result=skipped detail=任务 {task.id} 商品 {task.item_id} 已略过：{reason}", task=task, level="warning", ) return "skipped" started = time.monotonic() current_step = "db_write" def on_step(event): nonlocal current_step if isinstance(event, dict): step = str(event.get("step") or "apply_task") result = str(event.get("result") or "start") detail = event.get("detail") else: step = str(event) result = "start" detail = None current_step = step level = "error" if result == "failed" else "info" detail_text = "任务 {task_id} 商品 {item_id}".format( task_id=task.id, item_id=task.item_id, ) if detail: detail_text = f"{detail_text} {detail}" self._log_run_event( f"step={step} result={result} detail={detail_text}", task=task, level=level, ) try: self._log_run_event( f"step=apply_task result=start detail=任务 {task.id} 商品 {task.item_id} 开始更新，账号 {account.alias}", task=task, ) current_step = "db_write" self._log_run_event( f"step=db_write result=start detail=任务 {task.id} 商品 {task.item_id} 标记更新运行", task=task, ) db.mark_running(task.id, "apply", path=self.db_path) self.row_updated.emit(task.id, {"status": "running", "last_error": None}) result = editor.apply_task( account, task, close_success_tab=self.close_success_tab, on_step=on_step, ) committed = bool(result.get("committed")) and not result.get("error") error = result.get("error") current_step = "db_write" self._log_run_event( f"step=db_write result=start detail=任务 {task.id} 商品 {task.item_id} 保存更新结果", task=task, ) if committed: db.set_applied(task.id, True, path=self.db_path) elapsed_ms = self._elapsed_ms(started) self.row_updated.emit( task.id, { "stage": "applied", "status": "success", "committed": 1, "last_error": None, }, ) self._log_run_event( f"step=db_write result=success detail=任务 {task.id} 商品 {task.item_id} 更新成功 elapsed_ms={elapsed_ms}", task=task, ) return "applied" error = diagnostics.redact_log_text(error or "更新未提交") failed_step = self._failed_apply_step(result, current_step) db.set_applied(task.id, False, error, path=self.db_path) elapsed_ms = self._elapsed_ms(started) self.failed.emit(task.id, str(error)) self.row_updated.emit( task.id, {"status": "failed", "last_error": str(error), "committed": 0}, ) self._log_run_event( f"step={failed_step} result=failed detail={error} elapsed_ms={elapsed_ms}", task=task, level="error", ) self._log_run_event( f"step=db_write result=success detail=任务 {task.id} 商品 {task.item_id} 保存失败状态 elapsed_ms={elapsed_ms}", task=task, ) self._write_diagnostic_log( "蝦皮更新任务失败", level="ERROR", step=failed_step, task=task, elapsed_ms=elapsed_ms, payload={"error": error, "result": result}, ) return "failed" except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) db.set_applied(task.id, False, error, path=self.db_path) elapsed_ms = self._elapsed_ms(started) self.failed.emit(task.id, error) self.row_updated.emit( task.id, {"status": "failed", "last_error": error, "committed": 0}, ) self._log_run_event( f"step={current_step} result=failed detail={error} elapsed_ms={elapsed_ms}", task=task, level="error", ) self._write_diagnostic_log( "蝦皮更新任务异常", level="ERROR", step=current_step, task=task, elapsed_ms=elapsed_ms, payload={"error": error}, exc=exc, ) return "failed" def _record_outcome(self, counters, total, outcome): with self._progress_lock: counters["done"] += 1 if outcome == "applied": counters["applied"] += 1 elif outcome == "skipped": counters["skipped"] += 1 else: counters["failed"] += 1 self._emit_progress( counters["done"], total, counters["applied"], counters["skipped"], counters["failed"], ) def _account_payload(self, account, reason=None): payload = { "account_name": account.account_name, "alias": account.alias, "debug_port": account.debug_port, } if reason: payload["reason"] = reason return payload def _emit_progress(self, done, total, applied, skipped, failed): self.progress.emit( { "done": done, "total": total, "applied": applied, "skipped": skipped, "failed": failed, "dry_run": self.dry_run, "batch_size": self._current_batch_size, "batch_count": self._batch_count, } ) def _login_status(self, account): try: return accounts.detect_login(account, path=self.db_path, config=self.config) except Exception as exc: return { "logged_in": False, "reason": f"LOGIN_CHECK_FAILED: {exc}", } def _login_skip_reason(self, status): reason = status.get("reason") return f"账号未登录: {reason}" if reason else "账号未登录" def _batch_ids(self, tasks): batch_ids = [] for task in tasks: batch_id = getattr(task, "batch_id", None) if batch_id and batch_id not in batch_ids: batch_ids.append(batch_id) return batch_ids def _summary(self, ok, total, counters, batch_ids, blocked=False, extra=None): summary = { "ok": ok, "total": total, "done": counters["done"], "applied": counters["applied"], "skipped": counters["skipped"], "failed": counters["failed"], "batch_ids": batch_ids, "dry_run": self.dry_run, "parallel_accounts": self.parallel_accounts, "batch_size": self._current_batch_size, "batch_count": self._batch_count, "run_id": self._run_id, } if blocked: summary["blocked"] = True if extra: summary.update(extra) return summary def _create_run_log(self, eligible, batch_ids): try: return db.create_run_log( "apply", dry_run=self.dry_run, total=len(eligible), options={ "batch_ids": batch_ids, "close_success_tab": self.close_success_tab, "dry_run": self.dry_run, "parallel_accounts": self.parallel_accounts, "max_parallel_accounts": self.max_parallel_accounts, "batch_size": self._current_batch_size, "batch_count": self._batch_count, }, path=self.db_path, ) except Exception: return None def _finish_run_log(self, status, summary): if self._run_id is None: return try: db.finish_run_log( self._run_id, status=status, done=summary.get("done", 0), success_count=summary.get("applied", 0), skipped_count=summary.get("skipped", 0), failed_count=summary.get("failed", 0), summary_json=summary, path=self.db_path, ) except Exception: return def _log_run_event(self, message, task=None, level="info"): safe_message = diagnostics.redact_log_text(message) self.log.emit(str(safe_message)) if self._run_id is None: return try: db.add_run_log_event( self._run_id, safe_message, task_id=getattr(task, "id", None), alias=getattr(task, "alias", None), item_id=getattr(task, "item_id", None), level=level, path=self.db_path, ) except Exception: return def _log_preflight_blocked(self, blocked): if blocked.get("no_accounts"): self._log_run_event( "step=preflight result=blocked detail=当前没有配置账号", level="warning", ) for item in blocked.get("duplicate_ports") or []: self._log_run_event( "step=preflight result=blocked detail=调试端口重复 debug_port={port} aliases={aliases}".format( port=item.get("debug_port") or "", aliases=",".join(item.get("aliases") or []), ), level="warning", ) for item in blocked.get("not_running") or []: self._log_run_event( "step=check_chrome result=blocked detail=账号 {alias} Chrome 未启动或调试端口不可访问: {reason}".format( alias=item.get("alias") or "", reason=item.get("reason") or "", ), level="warning", ) for item in blocked.get("logged_out") or []: self._log_run_event( "step=login_check result=blocked detail=账号 {alias} 未登录蝦皮: {reason}".format( alias=item.get("alias") or "", reason=item.get("reason") or "", ), level="warning", ) def _failed_apply_step(self, result, fallback): if not isinstance(result, dict): return fallback or "apply_task" title = result.get("title") if isinstance(title, dict) and not title.get("ok", True): return "change_title" cover = result.get("cover") if isinstance(cover, dict) and not cover.get("ok", True): return "replace_cover" update = result.get("update") if isinstance(update, dict): return "click_update" return fallback or "apply_task" def _write_diagnostic_log( self, message, level="INFO", step=None, task=None, elapsed_ms=None, payload=None, exc=None, ): _safe_write_diagnostic_log( message, level=level, step=step, task=task, elapsed_ms=elapsed_ms, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) def _elapsed_ms(self, started): return _elapsed_ms(started) class CollectWorker(BaseWorker): """Collect old title and cover for imported tasks.""" def __init__( self, tasks, db_path=None, config=None, preflight=True, diagnostic_log_dir=None, ): super().__init__() self.tasks = list(tasks) self.db_path = db_path self.config = config self.preflight = preflight self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): account_rows = accounts.list_accounts(path=self.db_path, config=self.config) account_by_alias = { str(account.alias).strip(): account for account in account_rows if str(account.alias).strip() } eligible = [ task for task in self.tasks if getattr(task, "stage", None) == "imported" ] batch_ids = self._batch_ids(eligible) total = len(eligible) collected = 0 skipped = 0 failed = 0 done = 0 self._run_id = self._create_run_log(eligible, batch_ids) self._log_run_event( f"step=preflight result=start detail=采集运行开始 total={total}" ) if self.preflight: blocked = self._preflight_block(eligible, account_rows, account_by_alias) if blocked: self._log_preflight_blocked(blocked) summary = self._summary( ok=False, total=total, done=done, collected=collected, skipped=skipped, failed=failed, batch_ids=batch_ids, blocked=True, extra=blocked, ) self._finish_run_log("blocked", summary) return summary self._log_run_event("step=preflight result=success detail=账号检查通过") else: self._log_run_event( "step=preflight result=skipped detail=测试模式跳过采集前检查", level="warning", ) for task in eligible: if self.should_cancel(): break account = account_by_alias.get(str(task.alias).strip()) if account is None: skipped += 1 done += 1 reason = "别名未匹配账号" db.mark_skipped(task.id, reason, path=self.db_path) self.row_updated.emit(task.id, {"status": "skipped", "last_error": reason}) self._log_run_event( "step=preflight result=skipped detail=任务 {task_id} 商品 {item_id} {reason}".format( task_id=task.id, item_id=task.item_id, reason=reason, ), task=task, level="warning", ) self._emit_progress(done, total, collected, skipped, failed) continue status = self._login_status(account) if not status.get("logged_in"): skipped += 1 done += 1 reason = self._login_skip_reason(status) db.mark_skipped(task.id, reason, path=self.db_path) self.row_updated.emit(task.id, {"status": "skipped", "last_error": reason}) self._log_run_event( "step=preflight result=skipped detail=任务 {task_id} 商品 {item_id} {reason}".format( task_id=task.id, item_id=task.item_id, reason=reason, ), task=task, level="warning", ) self._emit_progress(done, total, collected, skipped, failed) continue started = time.monotonic() current_step = "db_write" def on_step(step): nonlocal current_step current_step = str(step) self._log_run_event( "step={step} result=start detail=任务 {task_id} 商品 {item_id}".format( step=current_step, task_id=task.id, item_id=task.item_id, ), task=task, ) try: self._log_run_event( "step=db_write result=start detail=任务 {task_id} 商品 {item_id} 标记采集运行".format( task_id=task.id, item_id=task.item_id, ), task=task, ) db.mark_running(task.id, "collect", path=self.db_path) self.row_updated.emit(task.id, {"status": "running"}) result = editor.collect( account, { "item_id": task.item_id, "old_cover_path": self._old_cover_path(account, task), }, on_step=on_step, ) current_step = "db_write" self._log_run_event( "step=db_write result=start detail=任务 {task_id} 商品 {item_id} 保存采集结果".format( task_id=task.id, item_id=task.item_id, ), task=task, ) db.set_collected( task.id, result.get("old_title", ""), result.get("old_cover_path", ""), path=self.db_path, ) collected += 1 elapsed_ms = self._elapsed_ms(started) self.row_updated.emit( task.id, { "stage": "collected", "status": "success", "old_title": result.get("old_title", ""), "old_cover_path": result.get("old_cover_path", ""), }, ) self._log_run_event( "step=db_write result=success detail=任务 {task_id} 商品 {item_id} 采集成功 elapsed_ms={elapsed_ms}".format( task_id=task.id, item_id=task.item_id, elapsed_ms=elapsed_ms, ), task=task, ) except Exception as exc: failed += 1 error = str(exc) or exc.__class__.__name__ safe_error = diagnostics.redact_log_text(error) elapsed_ms = self._elapsed_ms(started) db.mark_failed(task.id, "collect", safe_error, path=self.db_path) self.failed.emit(task.id, safe_error) self.row_updated.emit(task.id, {"status": "failed", "last_error": safe_error}) self._log_run_event( "step={step} result=failed detail={error} elapsed_ms={elapsed_ms}".format( step=current_step, error=safe_error, elapsed_ms=elapsed_ms, ), task=task, level="error", ) self._write_diagnostic_log( "采集任务失败", level="ERROR", step=current_step, task=task, elapsed_ms=elapsed_ms, payload={"error": safe_error}, exc=exc, ) finally: done += 1 self._emit_progress(done, total, collected, skipped, failed) summary = self._summary( ok=failed == 0, total=total, done=done, collected=collected, skipped=skipped, failed=failed, batch_ids=batch_ids, ) self._finish_run_log("cancelled" if self.should_cancel() else "done", summary) return summary def _preflight_block(self, eligible, account_rows, account_by_alias): if not account_rows: return { "reason": "NO_ACCOUNTS", "no_accounts": True, } required_accounts = [] seen_aliases = set() for task in eligible: alias = str(task.alias).strip() account = account_by_alias.get(alias) if account is not None and alias not in seen_aliases: required_accounts.append(account) seen_aliases.add(alias) not_running = [] logged_out = [] for account in required_accounts: self._log_run_event( f"step=check_chrome result=start detail=账号 {account.alias} debug_port={account.debug_port}", level="info", ) if not chrome.is_running(account.debug_port): self._log_run_event( f"step=check_chrome result=blocked detail=账号 {account.alias} CDP 端口未响应 debug_port={account.debug_port}", level="warning", ) not_running.append(self._account_payload(account, "CDP 端口未响应")) continue self._log_run_event( f"step=check_chrome result=success detail=账号 {account.alias} debug_port={account.debug_port}", level="info", ) self._log_run_event( f"step=login_check result=start detail=账号 {account.alias}", level="info", ) status = self._login_status(account) if not status.get("logged_in"): reason = self._login_skip_reason(status) self._log_run_event( f"step=login_check result=blocked detail=账号 {account.alias} {reason}", level="warning", ) logged_out.append( self._account_payload(account, reason) ) else: self._log_run_event( f"step=login_check result=success detail=账号 {account.alias}", level="info", ) if not_running or logged_out: return { "reason": "ACCOUNT_NOT_READY", "not_running": not_running, "logged_out": logged_out, } return None def _account_payload(self, account, reason=None): payload = { "account_name": account.account_name, "alias": account.alias, "debug_port": account.debug_port, } if reason: payload["reason"] = reason return payload def _emit_progress(self, done, total, collected, skipped, failed): self.progress.emit( { "done": done, "total": total, "collected": collected, "skipped": skipped, "failed": failed, } ) def _login_status(self, account): try: return accounts.detect_login(account, path=self.db_path, config=self.config) except Exception as exc: return { "logged_in": False, "reason": f"LOGIN_CHECK_FAILED: {exc}", } def _login_skip_reason(self, status): reason = status.get("reason") return f"账号未登录: {reason}" if reason else "账号未登录" def _old_cover_path(self, account, task): image_root = appconfig.image_dir(self.config) return image_paths.task_image_path(image_root, task, account, "old") def _batch_ids(self, tasks): batch_ids = [] for task in tasks: batch_id = getattr(task, "batch_id", None) if batch_id and batch_id not in batch_ids: batch_ids.append(batch_id) return batch_ids def _summary( self, ok, total, done, collected, skipped, failed, batch_ids, blocked=False, extra=None, ): summary = { "ok": ok, "total": total, "done": done, "collected": collected, "skipped": skipped, "failed": failed, "batch_ids": batch_ids, "run_id": self._run_id, } if blocked: summary["blocked"] = True if extra: summary.update(extra) return summary def _create_run_log(self, eligible, batch_ids): try: return db.create_run_log( "collect", dry_run=False, total=len(eligible), options={ "batch_ids": batch_ids, "preflight": self.preflight, }, path=self.db_path, ) except Exception: return None def _finish_run_log(self, status, summary): if self._run_id is None: return try: db.finish_run_log( self._run_id, status=status, done=summary.get("done", 0), success_count=summary.get("collected", 0), skipped_count=summary.get("skipped", 0), failed_count=summary.get("failed", 0), summary_json=summary, path=self.db_path, ) except Exception: return def _log_run_event(self, message, task=None, level="info"): safe_message = diagnostics.redact_log_text(message) self.log.emit(str(safe_message)) if self._run_id is None: return try: db.add_run_log_event( self._run_id, safe_message, task_id=getattr(task, "id", None), alias=getattr(task, "alias", None), item_id=getattr(task, "item_id", None), level=level, path=self.db_path, ) except Exception: return def _log_preflight_blocked(self, blocked): if blocked.get("no_accounts"): self._log_run_event( "step=preflight result=blocked detail=当前没有配置账号", level="warning", ) for item in blocked.get("not_running") or []: self._log_run_event( "step=preflight result=blocked detail=账号 {alias} Chrome 未启动或调试端口不可访问: {reason}".format( alias=item.get("alias") or "", reason=item.get("reason") or "", ), level="warning", ) for item in blocked.get("logged_out") or []: self._log_run_event( "step=preflight result=blocked detail=账号 {alias} 未登录蝦皮: {reason}".format( alias=item.get("alias") or "", reason=item.get("reason") or "", ), level="warning", ) def _write_diagnostic_log( self, message, level="INFO", step=None, task=None, elapsed_ms=None, payload=None, exc=None, ): try: diagnostics.write_diagnostic_log( message, level=level, step=step, task_id=getattr(task, "id", None), alias=getattr(task, "alias", None), item_id=getattr(task, "item_id", None), elapsed_ms=elapsed_ms, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) except Exception: return def _elapsed_ms(self, started): return int((time.monotonic() - started) * 1000) class WriteBackWorker(BaseWorker): """Write Excel fields back in a background thread.""" def __init__(self, batch_id, db_path=None, excel_path=None, mode="old", diagnostic_log_dir=None): super().__init__() self.batch_id = batch_id self.db_path = db_path self.excel_path = excel_path self.mode = mode self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): batch_ids = self._batch_ids() self._run_id = _safe_create_run_log( "write_back", db_path=self.db_path, total=len(batch_ids), options={ "batch_ids": batch_ids, "mode": self.mode, "excel_path": self.excel_path, }, ) self._log_run_event( f"step=start result=start detail=Excel 回写开始 mode={self.mode} batch_count={len(batch_ids)}" ) results = [] try: for batch_id in batch_ids: started = time.monotonic() self._log_run_event( f"step=write_excel result=start detail=batch_id={batch_id} mode={self.mode}" ) result = self._write_one(batch_id) results.append(result) self._log_run_event( "step=write_excel result=success detail=batch_id={batch_id} files={files} rows={rows} elapsed_ms={elapsed_ms}".format( batch_id=batch_id, files=result.get("files", 0), rows=result.get("rows", 0), elapsed_ms=self._elapsed_ms(started), ) ) except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) self._log_run_event( f"step=write_excel result=failed detail={error}", level="error", ) self._write_diagnostic_log( "Excel回写失败", level="ERROR", step="write_excel", payload={"batch_ids": batch_ids, "mode": self.mode, "error": error}, exc=exc, ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="failed", done=len(results), success_count=sum(result.get("rows", 0) for result in results), failed_count=1, summary_json={"ok": False, "error": error, "mode": self.mode}, ) raise result = results[0] if len(results) == 1 else self._combined_result(results) self.progress.emit( { "done": result.get("rows", 0), "total": result.get("rows", 0), "files": result.get("files", 0), } ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="done", done=len(batch_ids), success_count=result.get("rows", 0), failed_count=0, summary_json={"ok": result.get("ok", False), "mode": self.mode, "result": result}, ) return result def _batch_ids(self): if isinstance(self.batch_id, (list, tuple, set)): return list(self.batch_id) return [self.batch_id] def _write_one(self, batch_id): if self.mode == "results": return excel.write_back_results( batch_id, excel_path=self.excel_path, path=self.db_path, ) return excel.write_back( batch_id, excel_path=self.excel_path, path=self.db_path, ) def _combined_result(self, results): written_files = [] for result in results: for file_path in result.get("written_files", []): if file_path not in written_files: written_files.append(file_path) return { "ok": all(result.get("ok", False) for result in results), "batch_id": [result.get("batch_id") for result in results], "files": sum(result.get("files", 0) for result in results), "rows": sum(result.get("rows", 0) for result in results), "written_files": written_files, } def _log_run_event(self, message, level="info"): safe_message = _safe_add_run_log_event( self._run_id, message, db_path=self.db_path, level=level, ) self.log.emit(str(safe_message)) def _write_diagnostic_log(self, message, level="INFO", step=None, payload=None, exc=None): _safe_write_diagnostic_log( message, level=level, step=step, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) def _elapsed_ms(self, started): return _elapsed_ms(started) class AccountLoginCheckWorker(BaseWorker): def __init__(self, account, db_path=None, config=None, timeout=8, diagnostic_log_dir=None): super().__init__() self.account = account self.db_path = db_path self.config = config self.timeout = timeout self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): self._run_id = _safe_create_run_log( "login_check", db_path=self.db_path, total=1, options={ "alias": self.account.alias, "debug_port": self.account.debug_port, "timeout": self.timeout, }, ) started = time.monotonic() self._log_run_event( f"step=detect_login result=start detail=账号 {self.account.alias} debug_port={self.account.debug_port}" ) try: status = accounts.detect_login( self.account, timeout=self.timeout, path=self.db_path, config=self.config, ) except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) elapsed_ms = self._elapsed_ms(started) self._log_run_event( f"step=detect_login result=failed detail={error} elapsed_ms={elapsed_ms}", level="error", ) self._write_diagnostic_log( "登录检测失败", level="ERROR", step="detect_login", elapsed_ms=elapsed_ms, payload={"alias": self.account.alias, "error": error}, exc=exc, ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="failed", done=0, failed_count=1, summary_json={"ok": False, "alias": self.account.alias, "error": error}, ) raise elapsed_ms = self._elapsed_ms(started) logged_in = bool(status.get("logged_in")) result_text = "success" if logged_in else "failed" level = "info" if logged_in else "warning" self._log_run_event( "step=detect_login result={result} detail=账号 {alias} logged_in={logged_in} reason={reason} elapsed_ms={elapsed_ms}".format( result=result_text, alias=self.account.alias, logged_in=logged_in, reason=status.get("reason") or "", elapsed_ms=elapsed_ms, ), level=level, ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="done", done=1, success_count=1 if logged_in else 0, failed_count=0 if logged_in else 1, summary_json={"ok": logged_in, "alias": self.account.alias, "status": status}, ) self.row_updated.emit(self.account.id, status) return {"alias": self.account.alias, "status": status} def _log_run_event(self, message, level="info"): safe_message = _safe_add_run_log_event( self._run_id, message, db_path=self.db_path, account=self.account, level=level, ) self.log.emit(str(safe_message)) def _write_diagnostic_log( self, message, level="INFO", step=None, elapsed_ms=None, payload=None, exc=None, ): _safe_write_diagnostic_log( message, level=level, step=step, account=self.account, elapsed_ms=elapsed_ms, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) def _elapsed_ms(self, started): return _elapsed_ms(started) class CMHubSettingsWorker(BaseWorker): """Fetch cmhub aliases and optional balance without blocking the GUI.""" def __init__( self, base_url, api_key, connect_timeout=10, include_balance=True, db_path=None, diagnostic_log_dir=None, ): super().__init__() self.base_url = str(base_url or "").strip() self.api_key = str(api_key or "") self.connect_timeout = max(1, int(connect_timeout or 10)) self.include_balance = bool(include_balance) self.db_path = db_path self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): self._run_id = self._create_run_log() started = time.monotonic() action = "测试连接/查余额" if self.include_balance else "刷新别名" self._log_run_event(f"step=cmhub_settings result=start detail={action}") try: models = ai.fetch_cmhub_models( self.base_url, self.api_key, connect_timeout=self.connect_timeout, ) balance = None if self.include_balance: balance = ai.fetch_cmhub_balance( self.base_url, self.api_key, connect_timeout=self.connect_timeout, ) except Exception as exc: error = self._safe_error(exc) elapsed_ms = self._elapsed_ms(started) self._log_run_event( f"step=cmhub_settings result=failed detail={error} elapsed_ms={elapsed_ms}", level="error", ) self._write_diagnostic_log( "cmhub 设置检测失败", level="ERROR", step="cmhub_settings", elapsed_ms=elapsed_ms, payload={"base_url": self.base_url, "error": error}, exc=exc, ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="failed", done=0, failed_count=1, summary_json={"ok": False, "error": error}, ) raise RuntimeError(error) from exc elapsed_ms = self._elapsed_ms(started) payload = { "ok": True, "models": appconfig.sanitize_for_log(models), "balance": appconfig.sanitize_for_log(balance or {}), "points_balance": (balance or {}).get("points_balance"), } title_count = self._priced_count(models, "title") image_count = self._priced_count(models, "image") self._log_run_event( "step=cmhub_settings result=success detail=title_aliases={title_count} image_aliases={image_count} points_balance={points_balance} elapsed_ms={elapsed_ms}".format( title_count=title_count, image_count=image_count, points_balance=payload.get("points_balance") if payload.get("points_balance") is not None else "", elapsed_ms=elapsed_ms, ) ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="done", done=1, success_count=1, summary_json=payload, ) return payload def _priced_count(self, models, operation): return sum( 1 for model in models or [] if str(model.get("operation_type") or "").lower() == operation and str(model.get("pricing_status") or "").lower() != "unpriced" and str(model.get("alias") or "").strip() ) def _create_run_log(self): if not self.db_path: return None return _safe_create_run_log( "cmhub_settings_test", db_path=self.db_path, total=1, options={"base_url": self.base_url, "include_balance": self.include_balance}, ) def _log_run_event(self, message, level="info"): safe_message = _safe_add_run_log_event( self._run_id, message, db_path=self.db_path, level=level, ) self.log.emit(str(safe_message)) def _write_diagnostic_log( self, message, level="INFO", step=None, elapsed_ms=None, payload=None, exc=None, ): _safe_write_diagnostic_log( message, level=level, step=step, elapsed_ms=elapsed_ms, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) def _safe_error(self, exc): raw = str(exc) or exc.__class__.__name__ redacted = appconfig.redact_secrets(raw, [self.api_key]) return diagnostics.redact_log_text(redacted) def _elapsed_ms(self, started): return _elapsed_ms(started) class AIModelTestWorker(BaseWorker): """Test one AI model connection without blocking the GUI thread.""" def __init__(self, model_name, ai_models_path=None, db_path=None, diagnostic_log_dir=None): super().__init__() self.model_name = model_name self.ai_models_path = ai_models_path or appconfig.AI_MODELS_PATH self.db_path = db_path self.diagnostic_log_dir = diagnostic_log_dir self._run_id = None def execute(self): self._run_id = self._create_run_log() started = time.monotonic() self._log_run_event( f"step=test_connection result=start detail=AI模型 {self.model_name}" ) try: result = appconfig.test_ai_model(self.model_name, path=self.ai_models_path) except Exception as exc: error = diagnostics.redact_log_text(str(exc) or exc.__class__.__name__) elapsed_ms = self._elapsed_ms(started) self._log_run_event( f"step=test_connection result=failed detail={error} elapsed_ms={elapsed_ms}", level="error", ) self._write_diagnostic_log( "AI模型测试连接异常", level="ERROR", step="test_connection", elapsed_ms=elapsed_ms, payload={"model_name": self.model_name, "error": error}, exc=exc, ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="failed", done=0, failed_count=1, summary_json={"ok": False, "name": self.model_name, "error": error}, ) raise elapsed_ms = self._elapsed_ms(started) payload = dict(appconfig.sanitize_for_log(result or {})) payload["name"] = self.model_name ok = bool(payload.get("ok")) self._log_run_event( "step=test_connection result={result} detail=AI模型 {name} status={status} error={error} elapsed_ms={elapsed_ms}".format( result="success" if ok else "failed", name=self.model_name, status=payload.get("status") or "", error=payload.get("error") or "", elapsed_ms=elapsed_ms, ), level="info" if ok else "warning", ) _safe_finish_run_log( self._run_id, db_path=self.db_path, status="done", done=1, success_count=1 if ok else 0, failed_count=0 if ok else 1, summary_json=payload, ) return payload def _create_run_log(self): if not self.db_path: return None return _safe_create_run_log( "ai_model_test", db_path=self.db_path, total=1, options={"model_name": self.model_name}, ) def _log_run_event(self, message, level="info"): safe_message = _safe_add_run_log_event( self._run_id, message, db_path=self.db_path, level=level, ) self.log.emit(str(safe_message)) def _write_diagnostic_log( self, message, level="INFO", step=None, elapsed_ms=None, payload=None, exc=None, ): _safe_write_diagnostic_log( message, level=level, step=step, elapsed_ms=elapsed_ms, payload=payload, exc=exc, log_dir=self.diagnostic_log_dir, ) def _elapsed_ms(self, started): return _elapsed_ms(started)