"""Shopee editor operations built on the verified CDP primitives.""" import json import os import time from urllib.parse import urlparse from .cdp import CDP, close_tab, create_tab, create_tab_info, find_product_tab, http_get DEFAULT_REGION_HOST = "seller.shopee.tw" TITLE_XPATH = "//input[@class='eds-input__input' and string-length(@modelvalue)>24]" ITEMBOX_XPATH = ( "//div[@class='container']/div[@class='can-drag shopee-image-manager__itembox' " "and @data-draggable='true']" ) LOGIN_PATH_MARKERS = ( "/login", "account/signin", "seller/login", "seller/accounts/signin", ) JS_READY = ( "(function(){" f"var s=document.evaluate({json.dumps(ITEMBOX_XPATH)},document,null," "XPathResult.ORDERED_NODE_SNAPSHOT_TYPE,null);" f"var r=document.evaluate({json.dumps(TITLE_XPATH)},document,null," "XPathResult.FIRST_ORDERED_NODE_TYPE,null);" "var up=document.querySelector('.shopee-image-manager__upload input[type=file]');" "return (s.snapshotLength>0 && !!r.singleNodeValue && !!up);})()" ) JS_RECTS = ( "(function(){" f"var s=document.evaluate({json.dumps(ITEMBOX_XPATH)},document,null," "XPathResult.ORDERED_NODE_SNAPSHOT_TYPE,null);" "var a=[];for(var i=0;i0&&r.height>0&&s.visibility!=='hidden'&&s.display!=='none';}" "function text(e){return ((e&&((e.innerText||e.textContent)||''))||'').trim();}" "var manager=document.querySelector('.shopee-image-manager');" "var items=manager?[].slice.call(manager.querySelectorAll('.shopee-image-manager__itembox,[class*=image-manager__itembox]')):[];" "var imgs=items.map(function(item){var img=item.querySelector('img');return img?img.src:null;}).filter(Boolean);" "var upload=manager?manager.querySelector('.shopee-image-manager__upload input[type=file]'):null;" "var uploadBox=manager?manager.querySelector('.shopee-image-manager__upload,[class*=image-manager__upload]'):null;" "var busy=manager?[].slice.call(manager.querySelectorAll('[class*=loading],[class*=Loading],[class*=spinner],[class*=Spinner],[class*=progress],[class*=Progress],[class*=uploading],[class*=Uploading]')).filter(visible):[];" "var errRe=/(失敗|失败|錯誤|错误|不支援|不支持|格式|大小|尺寸|超過|超过|error|fail|invalid|unsupported)/i;" "var uploadRe=/(圖片|图片|封面|照片|相片|圖像|图像|image|photo|cover|upload|上傳|上传|檔案|文件|file|格式|大小|尺寸|像素|解析度|分辨率|超過|超过)/i;" "var texts=manager?[].slice.call(manager.querySelectorAll('*')).filter(visible).map(text).filter(Boolean):[];" "var errors=texts.filter(function(t){return errRe.test(t);}).slice(0,8);" "var roots=[].slice.call(document.querySelectorAll('.eds-modal__content,.eds-modal__box,[role=dialog]')).filter(visible);" "var modalTexts=roots.map(text).filter(Boolean);" "var crop=modalTexts.find(function(t){return /(裁剪|裁切|剪裁|crop)/i.test(t);})||'';" "var toastTexts=[].slice.call(document.querySelectorAll('[class*=toast],[class*=Toast],[class*=message],[class*=Message],[class*=notice]')).filter(visible).map(text).filter(Boolean).slice(0,8);" "var pageErrorToasts=toastTexts.filter(function(t){return errRe.test(t);}).slice(0,8);" "var errorToasts=pageErrorToasts.filter(function(t){return uploadRe.test(t);}).slice(0,8);" "return JSON.stringify({" "manager_exists:!!manager," "count:items.length," "srcs:imgs.slice(0,12)," "blob_count:imgs.filter(function(s){return /^blob:/.test(s);}).length," "cdn_count:imgs.filter(function(s){return /susercontent/.test(s);}).length," "upload_input_exists:!!upload," "upload_input_disabled:!!(upload&&(upload.disabled||upload.getAttribute('aria-disabled')==='true'))," "upload_tile_visible:visible(uploadBox)," "busy_count:busy.length," "errors:errors," "crop_modal:!!crop," "crop_text:crop," "toasts:toastTexts," "error_toasts:errorToasts," "page_error_toasts:pageErrorToasts" "});})()" ) JS_TITLE_STATE = ( "(function(){" f"var r=document.evaluate({json.dumps(TITLE_XPATH)},document,null," "XPathResult.FIRST_ORDERED_NODE_TYPE,null);" "var el=r.singleNodeValue;if(!el)return null;" "return JSON.stringify({value:el.value,modelvalue:el.getAttribute('modelvalue')});})()" ) JS_FIRST_COVER = ( "(function(){" f"var s=document.evaluate({json.dumps(ITEMBOX_XPATH)},document,null," "XPathResult.ORDERED_NODE_SNAPSHOT_TYPE,null);" "if(!s.snapshotLength)return null;" "var img=s.snapshotItem(0).querySelector('img');return img?img.src:null;})()" ) JS_FIND_UPDATE = ( "(function(){var bs=[].slice.call(document.querySelectorAll('button.eds-button'));" "var b=bs.find(function(b){var sp=b.querySelector('span');" "return (sp?sp.innerText:(b.innerText||'')).trim()==='更新' && b.offsetParent!==null;});" "if(!b)return JSON.stringify({found:false});" "var dis=b.disabled||/disabled/i.test(b.className);" "return JSON.stringify({found:true,disabled:dis});})()" ) JS_CLICK_UPDATE = ( "(function(){var bs=[].slice.call(document.querySelectorAll('button.eds-button'));" "var b=bs.find(function(b){var sp=b.querySelector('span');" "return (sp?sp.innerText:(b.innerText||'')).trim()==='更新' && b.offsetParent!==null;});" "if(!b)return 'NO_BTN';if(b.disabled||/disabled/i.test(b.className))return 'DISABLED';" "b.click();return 'CLICKED';})()" ) JS_FIND_UPDATE_CONFIRM = ( "(function(){" "function visible(e){if(!e)return false;var r=e.getBoundingClientRect();var s=getComputedStyle(e);" "return r.width>0&&r.height>0&&s.visibility!=='hidden'&&s.display!=='none';}" "function text(e){return ((e&&((e.innerText||e.textContent)||''))||'').trim();}" "function updateTitle(t){return /(確定|确定)/.test(t)&&/更新商品/.test(t);}" "function title(root){return text(root.querySelector('.eds-modal__title,[class*=modal__title],[class*=Modal__title]'))||text(root);}" "function buttons(root){var scope=root.querySelector('.eds-modal__footer,[class*=modal__footer],[class*=Modal__footer]')||root;" "return [].slice.call(scope.querySelectorAll('button,[role=button]')).filter(visible);}" "function hasUpdateButton(root){return buttons(root).some(function(b){return text(b)==='更新'&&!b.disabled&&!/disabled/i.test(b.className);});}" "function uniq(items){var out=[];items.forEach(function(e){if(e&&out.indexOf(e)<0)out.push(e);});return out;}" "var selector='.eds-modal__content,.eds-modal__box,[role=dialog]';" "var roots=[].slice.call(document.querySelectorAll(selector));" "var titles=[].slice.call(document.querySelectorAll('.eds-modal__title,[class*=modal__title],[class*=Modal__title]'))" ".filter(visible).filter(function(t){return updateTitle(text(t));})" ".map(function(t){return t.closest(selector);});" "roots=uniq(roots.concat(titles)).filter(visible).filter(function(r){return updateTitle(title(r));}).filter(hasUpdateButton);" "if(!roots.length)return JSON.stringify({present:false});" "var root=roots[roots.length-1];" "var seen=buttons(root).map(function(b){return text(b);}).filter(Boolean);" "return JSON.stringify({present:true,title:title(root),buttons:seen.slice(0,8)});})()" ) JS_CLICK_UPDATE_CONFIRM = ( "(function(){" "function visible(e){if(!e)return false;var r=e.getBoundingClientRect();var s=getComputedStyle(e);" "return r.width>0&&r.height>0&&s.visibility!=='hidden'&&s.display!=='none';}" "function text(e){return ((e&&((e.innerText||e.textContent)||''))||'').trim();}" "function updateTitle(t){return /(確定|确定)/.test(t)&&/更新商品/.test(t);}" "function title(root){return text(root.querySelector('.eds-modal__title,[class*=modal__title],[class*=Modal__title]'))||text(root);}" "function buttons(root){var scope=root.querySelector('.eds-modal__footer,[class*=modal__footer],[class*=Modal__footer]')||root;" "return [].slice.call(scope.querySelectorAll('button,[role=button]')).filter(visible);}" "function hasUpdateButton(root){return buttons(root).some(function(b){return text(b)==='更新'&&!b.disabled&&!/disabled/i.test(b.className);});}" "function uniq(items){var out=[];items.forEach(function(e){if(e&&out.indexOf(e)<0)out.push(e);});return out;}" "var selector='.eds-modal__content,.eds-modal__box,[role=dialog]';" "var roots=[].slice.call(document.querySelectorAll(selector));" "var titles=[].slice.call(document.querySelectorAll('.eds-modal__title,[class*=modal__title],[class*=Modal__title]'))" ".filter(visible).filter(function(t){return updateTitle(text(t));})" ".map(function(t){return t.closest(selector);});" "roots=uniq(roots.concat(titles)).filter(visible).filter(function(r){return updateTitle(title(r));}).filter(hasUpdateButton);" "if(!roots.length)return JSON.stringify({present:false,clicked:false,reason:'NO_UPDATE_CONFIRM_MODAL'});" "var root=roots[roots.length-1];" "var allButtons=buttons(root);" "var seen=allButtons.map(function(b){return text(b);}).filter(Boolean);" "var exact=allButtons.filter(function(b){var t=text(b);" "return t==='更新'&&!b.disabled&&!/disabled/i.test(b.className);});" "var primary=exact.filter(function(b){return /eds-button--primary/.test(b.className);});" "var candidates=primary.length?primary:exact;" "if(!candidates.length)return JSON.stringify({present:true,clicked:false,reason:'NO_UPDATE_CONFIRM_BUTTON',buttons:seen.slice(0,8)});" "candidates[0].click();" "return JSON.stringify({present:true,clicked:true,reason:null,text:text(candidates[0]),buttons:seen.slice(0,8)});})()" ) JS_TOASTS = ( "(function(){var es=[].slice.call(document.querySelectorAll(" "'[class*=toast],[class*=Toast],[class*=message],[class*=Message],[class*=notice]'));" "var t=es.filter(function(e){return e.offsetParent!==null;})" ".map(function(e){return (e.innerText||'').trim();}).filter(Boolean);" "return JSON.stringify(t.slice(0,5));})()" ) JS_POST_UPDATE_STATE = ( "(function(){" "function visible(e){if(!e)return false;var r=e.getBoundingClientRect();var s=getComputedStyle(e);" "return r.width>0&&r.height>0&&s.visibility!=='hidden'&&s.display!=='none';}" "function text(e){return ((e&&((e.innerText||e.textContent)||''))||'').trim();}" "var toastTexts=[].slice.call(document.querySelectorAll('[class*=toast],[class*=Toast],[class*=message],[class*=Message],[class*=notice]'))" ".filter(visible).map(text).filter(Boolean).slice(0,8);" "var errRe=/(失敗|失败|錯誤|错误|不支援|不支持|無法|无法|不允許|不允许|error|fail|invalid|請|请)/i;" "var okRe=/(成功|已更新|更新完成|儲存成功|保存成功|success|saved)/i;" "var url=location.href;" "return JSON.stringify({" "url:url," "redirected_to_list:url.indexOf('/portal/product/list/')>=0," "toasts:toastTexts," "error_toasts:toastTexts.filter(function(t){return errRe.test(t);})," "success_toasts:toastTexts.filter(function(t){return okRe.test(t);})" "});})()" ) JS_CLICK_FIRST_DELETE = ( "(function(){" f"var s=document.evaluate({json.dumps(ITEMBOX_XPATH)},document,null," "XPathResult.ORDERED_NODE_SNAPSHOT_TYPE,null);" "if(!s.snapshotLength)return JSON.stringify({clicked:false,reason:'NO_IMAGE'});" "var item=s.snapshotItem(0);" "['mouseover','mouseenter','mousemove'].forEach(function(n){" "item.dispatchEvent(new MouseEvent(n,{bubbles:true,view:window}));});" "var selectors=['.shopee-image-manager__icon--delete','[class*=\"icon--delete\"]','[class*=\"delete\"]'];" "var btn=null;" "for(var i=0;i0&&r.height>0&&s.visibility!=='hidden'&&s.display!=='none';}" "var rootSelectors='[role=dialog],[class*=dialog],[class*=Dialog],[class*=modal]," "[class*=Modal],[class*=popover],[class*=Popover],[class*=popup],[class*=Popup]';" "var roots=[].slice.call(document.querySelectorAll(rootSelectors)).filter(visible);" "var yes=/(删除|刪除|確認|确认|確定|确定|OK|Yes)/i;" "var no=/(取消|cancel|否|No)/i;" "var seen=[];var candidates=[];" "roots.forEach(function(root){[].slice.call(root.querySelectorAll('button,[role=button]')).forEach(function(b){" "var text=((b.innerText||b.textContent||b.getAttribute('aria-label')||'')+'').trim();" "if(!text||!visible(b))return;seen.push(text);" "if(b.disabled||/disabled/i.test(b.className)||no.test(text)||!yes.test(text))return;" "candidates.push({button:b,text:text,score:/(删除|刪除)/i.test(text)?0:1});});});" "if(!candidates.length)return JSON.stringify({clicked:false,reason:'NO_CONFIRM_BUTTON',buttons:seen.slice(0,8)});" "candidates.sort(function(a,b){return a.score-b.score;});" "candidates[0].button.click();" "return JSON.stringify({clicked:true,reason:null,text:candidates[0].text});})()" ) class EditorError(RuntimeError): """Raised for expected editor automation failures with user-readable messages.""" def _get(obj, *names, default=None): for name in names: if isinstance(obj, dict) and name in obj: return obj[name] if hasattr(obj, name): return getattr(obj, name) return default def _cdp_host(account=None): explicit = _get(account, "cdp_host", "debug_host") if explicit: return str(explicit).replace("http://", "").replace("https://", "").rstrip("/") port = _get(account, "debug_port", "port") if port: return f"127.0.0.1:{port}" return os.environ.get("CDP_HOST", "127.0.0.1:9222") def _region_host(account=None): value = _get(account, "region_host", "host", default=DEFAULT_REGION_HOST) value = str(value or DEFAULT_REGION_HOST).strip() if "://" in value: value = urlparse(value).netloc return value.strip("/") or DEFAULT_REGION_HOST def _seller_home_url(account): return f"https://{_region_host(account)}/" def _product_url(account, item_id): return ( f"https://{_region_host(account)}/portal/product/{item_id}" "?pageEntry=product_list&ignore-html-cache=1" ) def _item_id(task): value = _get(task, "item_id", "itemid", "product_id") if not value: raise EditorError("缺少商品 id，无法打开商品页") return str(value) def _image_root(account): alias = _get(account, "slug", "alias", "account_name", default="default") safe = "".join(ch.lower() if ch.isalnum() else "_" for ch in str(alias)).strip("_") return os.path.join("images", safe or "default") def _is_login_url(url): value = (url or "").lower() return any(marker in value for marker in LOGIN_PATH_MARKERS) def _json_value(cdp, expr, default=None): raw = cdp.val(expr) if not raw: return default return json.loads(raw) def _title_state(cdp): return _json_value(cdp, JS_TITLE_STATE, default={}) or {} def _image_rects(cdp): return _json_value(cdp, JS_RECTS, default=[]) or [] def _upload_state(cdp): return _json_value(cdp, JS_UPLOAD_STATE, default={}) or {} def _post_update_state(cdp): return _json_value(cdp, JS_POST_UPDATE_STATE, default=None) or {} def _wait_post_update(cdp, timeout=8): end = time.time() + timeout last_state = {} last_error_state = None while time.time() < end: last_state = _post_update_state(cdp) if not last_state: return {"ok": True, "observed_success": False, "observed_failure": False} error_toasts = last_state.get("error_toasts") or [] success_toasts = last_state.get("success_toasts") or [] redirected = bool(last_state.get("redirected_to_list")) if redirected or success_toasts: last_state.update({"ok": True, "reason": None, "observed_success": True, "observed_failure": False}) return last_state if error_toasts: last_error_state = dict(last_state) time.sleep(0.5) last_state = last_state or _post_update_state(cdp) if not last_state: return {"ok": True, "observed_success": False, "observed_failure": False} success_toasts = last_state.get("success_toasts") or [] redirected = bool(last_state.get("redirected_to_list")) if redirected or success_toasts: last_state.update({"ok": True, "reason": None, "observed_success": True, "observed_failure": False}) return last_state error_state = last_error_state or (last_state if last_state.get("error_toasts") else None) if error_state: error_state.update({"ok": False, "reason": "POST_UPDATE_ERROR", "observed_success": False, "observed_failure": True}) return error_state last_state.update({"ok": True, "reason": None, "observed_success": False, "observed_failure": False}) return last_state def _upload_input_ready(state): return bool((state or {}).get("upload_input_exists")) and not bool((state or {}).get("upload_input_disabled")) def _unstable_image_manager_reason(state, rects, expected_count, require_upload_input=True): state = state or {} if state.get("crop_modal"): return "UPLOAD_CROP_REQUIRED" if state.get("errors") or state.get("error_toasts"): return "UPLOAD_PAGE_ERROR" if expected_count is not None and len(rects) != expected_count: return "IMAGE_COUNT_NOT_READY" if state.get("busy_count") or any(str(r.get("src") or "").startswith("blob:") for r in rects): return "IMAGE_MANAGER_BUSY" if require_upload_input and not _upload_input_ready(state): return "UPLOAD_INPUT_NOT_READY" return "IMAGE_MANAGER_NOT_STABLE" def _wait_image_manager_stable( cdp, expected_count=None, timeout=20, settle_seconds=1.0, poll=0.5, require_upload_input=True, ): end = time.time() + timeout last_rects = [] last_state = {} last_signature = None stable_hits = 0 while time.time() < end: last_rects = _image_rects(cdp) last_state = _upload_state(cdp) signature = tuple(r.get("src") for r in last_rects) count_ok = expected_count is None or len(last_rects) == expected_count no_blob = not any(str(r.get("src") or "").startswith("blob:") for r in last_rects) ready = ( bool(last_rects) and count_ok and no_blob and not last_state.get("busy_count") and not last_state.get("crop_modal") and not last_state.get("errors") and not last_state.get("error_toasts") and (not require_upload_input or _upload_input_ready(last_state)) ) if ready and signature == last_signature: stable_hits += 1 elif ready: last_signature = signature stable_hits = 1 else: last_signature = signature stable_hits = 0 if stable_hits >= 2: if settle_seconds: time.sleep(settle_seconds) return {"ok": True, "rects": last_rects, "upload_state": last_state} time.sleep(poll) return { "ok": False, "reason": _unstable_image_manager_reason( last_state, last_rects, expected_count, require_upload_input=require_upload_input, ), "rects": last_rects, "upload_state": last_state, "expected_count": expected_count, "require_upload_input": require_upload_input, "count_after": len(last_rects), } def _cover_upload_error_message(result): reason = result.get("reason") or "COVER_UPDATE_FAILED" state = result.get("upload_state") or {} if reason == "UPLOAD_PAGE_ERROR": details = state.get("errors") or state.get("error_toasts") or state.get("toasts") or [] return "新封面上传失败：" + "；".join(map(str, details[:3])) if details else "新封面上传失败" if reason == "UPLOAD_CROP_REQUIRED": return "新封面上传后出现裁剪确认框，当前版本未自动处理裁剪弹窗" if reason == "UPLOAD_STILL_PROCESSING": return "新封面上传仍在处理中，未取得 Shopee CDN 地址" if reason == "UPLOAD_TIMEOUT": return "新封面上传超时，未取得 Shopee CDN 地址" if reason == "IMAGE_MANAGER_BUSY": return "商品图片区域仍在加载，未开始上传新封面" if reason == "IMAGE_COUNT_NOT_READY": return "删除旧封面后图片数量未稳定，未开始上传新封面" if reason == "UPLOAD_INPUT_NOT_READY": return "商品图片上传入口未恢复可用，未开始上传新封面" if reason == "IMAGE_MANAGER_NOT_STABLE": return "商品图片区域未稳定，未开始上传新封面" return str(reason) def _wait_ready(cdp, timeout=60): end = time.time() + timeout while time.time() < end: try: if cdp.val(JS_READY): cdp.val( "(function(){var m=document.querySelector('.shopee-image-manager');" "if(m)m.scrollIntoView({block:'center'});return 1;})()" ) time.sleep(0.5) return True except Exception: pass time.sleep(1) raise TimeoutError("等待 Shopee 商品编辑器就绪超时") def _ensure_page_domains(cdp): for domain in ("Page", "Runtime", "DOM", "Network"): try: cdp.send(f"{domain}.enable") except Exception: pass def _current_url(cdp): try: return cdp.val("location.href") or "" except Exception: return "" def _wait_for_login_probe(cdp, timeout=8): end = time.time() + timeout last_url = "" cookie_names = set() while time.time() < end: last_url = _current_url(cdp) or last_url try: cookies = cdp.send("Network.getAllCookies").get("cookies", []) cookie_names = { c.get("name") for c in cookies if "shopee" in (c.get("domain") or "") } except Exception: cookie_names = set() if _is_login_url(last_url) or "SPC_ST" in cookie_names or "SPC_U" in cookie_names: break time.sleep(0.5) return last_url, cookie_names def login_status(account, timeout=8) -> dict: """Return detailed Shopee login status for an account's CDP session.""" host = _cdp_host(account) pages = [t for t in http_get("/json", host=host) if t.get("type") == "page"] shopee_page = next((p for p in pages if "shopee" in (p.get("url") or "")), None) if not shopee_page: ws = create_tab(_seller_home_url(account), host=host) initial_url = _seller_home_url(account) cdp = CDP(ws) else: initial_url = shopee_page.get("url") or "" if _is_login_url(initial_url): return { "logged_in": False, "reason": "LOGIN_PAGE", "url": initial_url, "host": host, "cookie_names": [], } cdp = CDP(shopee_page["webSocketDebuggerUrl"]) try: _ensure_page_domains(cdp) url, names = _wait_for_login_probe(cdp, timeout=timeout) url = url or initial_url if _is_login_url(url): reason = "LOGIN_PAGE" logged_in = False elif "SPC_ST" in names or "SPC_U" in names: reason = None logged_in = True else: reason = "NO_SESSION_COOKIE" logged_in = False return { "logged_in": logged_in, "reason": reason, "url": url, "host": host, "cookie_names": sorted(name for name in names if name), } finally: cdp.close() def is_logged_in(account) -> bool: """Return whether the account's current Shopee session appears logged in.""" return bool(login_status(account).get("logged_in")) def open_product(account, item_id, on_step=None) -> CDP: """Open or reuse a product edit tab, navigate to a clean edit URL, and wait ready.""" _notify_collect_step(on_step, "open_product") host = _cdp_host(account) item_id = str(item_id) url = _product_url(account, item_id) tab = find_product_tab(item_id, host=host) created_by_app = tab is None if tab is None: tab = create_tab_info(url, host=host) ws = tab["webSocketDebuggerUrl"] cdp = CDP(ws) cdp.target_id = tab.get("id") cdp.created_by_app = created_by_app cdp.cdp_host = host _ensure_page_domains(cdp) try: cdp.send("Page.bringToFront") except Exception: pass cdp.send("Page.navigate", {"url": url}) _notify_collect_step(on_step, "wait_ready") _wait_ready(cdp) return cdp def read_title(cdp) -> str: """Read the current Shopee title input value.""" state = _title_state(cdp) return state.get("value") or state.get("modelvalue") or "" def read_cover_src(cdp) -> str: """Read the first product image URL, which is the current cover.""" return cdp.val(JS_FIRST_COVER) or "" def download_cover(src, out_path) -> str: """Download a cover image to a local path and return the absolute path.""" if not src: raise EditorError("旧封面链接为空，无法下载") import requests out_path = os.path.abspath(str(out_path)) root, ext = os.path.splitext(out_path) if not ext: path = urlparse(src).path _, guessed = os.path.splitext(path) out_path = root + (guessed if guessed and len(guessed) <= 5 else ".jpg") os.makedirs(os.path.dirname(out_path), exist_ok=True) session = requests.Session() session.trust_env = False resp = session.get(src, timeout=30) resp.raise_for_status() with open(out_path, "wb") as fh: fh.write(resp.content) return out_path def collect(account, task, on_step=None) -> dict: """Collect current title and cover snapshot before any edits.""" item_id = _item_id(task) cdp = open_product(account, item_id, on_step=on_step) try: _notify_collect_step(on_step, "read_title") old_title = read_title(cdp) _notify_collect_step(on_step, "read_cover") old_cover_src = read_cover_src(cdp) out_path = _get(task, "old_cover_path") if not out_path: out_path = os.path.join(_image_root(account), f"{item_id}_old.jpg") _notify_collect_step(on_step, "download_cover") old_cover_path = download_cover(old_cover_src, out_path) return { "old_title": old_title, "old_cover_src": old_cover_src, "old_cover_path": old_cover_path, } finally: _close_collected_product(cdp) def _notify_collect_step(callback, step): if callback is None: return try: callback(step) except Exception: pass def _close_collected_product(cdp): target_id = getattr(cdp, "target_id", None) created_by_app = bool(getattr(cdp, "created_by_app", False)) host = getattr(cdp, "cdp_host", None) try: cdp.close() finally: if created_by_app and target_id: try: close_tab(target_id, host=host) except Exception: pass def change_title(cdp, new_title) -> dict: """Write a new title with the verified native setter + input/change events.""" expr = ( "(function(){" f"var r=document.evaluate({json.dumps(TITLE_XPATH)},document,null," "XPathResult.FIRST_ORDERED_NODE_TYPE,null);var el=r.singleNodeValue;" "if(!el)return 'NO_INPUT';" "var s=Object.getOwnPropertyDescriptor(window.HTMLInputElement.prototype,'value').set;" f"el.focus();s.call(el,{json.dumps(str(new_title))});" "el.dispatchEvent(new Event('input',{bubbles:true}));" "el.dispatchEvent(new Event('change',{bubbles:true}));el.blur();return el.value;})()" ) written = cdp.val(expr) time.sleep(0.4) state = _title_state(cdp) value = state.get("value") modelvalue = state.get("modelvalue") ok = value == str(new_title) and modelvalue == str(new_title) return {"ok": ok, "written": written, "value": value, "modelvalue": modelvalue} def replace_cover(cdp, image_win_path, old_cover_path=None, timeout=180) -> dict: """Upload an image and drag it to the first position. When the image manager is full, only delete the current first image if the old cover backup from the collect stage exists locally. """ image_win_path = os.path.abspath(str(image_win_path)) if not os.path.exists(image_win_path): raise FileNotFoundError(f"封面图片不存在: {image_win_path}") cdp.val( "(function(){var m=document.querySelector('.shopee-image-manager');" "if(m)m.scrollIntoView({block:'center'});return !!m;})()" ) time.sleep(0.5) stable = _wait_image_manager_stable( cdp, timeout=20, settle_seconds=1.0, require_upload_input=False, ) if not stable.get("ok"): rects = stable.get("rects") or [] return { "ok": False, "reason": stable.get("reason"), "count_before": len(rects), "count_after": len(rects), "upload_state": stable.get("upload_state"), "stable": stable, } before = stable.get("rects") or _image_rects(cdp) count_before = len(before) delete_result = None if len(before) >= 9: backup_path = _validated_old_cover_backup(old_cover_path) if not backup_path: return { "ok": False, "reason": "OLD_COVER_BACKUP_MISSING", "count_before": count_before, "old_cover_path": old_cover_path, } delete_result = _delete_first_cover(cdp, before) if not delete_result.get("ok"): return { "ok": False, "reason": delete_result.get("reason"), "count_before": count_before, "delete": delete_result, } stable = _wait_image_manager_stable( cdp, expected_count=delete_result.get("count_after"), timeout=30, settle_seconds=2.0, require_upload_input=True, ) if not stable.get("ok"): return { "ok": False, "reason": stable.get("reason"), "count_before": count_before, "count_after": stable.get("count_after"), "delete": delete_result, "upload_state": stable.get("upload_state"), "stable": stable, } before = stable.get("rects") or _image_rects(cdp) else: stable = _wait_image_manager_stable( cdp, expected_count=len(before), timeout=20, settle_seconds=1.0, require_upload_input=True, ) if not stable.get("ok"): return { "ok": False, "reason": stable.get("reason"), "count_before": count_before, "count_after": stable.get("count_after"), "upload_state": stable.get("upload_state"), "stable": stable, } before = stable.get("rects") or _image_rects(cdp) before_srcs = {r.get("src") for r in before} oid = cdp.object_id("document.querySelector('.shopee-image-manager__upload input[type=file]')") if not oid: return { "ok": False, "reason": "NO_UPLOAD_INPUT", "count_before": count_before, "delete": delete_result, "upload_state": _upload_state(cdp), } cdp.send("DOM.setFileInputFiles", {"objectId": oid, "files": [image_win_path]}) changed = cdp.val( "(function(){var up=document.querySelector('.shopee-image-manager__upload input[type=file]');" "if(!up)return false;" "up.dispatchEvent(new Event('input',{bubbles:true}));" "up.dispatchEvent(new Event('change',{bubbles:true}));return true;})()" ) time.sleep(2) new_src = None last_state = _upload_state(cdp) if not changed and not last_state.get("busy_count") and not last_state.get("blob_count"): return { "ok": False, "reason": "UPLOAD_INPUT_NOT_READY", "count_before": count_before, "delete": delete_result, "upload_state": last_state, } blob_seen = bool(last_state.get("blob_count")) file_size = os.path.getsize(image_win_path) if os.path.exists(image_win_path) else None end = time.time() + timeout while time.time() < end: cur = _image_rects(cdp) last_state = _upload_state(cdp) blob_seen = blob_seen or any(str(r.get("src") or "").startswith("blob:") for r in cur) ready = [ r for r in cur if r.get("src") not in before_srcs and r.get("src") and "susercontent" in r.get("src") and "blob:" not in r.get("src") ] if ready and (len(cur) > len(before) or len(ready) == 1): new_src = ready[-1]["src"] break if (last_state.get("errors") or last_state.get("error_toasts")) and not last_state.get("busy_count"): return { "ok": False, "reason": "UPLOAD_PAGE_ERROR", "count_before": count_before, "count_after": len(cur), "delete": delete_result, "upload_state": last_state, "file_size": file_size, } if last_state.get("crop_modal"): return { "ok": False, "reason": "UPLOAD_CROP_REQUIRED", "count_before": count_before, "count_after": len(cur), "delete": delete_result, "upload_state": last_state, "file_size": file_size, } time.sleep(1.5) if not new_src: cur = _image_rects(cdp) reason = "UPLOAD_STILL_PROCESSING" if blob_seen or (last_state or {}).get("busy_count") else "UPLOAD_TIMEOUT" return { "ok": False, "reason": reason, "count_before": count_before, "count_after": len(cur), "delete": delete_result, "upload_state": last_state, "blob_seen": blob_seen, "file_size": file_size, } time.sleep(1) cur = _image_rects(cdp) new_rect = next((r for r in cur if r.get("src") == new_src), None) if not new_rect: return {"ok": False, "reason": "NEW_IMAGE_NOT_FOUND", "new_src": new_src, "delete": delete_result} first = cur[0] cdp.drag(new_rect["x"], new_rect["y"], first["left"] - first["w"] * 0.30, first["y"]) time.sleep(1.2) after = _image_rects(cdp) cover_ok = bool(after and after[0].get("src") == new_src) index = next((r["i"] for r in after if r.get("src") == new_src), None) return { "ok": cover_ok, "reason": None if cover_ok else "DRAG_NOT_FIRST", "new_src": new_src, "index": index, "count_before": count_before, "count_after": len(after), "delete": delete_result, } def _validated_old_cover_backup(old_cover_path): if not old_cover_path: return None path = os.path.abspath(str(old_cover_path)) if not os.path.exists(path): return None return path def _delete_first_cover(cdp, before, timeout=15): count_before = len(before) if not before: return {"ok": False, "reason": "NO_IMAGE", "count_before": 0} first = before[0] try: cdp.send( "Input.dispatchMouseEvent", {"type": "mouseMoved", "x": first["x"], "y": first["y"]}, ) time.sleep(0.2) except Exception: pass click_result = _json_value( cdp, JS_CLICK_FIRST_DELETE, default={"clicked": False, "reason": "NO_DELETE_BUTTON"}, ) or {} if not click_result.get("clicked"): return { "ok": False, "reason": click_result.get("reason") or "NO_DELETE_BUTTON", "count_before": count_before, "click": click_result, } confirm_result = None end = time.time() + timeout while time.time() < end: cur = _image_rects(cdp) if len(cur) < count_before: return { "ok": True, "reason": None, "count_before": count_before, "count_after": len(cur), "click": click_result, "confirm": confirm_result, } confirm_attempt = _json_value( cdp, JS_CLICK_DELETE_CONFIRM, default={"clicked": False, "reason": "NO_CONFIRM_BUTTON"}, ) or {} if confirm_attempt.get("clicked"): confirm_result = confirm_attempt time.sleep(0.5) cur = _image_rects(cdp) return { "ok": False, "reason": "DELETE_TIMEOUT", "count_before": count_before, "count_after": len(cur), "click": click_result, "confirm": confirm_result, } def click_update(cdp, confirm_timeout=3, post_timeout=8) -> dict: """Click the Shopee update button, confirm Shopee's final modal, and observe submit result.""" info = _json_value(cdp, JS_FIND_UPDATE, default={"found": False}) if not info.get("found"): return {"clicked": False, "reason": "NO_UPDATE_BUTTON", "toasts": []} if info.get("disabled"): return {"clicked": False, "reason": "UPDATE_DISABLED", "toasts": []} result = cdp.val(JS_CLICK_UPDATE) if result != "CLICKED": toasts = _json_value(cdp, JS_TOASTS, default=[]) or [] return {"clicked": False, "reason": result, "toasts": toasts} confirm_result = _confirm_update_modal(cdp, timeout=confirm_timeout) if confirm_result.get("present") and not confirm_result.get("clicked"): toasts = _json_value(cdp, JS_TOASTS, default=[]) or [] return { "clicked": False, "reason": confirm_result.get("reason") or "UPDATE_CONFIRM_NOT_CLICKED", "toasts": toasts, "confirm": confirm_result, } post_update = _wait_post_update(cdp, timeout=post_timeout) toasts = post_update.get("toasts") if post_update else None if toasts is None: toasts = _json_value(cdp, JS_TOASTS, default=[]) or [] if post_update.get("observed_failure"): return { "clicked": False, "reason": post_update.get("reason") or "POST_UPDATE_ERROR", "toasts": toasts, "confirm": confirm_result, "post_update": post_update, } return { "clicked": True, "reason": None, "toasts": toasts, "confirm": confirm_result, "post_update": post_update, } def _confirm_update_modal(cdp, timeout=3): end = time.time() + timeout last_present = None while time.time() < end: state = _json_value( cdp, JS_FIND_UPDATE_CONFIRM, default={"present": False}, ) or {"present": False} if state.get("present"): confirm_attempt = _json_value( cdp, JS_CLICK_UPDATE_CONFIRM, default={"present": True, "clicked": False, "reason": "NO_UPDATE_CONFIRM_BUTTON"}, ) or {} last_present = confirm_attempt if confirm_attempt.get("clicked"): time.sleep(1) return confirm_attempt time.sleep(0.5) return last_present or {"present": False, "clicked": False, "reason": "NO_UPDATE_CONFIRM_MODAL"} def apply_task(account, task, close_success_tab=False) -> dict: """Apply generated title/cover to Shopee. The caller must perform the batch confirmation before calling this function. """ item_id = _item_id(task) cdp = open_product(account, item_id) committed = False try: title_result = None cover_result = None new_title = _get(task, "new_title") new_cover_path = _get(task, "new_cover_path") if new_title: title_result = change_title(cdp, new_title) if not title_result.get("ok"): return {"committed": False, "error": "标题写入后 value/modelvalue 未同步", "title": title_result} if new_cover_path: cover_result = replace_cover( cdp, new_cover_path, old_cover_path=_get(task, "old_cover_path"), ) if not cover_result.get("ok"): return {"committed": False, "error": _cover_upload_error_message(cover_result), "cover": cover_result} update_result = click_update(cdp) committed = bool(update_result.get("clicked", False)) return { "committed": committed, "error": update_result.get("reason"), "title": title_result, "cover": cover_result, "update": update_result, } except Exception as exc: return {"committed": False, "error": str(exc)} finally: _close_applied_product(cdp, close_success_tab=close_success_tab and committed) def _close_applied_product(cdp, close_success_tab=False): target_id = getattr(cdp, "target_id", None) created_by_app = bool(getattr(cdp, "created_by_app", False)) host = getattr(cdp, "cdp_host", None) try: cdp.close() finally: if close_success_tab and created_by_app and target_id: try: close_tab(target_id, host=host) except Exception: pass