9i6ndZddlZddlZddlmZddlmZddlmZddl m Z e e Z gdZ gdZgd Zgd Zgd Zgd Zd ZdZedZdZdZdZGddZdS)uZhihu Playwright Engine — authentication and answer publishing. Handles: - Persistent session via zhihu_state.json (cookies) - QR-code login flow with Slack notification + 2-minute polling - Answer publishing via headless Chromium + playwright-stealth N)Path)Optional) WebClient) setup_logger)z.SignFlow-qrcode imgz.LoginQrcode imgzcanvas.SignFlow-qrcodeCanvasz.SignFlow-qrcode canvas)u)div.SignFlow-tab:has-text('扫码登录')u"li.SignFlow-tab:has-text('扫码')uspan:has-text('扫码登录'))z.AppHeader-profileAvatarz;div[data-za-detail-view-path-module='AppHeader'] img.Avatarz .TopstoryTabs) ubutton:has-text('写回答')ua:has-text('写回答')ubutton:has-text('编辑回答')ua:has-text('编辑回答')u%[role='button']:has-text('写回答')u([role='button']:has-text('编辑回答')z0[data-za-detail-view-element-name='WriteAnswer']z3[data-za-element-name='QuestionDetail_WriteAnswer']z%.QuestionDetailPageHeaderRight buttonz".QuestionDetail-buttonGroup buttonz[class*='WriteAnswer'])z,.DraftEditor-content[contenteditable='true']z".RichText [contenteditable='true']z3[contenteditable='true'].public-DraftEditor-content)ubutton:has-text('发布回答')u%.AnswerForm button:has-text('发布')u%.AnswerForm button:has-text('提交')ubutton:has-text('发布')ubutton:has-text('提交')zbutton.SubmitAnswerz%[data-za-element-name='SubmitAnswer']zhttps://www.zhihu.com/signinzhttps://www.zhihu.com/z/tmp/zhihu_qr.pngxc eZdZdZ ddededeeddfdZde fdZ d ed edefd Z de fd Z dd Z de fdZde fdZddZdeefdZdededdfdZde fdZddZdeddfdZdefdZdeddfdZdS)ZhihuPlaywrightEnginezPlaywright-powered Zhihu auth and answer publisher. Example: engine = ZhihuPlaywrightEngine(slack_client, channel_id) if engine.ensure_logged_in(): engine.publish_answer(question_url, answer_text) N slack_clientnotify_channel state_filereturnc||_||_|p6ttjddz dz |_|jjdddS)aCInitialize the engine. Args: slack_client: Authenticated Slack WebClient for QR upload. notify_channel: Slack channel/DM ID to send QR images to. state_file: Override path for session cookie persistence. Defaults to /data/zhihu_state.json. datazzhihu_state.jsonT)parentsexist_okN) r r r__file__resolverrparentmkdir)selfr r rs @/root/projects/butler/slack_bot/zhihu/zhihu_playwright_engine.py__init__zZhihuPlaywrightEngine.__init___sq),$ NN " " $ $ ,Q /& 8;M M  $$TD$AAAAAcddlm}ddlm}|5}|jd}|dddd }|}||| |rl| |r\'")\].,;]+$/zNavigating to question: domcontentloaded: wait_untiltimeouti0u signinloginuu知乎 session 已失效，导航时跳转至登录页。请重新点击「确认发布」重新触发扫码登录。uC() => { const t = document.title || ''; const b = document.body ? document.body.textContent : ''; return t.includes('404') || b.includes('你访问的页面不存在') || b.includes('页面不存在'); }uB问题页面返回 404，可能该问题已删除或 URL 有误：)r-rr.r!r/r0r1r4 RuntimeErrorr2r3researchsubgrouprstripr6r7gotoZHIHU_HOME_URLwait_for_timeouturlevaluate_click_write_answer _fill_answer_submit_answer _save_stater8) rr@rArr!r:r;r<r=_re_mis_404 answer_urls rpublish_answerz$ZhihuPlaywrightEngine.publish_answers% 877777...... _  C !j'''66G))#'3775*G9 ''00&W'')) ,,T222!   ZZ>>R&'P 'P'PQQQ"ww'8"bhhqkkJJQQRUVV  E|EEFFF  .5GQV WWW%%d+++ ,3Eu UUU%%d+++tx''7dh+>+>&W&k]ikk((...!!$ 444!0066   )))! GC C C C C C C C F GC C C C C C C C C C s*6I$GI *I$ I!!I$$I(+I(c|jsdS tj|jd}|dg}|sdS||tdt|d|jdS#t$r(}t d|Yd }~dSd }~wwxYw) zRestore saved cookies into a browser context. Args: context: Playwright BrowserContext to inject cookies into. Returns: True if state file existed and cookies were loaded. Futf-8encodingcookieszLoaded z cookies from TzState load failed: N) rexistsjsonloads read_textget add_cookiesr6debuglen Exceptionwarning)rr<rrfes rr4z!ZhihuPlaywrightEngine._load_states%%'' 5 :do777IIJJDhhy"--G u    ( ( ( LLP3w<<PPtPP Q Q Q4    NN444 5 5 555555 sAB-$AB-- C7CCc(|tjd}|jt j|dddt dt|dd |jd S) zPersist current browser cookies to the state file. Args: context: Playwright BrowserContext after successful authentication. )rfsaved_atFr ) ensure_asciiindentrcrdzSaved rfz cookies to N) rftimer write_textrhdumpsr6r7rn)rr<states rr\z!ZhihuPlaywrightEngine._save_states((    "" Ju5 ; ; ; #     QSy!122QQQQRRRRRrcN |tddtD]I} ||dtd|dS#t $rYFwxYwn4#t $r'}td|Yd }~nd }~wwxYwd S) zNavigate to Zhihu home and verify login-specific elements appear. Args: page: Playwright Page with cookies loaded. Returns: True if logged-in UI elements are found. rErFrGrIzLogin confirmed via selector: TzLogin check navigation failed: NF)rTrU_LOGIN_CHECK_SELECTORSwait_for_selectorr6rmro)rr=selrqs rr5z#ZhihuPlaywrightEngine._is_logged_ins @ IIn1CUI S S S-  **3*===LL!G#!G!GHHH44 H   @ @ @ LL>1>> ? ? ? ? ? ? ? ? @us:&A14A A1 A-*A1,A--A11 B";BB"c|tdd||tdtdzD]}t d|dt||}||d|d d S| ||| |r-| ||d d S|tkrEt d | d|||dtdd S)aExecute the full QR login loop (up to _MAX_QR_ATTEMPTS refreshes). Navigates to sign-in, switches to QR tab, screenshots the code, uploads to Slack, then polls for login success. Args: context: Playwright BrowserContext (for saving state on success). page: Active Playwright Page. Returns: True if login succeeded before all attempts exhausted. rErFrGzQR login attempt rDNu&❌ 无法截取知乎二维码（第 u 次），放弃登录。Fu,✅ 知乎登录成功，Cookie 已保存。Tz0QR timed out, refreshing page for new QR code...)rHu,❌ 知乎二维码扫码超时（已重试 u> 次）。 请稍后重新发起 `mode zhihu-hunter` 流程。) rTZHIHU_SIGNIN_URL_switch_to_qr_tabrange_MAX_QR_ATTEMPTSr6r7 _capture_qr _notify_slack_send_qr_to_slack_poll_for_loginr\reload)rr<r=attemptqr_paths rr9z$ZhihuPlaywrightEngine._qr_login_flow2s "/A5 QQQ t$$$Q 01 455 - -G KKHGHH6FHH I I I&&t,,G""#nG#n#n#nooouu  " "7G 4 4 4##D))   )))""#QRRRtt))) NOOO '9 :::&&t,,,  B;K B B B   urc.tD]} ||j}|drI||dt d|dS}#t$rYwxYwdS)zClick the QR-code tab on the login page if it exists. Args: page: Page showing the Zhihu sign-in form. rJr|r%zSwitched to QR tab via: N) _QR_TAB_SELECTORSlocatorfirst is_visibleclickrVr6rmro)rr=rtabs rrz'ZhihuPlaywrightEngine._switch_to_qr_tab]s %  C ll3''->>$>//IIKKK))#...LL!AC!A!ABBBFF        sA6B BBcDtD]} ||d||j}|t t td|dt t cS#t$rYwxYwt d |t t dt S#t$r(}t d |Yd }~d Sd }~wwxYw) aLScreenshot the QR code element and save it to a temp file. Tries specific element selectors first; falls back to a full-page screenshot if none match. Args: page: Page currently showing the QR code. Returns: Path to the saved PNG file, or None on complete failure. @r|)pathzQR captured via selector 'u' → zAElement QR selectors failed; falling back to full-page screenshotFr full_pagez"Full-page screenshot also failed: N) _QR_IMAGE_SELECTORSr~rr screenshotstr _QR_TMP_PATHr6r7rorperror)rr=relemrqs rrz!ZhihuPlaywrightEngine._capture_qrns5'  C &&sD&999||C((.S%6%6777 RRRLRRSSS####     Z[[[  OO\!2!2eO D D D     LLAaAA B B B44444 s*BB BB=/C-- D7DDrrc ld|dtd} |d5}|j|j|dd|d|d d d n #1swxYwYt d d S#t$r(}t d |Yd }~d Sd }~wwxYw) zUpload the QR code PNG to Slack with instructions. Args: qr_path: Local path to the QR code image. attempt: Current attempt number shown in the message. u@🔑 *请用知乎 App 扫描以下二维码完成登录*（第 rDuE 次） ⏱ 二维码有效期约 2 分钟，超时将自动刷新。rbz zhihu_qr.pngu知乎登录二维码（第 u 次）)channelfilefilenametitleinitial_commentNzQR image uploaded to SlackzSlack QR upload failed: ) ropenr files_upload_v2r r6r7ror)rrrcaptionfrqs rrz'ZhihuPlaywrightEngine._send_qr_to_slacks= Mw M MYi M M M  9d## q!11 /+IIII$+ 2                KK4 5 5 5 5 5 9 9 9 LL7A77 8 8 8 8 8 8 8 8 8 9s:B)A BAB!A"B B3 B..B3ctjtz}tj|kr |dtdzt d|jdS#t$rDt|tjz }t d|dYnwxYwtj|kdS) a.Poll until the page leaves the sign-in URL or timeout expires. Checks every _POLL_INTERVAL_S seconds for up to _LOGIN_TIMEOUT_S total. Args: page: Page that started at ZHIHU_SIGNIN_URL. Returns: True if login was detected before the deadline. cd|vod|vS)Nz zhihu.comrL)rWs rz7ZhihuPlaywrightEngine._poll_for_login..s s 2 Jxs7Jrir|u"Login confirmed — redirected to TzWaiting for QR scan... zs leftF) rv monotonic_LOGIN_TIMEOUT_S wait_for_url_POLL_INTERVAL_Sr6r7rWrointrm)rr=deadline remainings rrz%ZhihuPlaywrightEngine._poll_for_logins>##&66n)) J!!JJ,t3" KKKLLLt J J J4>+;+; ;<<  HyHHHIIIII Jn))usAA88A CCc6|d|dtD]} ||j}|dd||dtd|dS#t$rYwxYw|d }t d |dd |d }|r5|dt d |ddS ddl }t|d}|t!|d|j|jt!|dd|dn4#t$r'}t d|Yd}~nd}~wwxYwt+d)uFind and click the 写回答 button on a question page. Args: page: Page showing a Zhihu question. Raises: RuntimeError: If no write-answer button is found by any strategy. zwindow.scrollBy(0, 400)r%visiblerKryrIiuClicked '写回答' via: Na() => { const els = document.querySelectorAll('button, a, [role="button"]'); return Array.from(els) .map(e => e.textContent.trim()) .filter(t => t.length > 0 && t.length < 30); }u.'写回答' not found. Buttons/links on page: u'() => { const TARGETS = new Set(['写回答','回答问题','回答','发起回答','编辑回答']); const ZW_RE = /[​‌‍ ]/g; const walker = document.createTreeWalker( document.body, NodeFilter.SHOW_TEXT ); let node; while ((node = walker.nextNode())) { const cleaned = node.textContent.replace(ZW_RE, '').trim(); if (TARGETS.has(cleaned)) { const el = node.parentElement; const btn = el.closest('button, a, [role="button"]') || el; btn.click(); return cleaned; } } return null; }u2Clicked '写回答' via text-node fallback (text='z')rz.png)suffixFrzzhihu_debug.pngu🔍 发布调试截图：找不到「写回答」按钮，请确认页面状态（是否有弹窗/登录墙/按钮文字变了）。)rrrrT) missing_okz Debug screenshot upload failed: u~Could not find '写回答' button. Zhihu DOM may have changed — update _WRITE_ANSWER_SELECTORS in zhihu_playwright_engine.py)rXrV_WRITE_ANSWER_SELECTORSrrwait_forrr6rmrorpr7tempfilermktemprrr rr unlinkrN) rr=rbtn all_textsclickedrtmp_ss_ss_errs rrYz)ZhihuPlaywrightEngine._click_write_answersg /000 c"""*  C ll3''- 9d ;;; %%d+++ >>>???    MM    X RUSURUXXYYY--   (    ! !$ ' ' ' KKXWXXX Y Y Y F I OOO(///8899F OOVO > > >   - -+[[*a .    MMTM * * * * I I I NNGgGG H H H H H H H H I K   s+A7B// B<;B<BG H "HH textcd}tD]b} |d|j}td|n#t $rY_wxYw|td| ddtffd } d | d |t|d zkr-t d t|d dStd d| d |t|d zkr-t d t|ddStd|jdjdj|d dt d t|ddS)uInput answer text into Zhihu's Draft.js rich-text editor. Tries three strategies in order, verifying content after each attempt: 1. execCommand('insertText') — most reliable for contenteditable 2. Synthetic ClipboardEvent paste 3. keyboard.type() — slow but universally compatible Args: page: Page with the answer editor open. text: Plain-text answer to insert. Nrr|zFound editor via: zTCould not find answer editor. Update _EDITOR_SELECTORS in zhihu_playwright_engine.pyirc.dS)z1Return current character count inside the editor.az() => { const ed = document.querySelector( '.DraftEditor-content[contenteditable="true"],' + '.RichText [contenteditable="true"],' + '[contenteditable="true"].public-DraftEditor-content' ); return ed ? ed.textContent.length : 0; })rX)r=sr_content_lengthz;ZhihuPlaywrightEngine._fill_answer.._content_length8s==   ra(text) => { const ed = document.querySelector( '.DraftEditor-content[contenteditable="true"],' + '.RichText [contenteditable="true"],' + '[contenteditable="true"].public-DraftEditor-content' ); if (!ed) return false; ed.focus(); document.execCommand('selectAll'); document.execCommand('insertText', false, text); return true; }iXr zFilled z! chars via execCommand insertTextz:execCommand insertText insufficient; trying ClipboardEventab(text) => { const ed = document.querySelector( '.DraftEditor-content[contenteditable="true"],' + '.RichText [contenteditable="true"],' + '[contenteditable="true"].public-DraftEditor-content' ); if (!ed) return; ed.focus(); const dt = new DataTransfer(); dt.setData('text/plain', text); ed.dispatchEvent(new ClipboardEvent('paste', { bubbles: true, cancelable: true, clipboardData: dt })); }z chars via ClipboardEventzAClipboardEvent also insufficient; falling back to keyboard.type()z Control+aDelete)delayz chars via keyboard.type())_EDITOR_SELECTORSr~rrr6rmrorNrrVrrXrnr7rpkeyboardpresstype)rr=reditorrrs ` rrZz"ZhihuPlaywrightEngine._fill_answers$  C &&sD&999c**0 7#77888     >I    c"""              c""" ?  D Q . . KKN#d))NNN O O O F STTT      " c""" ?  D Q . . KKF#d))FFF G G G F Z[[[  K((( H%%% 4q))) c""" Cc$iiCCCDDDDDsAA A+*A+ctD]} ||j}|dd|t d||d|d}|r!t d||cSt d|j |j cS#t$rYwxYwtd ) u Click the publish button and return the answer URL. After clicking 发布, waits for the page to update then extracts the answer URL (``/question//answer/``) from the DOM. Falls back to the question URL if no answer link is found. Args: page: Page with the answer text already filled in. Returns: URL of the published answer, or question URL as fallback. Raises: RuntimeError: If the publish button cannot be found. rirzClicked publish via: r{aH() => { // Look for a canonical answer link on the page const links = Array.from( document.querySelectorAll('a[href*="/answer/"]') ); // Prefer links that point to the current question const sorted = links .map(a => a.href) .filter(h => /\/question\/\d+\/answer\/\d+/.test(h)); return sorted.length ? sorted[sorted.length - 1] : null; }zAnswer published: z!Answer submitted (URL fallback): zVCould not find publish button. Update _PUBLISH_SELECTORS in zhihu_playwright_engine.py) _PUBLISH_SELECTORSrrrrr6r7rVrXrWrorN)rr=rrr`s rr[z$ZhihuPlaywrightEngine._submit_answer}s- &  C ll3''- 9d ;;;  9C99:::%%d+++"]]    &KK AZ A ABBB%%%% JJJKKKx     F   sB,C%:(C%% C21C2messagec |j|j|dS#t$r(}td|Yd}~dSd}~wwxYw)zSend a plain text message to the notify channel. Args: message: Message text (supports Slack mrkdwn formatting). )rrzSlack notify failed: N)r chat_postMessager ror6r)rrrqs rrz#ZhihuPlaywrightEngine._notify_slacks  6   . .+ /      6 6 6 LL444 5 5 5 5 5 5 5 5 5 6s!% AAA)N)rN)__name__ __module__ __qualname____doc__rrrrrboolr?rar4r\r5r9rrrrrrYrZr[rrrrr r Vs$&* BBBBTN B  BBBB."$""""HS 3S SS SS S S S nd.SSSS$T0)t))))V"8D>@99999990t:S S S S jcEscEtcEcEcEcEJ4 c4 4 4 4 p 6S 6T 6 6 6 6 6 6rr )rrhrvpathlibrtypingr slack_sdkrhealth.utils.logging_configrrr6rrr}rrrrrUrrrrr rrrrsQ 444444 h       2+D,-- k 6k 6k 6k 6k 6k 6k 6k 6k 6k 6r