U ^{j A@s@dZddlmZddlZddlZddlZddlZddlmZddlm Z ddl Z ddl m Z ddl mZdd lmZmZmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZe rddlm Z ddl!m"Z"ddlm#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+e ,e-Z.GdddeZ/GdddeZ0dS)z)Provides the async api: `Browser`, `Tab`.) annotationsN)Lock) TYPE_CHECKING)protocol)Broker)BrowserClosedErrorBrowserDepsErrorBrowserFailedErrorChromium)ChannelClosedErrorPipe)SessionTarget) TmpDirWarning_manual_thread_pool)kill)Path) TracebackType)Any GeneratorMutableMapping)Self)BrowserImplInterface)ChannelInterfacec@seZdZdZddddZdS)Tabzd?Zd@d dAdBZd d dCdDZdEd dFdGZdSdIdJds zBrowser._is_opencCs@z$|jr|jWdSWdSWntk r:YdSXdSr5)r2r6release RuntimeErrorr"r$r$r% _release_lockGs   zBrowser._release_lock) browser_cls channel_clszstr | Path | Noneztype[BrowserImplInterface]ztype[ChannelInterface]r)pathr<r=kwargsrcKsZtdtjddd|_|i|_i|_||_t ||j|_ ||j|f||_ dS)a Construct a new browser instance. Args: path: The path to the browser executable. browser_cls: The type of browser (default: `Chromium`). channel_cls: The type of channel to browser (default: `Pipe`). kwargs: The arguments that the browser_cls takes. For example, headless=True/False, enable_gpu=True/False, etc. zAttempting to open new browser.Zchecking_close max_workersnameN) _loggerdebugrManualThreadExecutor_process_executorr3r.r/_channelrr _browser_impl)r#r>r<r=r?r$r$r%__init__Qs zBrowser.__init__cCs |jS)zReturn if process is isolated.)rI is_isolatedr"r$r$r%rKsszBrowser.is_isolatedc stdIdHr tdtjdr6jj}nd}tjd|d\_ }ddfd d }t d t }| j|IdH_td jtd jznt dt _t djt djt dt dIdHIdHWnfttt jfk r~}z>tjdrRjj rRt!|tddddjj"d|W5d}~XYnXdS)zOpen the browser.zOpening browser.NzCan't re-open the browser logger_parserZ browser_proc)parserr,rcsFjj}j}j}j}tj|f||d|S)N)stderrenv)rIZpre_openZget_cli _logger_pipeZget_envZget_popen_argsr-Popen)clirNrOargsr"r$r%runs    zBrowser.open..runzTrying to open browser.0zStarting watchdogzOpening channel.zRunning read loopzPopulating Targetsr missing_libsz7The browser seemed to close immediately after starting.z@You can set the `logging.Logger` level lower to see more output.zAYou may try installing a known working copy of Chrome by running z`$ choreo_get_chrome`.It may be your browser auto-updated and will now work upon restart. The browser we tried to start is located at .)#rDinfor8r:hasattrrIrLlogistroZ getPipeLoggerrPrEasyncioget_running_looprun_in_executorrGr-superrJr _add_sessionrZ create_task _watchdogr0rHopenZ run_read_loopsleeppopulate_targetsrr CancelledErrorrWr r>)r#rM_rTloope __class__r"r%rbwsV              z Browser.openrcs|IdH|S)z=Open browser as context to launch on entry and close on exit.N)rbr"r$r$r% __aenter__szBrowser.__aenter__zGenerator[Any, Any, Browser]cCs |S)z?If you await the `Browser()`, it will implicitly call `open()`.)rk __await__r"r$r$r%rlszBrowser.__await__rz int | None)waitrcst|dsdS|dkr*|jdk}| Sz&t}||j|jj|IdHWn.tjk rhYdStj k r~YdSXdS)Nr-TrF) rZr-pollr\r]r^rGrmTimeoutExpiredre)r#rmr8rgr$r$r% _is_closeds" zBrowser._is_closedc s|IdHrtddSztd|dIdHWn^ttfk r^tdYdStk rztdYn tjk rtdYnX|j |jddIdHrdS|IdHrtd dSt d t |j |jd dIdHrdStd dS) NzNo _close(), already is closedzTrying Browser.close Browser.closez.Browser is closed trying to send Browser.closez)Can't send Browser.close on close channelz3Close was cancelled, _broker must be shutting down.r@)rmz'Browser is closed after closing channelz"Resorting to unclean kill browser.z)Couldn't close or kill browser subprocess)rprDrE send_commandrr r r\rerHr&warningrr-r:r"r$r$r%_closes0       zBrowser._closecstd|jr$td|j|s0dSz&td|IdHtdWntk rjYnX|j td|j rt |j td|j td|j td |jjd d d dS) Close the browser.zClosing browser.zCancelling watchdog.NzStarting browser close methods.zBrowser close methods finished.zBroker cleaned up.zLogging pipe closed.zBrowser channel closed.z"Browser implementation cleaned up.FTrmZcancel_futures)rDrYr0rEcancelr;ruProcessLookupErrorr cleanrPosr&rHrIrGshutdownr"r$r$r%r&s,            rqztype[BaseException] | NonezBaseException | NonezTracebackType | Nonez bool | None)type_value tracebackrcs|IdHdS)rvN)r&)r#r}r~rr$r$r% __aexit__szBrowser.__aexit__c stjddd}zzttjdtd td t }t d | ||j jIdHtd d|_|IdHt dIdH| ||jjIdHW5QRXWnt jk rYnXW5|jdddtdXdS) NrZ watchdog_waitrAFTrwz$Watchdog full shutdown (in finally:)ignore)categoryz In watchdogz Running wait.z2Wait expired, Browser is being closed by watchdog.)rrFr|rDrEwarningscatch_warningsfilterwarningsrr\r]Zdebug2r^r-rmrtr0r&rcrIrzre)r#Z _executorrgr$r$r%ra"s6      zBrowser._watchdogr)tabrcCs*t|tstd|j||j|j<dS)Nztab must be an object of ) isinstancer TypeErrorZ _tab_typer.r)r#rr$r$r%_add_tabBs zBrowser._add_tabstr)rrcCst|tr|j}|j|=dSr1)rrrr.)r#rr$r$r% _remove_tabGs zBrowser._remove_tabz Tab | NonecCs |jrtt|jSdS)zz Get the first tab if there is one. Useful for default tabs. Returns: A tab object. N)r.valuesnextiterr"r$r$r%get_tabLs zBrowser.get_tabc s|IdHrtd|dIdH}d|kr@tdt|d|ddD]}|dd krL|d |jkrL|d }t||j}z|IdHWnPt j k r}z0|j t j j krtd |d WY qLnW5d}~XYnX||td |dqLdS)zHSolicit the actual browser for all targets to add to the browser object.Nz-populate_targets() called on a closed browserzTarget.getTargetserrorzCould not get targetsresultZ targetInfostypepagetargetIdzTarget z# not found (could be closed before)z The target z was added)rprrsr: Exceptionr.rr create_sessionrDevtoolsProtocolErrorcodeZEcodeZTARGET_NOT_FOUNDrDrtrrE)r#responseZ json_responsernew_tabrhr$r$r%rdXs2      zBrowser.populate_targetsrcsr|IdHrtdtdtj|dIdH}d|krLtdt||dd}t ||j }| ||S) z Create a browser session. Only in supported browsers, is experimental. Returns: A session object. Nz+create_session() called on a closed browserz_Creating new sessions on Browser() only works with some versions of Chrome, it is experimental.zTarget.attachToBrowserTargetrzCould not create sessionrZ sessionId) rprrwarnrZExperimentalFeatureWarningrsr:rrr r`)r#rZ session_idZ new_sessionr$r$r%rvs&   zBrowser.create_sessionrVF)window)urlwidthheightrrc s|IdHrtdd|i}|r*||d<|r6||d<|rBd|d<|jd|d IdH}d |krntd t||d d }t||j}||| IdH|S)aE Create a new tab. Args: url: the url to navigate to, default "" width: the width of the tab (headless only) height: the height of the tab (headless only) window: default False, if true, create new window, not tab Returns: a tab. Nz(create_tab() called on a closed browser.rrrTZ newWindowzTarget.createTarget)paramsrzCould not create tabrr) rprrsr:rrrr rr) r#rrrrrrrrr$r$r% create_tabs.   zBrowser.create_tabzprotocol.BrowserResponsecsd|IdHrtdt|tr&|j}|jdd|idIdH}||d|kr`tdt ||S)zq Close a tab by its id. Args: target_id: the targetId of the tab to close. Nz&close_tab() called on a closed browserzTarget.closeTargetr)commandrrzCould not close tab) rprrrrrsrr:rr)r#rrr$r$r%r!s$   zBrowser.close_tab)N)r)rVNN)r'r(r)r*__annotations__r0r3r8r;r r rJrKrbrkrlrprur&rrarrrrdrrr! __classcell__r$r$rir%r+.sD    "B    ,r+)1r* __future__rr\r{r-rrtypingrr[Z choreographerrZ_brokersrZbrowsersrr r r Zchannelsr r Zprotocol.devtools_asyncrrutilsrrZ utils._killrpathlibrtypesrrrrtyping_extensionsrZbrowsers._interface_typerZchannels._interface_typer getLoggerr'rDrr+r$r$r$r%s2