o _c@sdZddlZddlmZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZddlmZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZmZmZdd l m!Z!e"e#Z$GdddeZ%Gddde&Z'Gddde&Z(dZ)dS)zFrame Manager module.N) OrderedDict)SimpleNamespace)Any AwaitableDict GeneratorListOptionalSetUnion) EventEmitter)helper) CDPSession) ElementHandle) NetworkError)ExecutionContextJSHandle)ElementHandleError PageError TimeoutError) merge_dictcsleZdZdZeddddddZded ed ed d ffd d Z ded d fddZ de d d fddZ d ed d fddZ ed edfddZd edfddZde d edfddZde de d d fdd Zd!ed d fd"d#Zde d$e d d fd%d&Zde d d fd'd(Zd)ed d fd*d+Zd,e d d fd-d.Zd;d/d0Zd1e d efd2d3Z d FrameManagerzFrameManager class.Z frameattachedZframenavigatedZ framedetachedZlifecycleeventZframenavigatedwithindocument) FrameAttachedFrameNavigated FrameDetachedLifecycleEventFrameNavigatedWithinDocumentclient frameTreepagereturnNcst|_|_t_d_t_| dfdd| dfdd| dfdd| d fd d| d fd d| d fdd| dfdd| dfdd| dfdd |dS)zMake new frame manager.NzPage.frameAttachedcs|dd|ddS)NframeId parentFrameId)_onFrameAttachedgeteventselfB/usr/local/lib/python3.10/dist-packages/pyppeteer/frame_manager.py.sz'FrameManager.__init__..zPage.frameNavigatedc|dS)Nframe)_onFrameNavigatedr%r&r(r*r+r,2zPage.navigatedWithinDocumentcs|d|dS)Nr!url)_onFrameNavigatedWithinDocumentr%r&r(r*r+r,4szPage.frameDetachedcr-Nr!)_onFrameDetachedr%r&r(r*r+r,8r0zPage.frameStoppedLoadingcr-r3)_onFrameStoppedLoadingr%r&r(r*r+r,:zRuntime.executionContextCreatedcr-)Ncontext)_onExecutionContextCreatedr%r&r(r*r+r,>r6z!Runtime.executionContextDestroyedcr-)NexecutionContextId)_onExecutionContextDestroyedr%r&r(r*r+r,Ar6z Runtime.executionContextsClearedcsSN)_onExecutionContextsClearedr&r(r*r+r,DszPage.lifecycleEventcs |Sr;)_onLifecycleEventr&r(r*r+r,Fs ) super__init___client_pager_frames _mainFramedict_contextIdToContexton_handleFrameTree)r)rrr __class__r(r+r?$sD          zFrameManager.__init__r'cCs@|j|d}|s dS||d|d|tjj|dS)Nr!loaderIdname)rBr%r=emitrEventsr)r)r'r.r*r*r+r=Js zFrameManager._onLifecycleEventr!cCs0|j|}|s dS||tjj|dSr;)rBr%_onLoadingStoppedrLrrMrr)r!r.r*r*r+r5Qs z#FrameManager._onFrameStoppedLoadingcCsV|d}d|vr||d|d||d|vrdS|dD]}||q!dS)Nr.parentIdid childFrames)r$r/rG)r)rr.childr*r*r+rGXs   zFrameManager._handleFrameTreeFramecC|jS)zReturn main frame.)rCr(r*r*r+ mainFrameezFrameManager.mainFramecCst|jS)zReturn all frames.)listrBvaluesr(r*r*r+framesjszFrameManager.framescCs |j|S)z%Return :class:`Frame` of ``frameId``.)rBr%)r)r!r*r*r+r.ns zFrameManager.framer#cCsF||jvrdS|j|}t|j||}||j|<|tjj|dSr;)rBr%rTr@rLrrMr)r)r!r# parentFramer.r*r*r+r$rs   zFrameManager._onFrameAttached framePayloadcCs|d }|r |j}n |j|dd}|s|std|r+|jD]}||q#|dd}|rP|rA|j|jd||_nt|j d|}||j|<||_| || t j j|dS)NrPrQr"zGWe either navigate top level or have old version of the navigated frame)r%rCrBrrR_removeFramesRecursivelypop_idrTr@ _navigatedrLrrMr)r)r\Z isMainFramer.rSr_r*r*r+r/zs&      zFrameManager._onFrameNavigatedr1cCsB|j|}|s dS|||tjj||tjj|dSr;)rBr%_navigatedWithinDocumentrLrrMrr)r)r!r1r.r*r*r+r2s  z,FrameManager._onFrameNavigatedWithinDocumentcCs"|j|}|r||dSdSr;)rBr%r]rOr*r*r+r4s zFrameManager._onFrameDetachedcontextPayloadcsdrddrdd}nd}j|}dtdtffdd }tj||}|jd<|r?||dSdS)NZauxDatar!objr csd}||S)NrQ)executionContextByIdcreateJSHandle)rcr7rbr)r*r+_createJSHandles z@FrameManager._onExecutionContextCreated.._createJSHandlerQ)r%rBrrrr@rE_addExecutionContext)r)rbr!r.rgr7r*rfr+r8s"   z'FrameManager._onExecutionContextCreatedr9cCs8|j|}|s dS|j|=|j}|r||dSdSr;)rEr%r._removeExecutionContext)r)r9r7r.r*r*r+r:s z)FrameManager._onExecutionContextDestroyedcCs2|jD] }|j}|r||q|jdSr;)rErYr.riclear)r)r7r.r*r*r+r<s  z(FrameManager._onExecutionContextsCleared contextIdcCs"|j|}|std||S)z*Get stored ``ExecutionContext`` by ``id``.z*INTERNAL ERROR: missing context with id = )rEr%r)r)rkr7r*r*r+rds z!FrameManager.executionContextByIdr7 remoteObjectcCs>|durt}|ddkrt||j||j|St||j|S)z@Create JS handle associated to the context id and remote object.Nsubtypenode)rDr%rr@rAr)r)r7rlr*r*r+res zFrameManager.createJSHandler.cCsB|jD]}||q||j|jd|tjj |dSr;) rRr]_detachrBr^r_rLrrMr)r)r.rSr*r*r+r]s  z%FrameManager._removeFramesRecursivelyr Nr;)r.rTr N)"__name__ __module__ __qualname____doc__rrMrrrr?r=strr5rGpropertyr rVrrZr.r$rDr/r2r4r8r:r<rrdrrer] __classcell__r*r*rHr+rs@&      rc @seZdZdZdededdeddfddZd eddfd d Z d eddfd d Z d eeddfddZ deefddZ dede defddZdddede dede fddZdedeefddZdefddZd edeefd!d"Zdedede de fd#d$Zdedede deefd%d&Zdedeefd'd(ZeZeZeZeZeZdefd)d*Z d+eddfd,d-Z!e"defd.d/Z#e"defd0d1Z$e"dedfd2d3Z%e"dedfd4d5Z&defd6d7Z'd8edefd9d:Z(d;edefdd?Z*dnded;e+d@e ddfdAdBZ,deddfdCdDZ-deddfdEdFZ.dedGedeefdHdIZ/deddfdJdKZ0dndedLed;e+d@e ddf dMdNZ1 dndOe2ee3e4fd;e+de d@e de2e5dPff dQdRZ6dnded;e+d@e ddPfdSdTZ7dndUed;e+d@e ddPfdVdWZ8dnded;e+de d@e ddPf dXdYZ9 dndZed[ed;e+d@e ddPf d\d]Z:defd^d_Z;d`e+ddfdadbZdodjdkZ?dodldmZ@dS)prTz_Frame class. Frame objects can be obtained via :attr:`pyppeteer.page.Page.mainFrame`. rr[r!r NcCsr||_||_d|_d|_||_d|_dd|_|dt|_ d|_ t|_ t|_ |jr7|jj |dSdS)Nr"FcSdSr;r*_r*r*r+r,z Frame.__init__..)r@ _parentFrame_url _detachedr__documentPromise_contextResolveCallback_setDefaultContextset _waitTasks _loaderId_lifecycleEvents _childFramesadd)r)rr[r!r*r*r+r?s  zFrame.__init__r7cCs|jr ||dSdSr;Z _isDefaultrr)r7r*r*r+rhzFrame._addExecutionContextcCs|jr |ddSdSr;rrr*r*r+rirzFrame._removeExecutionContextcsd|dur|dd_jD] }jj|qdSd_jj_fdd_dS)NcSrxr;r*ryr*r*r+r,r{z*Frame._setDefaultContext..cs j|Sr;)_contextPromise set_result)_contextr(r*r+r,s ) rrr@_loop create_taskrerunr create_futurer)r)r7waitTaskr*r(r+r s    zFrame._setDefaultContextcs|jIdHS)zReturn execution context of this frame. Return :class:`~pyppeteer.execution_context.ExecutionContext` associated to this frame. N)rr(r*r*r+executionContexts zFrame.executionContext pageFunctionargscs8|IdH}|durtd|j|g|RIdHS)ziExecute function on this frame. Details see :meth:`pyppeteer.page.Page.evaluateHandle`. Nzthis frame has no context.)rrevaluateHandle)r)rrr7r*r*r+r!s zFrame.evaluateHandleF) force_exprrcs>|IdH}|durtd|j|g|Rd|iIdHS)zhEvaluate pageFunction on this frame. Details see :meth:`pyppeteer.page.Page.evaluate`. NExecutionContext is None.r)rrevaluate)r)rrrr7r*r*r+r+s zFrame.evaluateselectorc$|IdH}||IdH}|S)zuGet element which matches `selector` string. Details see :meth:`pyppeteer.page.Page.querySelector`. N) _document querySelectorr)rdocumentvaluer*r*r+r7zFrame.querySelectorcsZ|jr|jS|IdH}|durtd|dIdH}||_|dur+td|S)NzNo context exists.rzCould not find `document`.)rrrr asElement)r)r7rr*r*r+r@szFrame._document expressioncr)zEvaluate the XPath expression. If there are no such elements in this frame, return an empty list. :arg str expression: XPath string to be evaluated. N)rxpath)r)rrrr*r*r+rLsz Frame.xpathcs*|IdH}|j||g|RIdHS)zExecute function on element which matches selector. Details see :meth:`pyppeteer.page.Page.querySelectorEval`. N)rquerySelectorEval)r)rrrrr*r*r+rWszFrame.querySelectorEvalcs.|IdH}|j||g|RIdH}|S)zExecute function on all elements which matches selector. Details see :meth:`pyppeteer.page.Page.querySelectorAllEval`. N)rJJeval)r)rrrrrr*r*r+querySelectorAllEval`szFrame.querySelectorAllEvalcr)zvGet all elements which matches `selector`. Details see :meth:`pyppeteer.page.Page.querySelectorAll`. N)rquerySelectorAllrr*r*r+rjrzFrame.querySelectorAllcs|dIdHS)z(Get the whole HTML contents of the page.z () => { let retVal = ''; if (document.doctype) retVal = new XMLSerializer().serializeToString(document.doctype); if (document.documentElement) retVal += document.documentElement.outerHTML; return retVal; } N)rstripr(r*r*r+content~s z Frame.contenthtmlcsd}|||IdHdS)zSet content to this page.zS function(html) { document.open(); document.write(html); document.close(); } Nr)r)rfuncr*r*r+ setContentszFrame.setContentcCs|jddS)zGet frame name._namer")__dict__r%r(r*r*r+rKsz Frame.namecCrU)zGet url of the frame.r}r(r*r*r+r1rWz Frame.urlcCrU)zcGet parent frame. If this frame is main frame or detached frame, return ``None``. )r|r(r*r*r+r[szFrame.parentFramecCs t|jS)zGet child frames.)rXrr(r*r*r+rRs zFrame.childFramescCrU)zXReturn ``True`` if this frame is detached. Otherwise return ``False``. )r~r(r*r*r+ isDetachedszFrame.isDetachedfilePathcsbtdt| }|}Wdn1swY|d|dd7}||IdHS)z&[Deprecated] Inject file to the frame.zE`injectFile` method is deprecated. Use `addScriptTag` method instead.Nz/* # sourceURL= {} */ r")loggerwarningopenreadformatreplacer)r)rfcontentsr*r*r+ injectFiles   zFrame.injectFileoptionsc s~|IdH}|durtdd}d}t|dtrP|d}||g}d|vr/||dz |j|IdHWStyO}z td|d|d}~wwt|d trt |d  }| } Wdn1smwY| d |d  d d } || g}d|vr||d|j|IdHSt|d tr||d g}d|vr||d|j|IdHSt d)zeAdd script tag to this frame. Details see :meth:`pyppeteer.page.Page.addScriptTag`. Nra async function addScriptUrl(url, type) { const script = document.createElement('script'); script.src = url; if (type) script.type = type; const promise = new Promise((res, rej) => { script.onload = res; script.onerror = rej; }); document.head.appendChild(script); await promise; return script; }a function addScriptContent(content, type = 'text/javascript') { const script = document.createElement('script'); script.type = type; script.text = content; let error = null; script.onerror = e => error = e; document.head.appendChild(script); if (error) throw error; return script; }r1typezLoading script from  failedpathz//# sourceURL={}rr"r { link.onload = res; link.onerror = rej; }); document.head.appendChild(link); await promise; return link; }a async function (content) { const style = document.createElement('style'); style.type = 'text/css'; style.appendChild(document.createTextNode(content)); const promise = new Promise((res, rej) => { style.onload = res; style.onerror = rej; }); document.head.appendChild(style); await promise; return style; }r1zLoading style from rrz/*# sourceURL={}*/rr"rr) rrrr%rurrrrrrrr) r)rr7Z addStyleUrlZaddStyleContentr1rrrr*r*r+ addStyleTagsN zFrame.addStyleTagkwargscsNt||}||IdH}|std|||IdH|IdHdS)zjClick element which matches ``selector``. Details see :meth:`pyppeteer.page.Page.click`. NNo node found for selector: )rJrclickdispose)r)rrrhandler*r*r+r>s  z Frame.clickcsF||IdH}|std||d|IdH|IdHdS)zjFocus element which matches ``selector``. Details see :meth:`pyppeteer.page.Page.focus`. Nrzelement => element.focus())rrrrr)rrr*r*r+focusKs  z Frame.focuscB||IdH}|std||IdH|IdHdS)ztMouse hover the element which matches ``selector``. Details see :meth:`pyppeteer.page.Page.hover`. Nr)rrhoverrrr*r*r+rV  z Frame.hoverrYcsB|D]}t|tstd|dt|q||d|IdHS)zlSelect options and return selected values. Details see :meth:`pyppeteer.page.Page.select`. zValues must be string. Found z of type al (element, values) => { if (element.nodeName.toLowerCase() !== 'select') throw new Error('Element is not a