1075 lines
42 KiB
Python
1075 lines
42 KiB
Python
"""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;i<s.snapshotLength;i++){"
|
||
"var el=s.snapshotItem(i);var r=el.getBoundingClientRect();var im=el.querySelector('img');"
|
||
"a.push({i:i,x:r.left+r.width/2,y:r.top+r.height/2,left:r.left,top:r.top,"
|
||
"w:r.width,h:r.height,src:im?im.src:null});}"
|
||
"return JSON.stringify(a);})()"
|
||
)
|
||
|
||
|
||
JS_UPLOAD_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 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;i<selectors.length&&!btn;i++){btn=item.querySelector(selectors[i]);}"
|
||
"if(!btn){var all=[].slice.call(item.querySelectorAll('*'));"
|
||
"btn=all.find(function(e){var t=[e.className,e.innerText,e.title,e.getAttribute('aria-label')].join(' ');"
|
||
"return /(delete|删除|刪除)/i.test(t);});}"
|
||
"if(!btn)return JSON.stringify({clicked:false,reason:'NO_DELETE_BUTTON'});"
|
||
"if(btn.scrollIntoView)btn.scrollIntoView({block:'center'});"
|
||
"btn.click();return JSON.stringify({clicked:true,reason:null});})()"
|
||
)
|
||
|
||
JS_CLICK_DELETE_CONFIRM = (
|
||
"(function(){"
|
||
"function visible(e){var r=e.getBoundingClientRect();var s=getComputedStyle(e);"
|
||
"return r.width>0&&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
|