U ^{j,@s.dZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl Z e dkrlddlZddlmZddlmZmZd d lmZe rddlZdd l mZmZmZmZdd lmZe ejd Z e ejdZ!e "e#Z$dddddZ%e&dZ'Gddde(Z)GdddZ*dS)z=Provides a class proving tools for running chromium browsers.) annotationsN)Path) TYPE_CHECKINGWindows)Pipe) TmpDirectoryget_browser_path)chromium_based_browsers)AnyMappingMutableMappingSequence)ChannelInterfacez_unix_pipe_chromium_wrapper.pyZpackaged_chromium_libsz str | Pathbool)pathreturncCs$zt|tjWSYdSXdS)NF)osaccessX_OK)rr^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/choreographer/browsers/chromium.py_is_exe&srz\d*:\d*:\d*\/\d*\.\d*:c@seZdZdZdS)ChromeNotFoundErrorz,Raise when browser path can't be determined.N)__name__ __module__ __qualname____doc__rrrrr0src@seZdZUdZded<ded<ded<ded<ded<d ed <ed d ddd dddZeddddddZddddZd2ddddddd Z ddd!d"Z ddd#d$Z d%dd&d'Z d(dd)d*Z d+dd,d-Zddd.d/Zddd0d1ZdS)3Chromiumz Chromium represents an implementation of the chromium browser. It also includes chromium-like browsers (chrome, edge, and brave). zstr | Path | Nonerr gpu_enabledheadlesssandbox_enabled skip_localrtmp_dirF) skip_typicalz str | None)r"r$rcCsjtD]\\}}td|dt|j||jd}|sX|sX|jD]}t|rB|}qXqB|r|SqdS)zFind a chromium based browser.zLooking for a z browser.)Zexecutable_namesr" ms_prog_idN) r items_loggerdebugrZ exe_namesr%Z typical_pathsr)clsr"r$nameZ browser_datar candidaterrr find_browserHs  zChromium.find_browserzlogging.LogRecordzMutableMapping[str, Any])record_oldrcCstd|j|_dS)a^ Remove chromium timestamp from chromium's logs. This method will be used as the `filter()` method on the `logging.Filter()` attached to all incoming logs from the browser process. Args: record: the `logging.LogRecord` object to read/modify _old: data that was already stripped out. T)_logs_parser_regexsubmsg)r)r-r.rrr logger_parser`szChromium.logger_parser)rc Cs|jrtddStdtdkr8tddSd}z2td|jtjdt|jgdd dd }Wn^t k r}z@d }|r|j r|j nd}t |d |d |WYdSd}~XYnXd|j krd}t|d|j dStddS)zReturn true if libs ok.zIf we HAVE to skip local.TzChecking for libs needed.Linuxz)We're not in linux, so no need for check.Nz Trying ldd Zldd)capture_outputtimeoutcheckz ldd failed.z e: z , stderr: Fs not foundzFound deps missing in chrome z#No problems found with dependencies)r"r'r(platformsystemr subprocessrunstr ExceptionstderrdecodeinfostdoutZdebug2)selfper2r@rrr_libs_okvsF      zChromium._libs_okNrzPath | str | Noner None)channelrkwargsrcKstd|||_|dd|_|dd|_|dd|_|dd|_|rdtd | t d t koz|j|_|jrtd |jstj|jd |_|jstd |jdtd|j||_t|tstddt|jk|_dS)a Construct a chromium browser implementation. Args: channel: the `choreographer.Channel` we'll be using (WebSockets? Pipe?) path: path to the browser kwargs: gpu_enabled (default False): Turn on GPU? Doesn't work in all envs. headless (default True): Actually launch a browser? sandbox_enabled (default False): Enable sandbox- a persnickety thing depending on environment, OS, user, etc tmp_dir (default None): Manually set the temporary directory Raises: RuntimeError: Too many kwargs, or browser not found. NotImplementedError: Pipe is the only channel type it'll accept right now. zChromium init'ed with kwargs Z enable_gpuFr TZenable_sandboxr#Nz*Chromium.get_cli() received invalid args: Zubuntuz?Forced skipping local. Ubuntu sandbox requires package manager.)r"zzBrowser not found. You can use get_chrome() or choreo_get_chrome from bash. please see documentation. Local copy ignored: .zFound chromium path: z-Websocket style channels not implemented yet.Zsnap)r'rBrpoprr r! _tmp_dir_path RuntimeErrorkeysrr:versionlowerr"warningrr,r_channel isinstancerNotImplementedErrorr> _is_isolated)rDrIrrJrrr__init__s8   zChromium.__init__cCs6t|j|jd|_| |_td|jjdS)zPrepare browser for opening.)rZsneakzTemporary directory at: N) rrMrVr#rGZ missing_libsr'rBrrDrrrpre_opens  zChromium.pre_opencCs|jS)z Return if /tmp directory is isolated by OS. Returns: bool indicating if /tmp is isolated. )rVrXrrr is_isolatedszChromium.is_isolatedzMapping[str, Any]cCsdi}tdkr$tj|d<d|d<n,d|d<t|jtrP|jj|d<|jj|d<t d||S) zBReturn the args needed to runc chromium with `subprocess.Popen()`.r creationflagsF close_fdsTstdinrCzReturning args: ) r:r;r<ZCREATE_NEW_PROCESS_GROUPrTrSrfrom_choreo_to_externalfrom_external_to_choreor'r()rDargsrrrget_popen_argss      zChromium.get_popen_argsz Sequence[str]cCs0tdkr(ttjttt|jg}n t|jg}|jsD|d|j rT|d|j sd|d| dddd|j jd d d d d dddddddddddddddgt |jtr|dtdkrt|jj}t|jj}d}t||t|||d|d |g7}td!||S)"z$Return the CLI command for chromium.rz --disable-gpuz --headlessz --no-sandboxz--disable-breakpadz--allow-file-access-from-filesz--enable-logging=stderrz--user-data-dir=z--no-first-runz--enable-unsafe-swiftshaderz--disable-dev-shm-usagez"--disable-background-media-suspendz--disable-lazy-loadingz%--disable-background-timer-throttlingz(--disable-backgrounding-occluded-windowsz --disable-renderer-backgroundingz--disable-component-updatez--disable-hang-monitorz--disable-popup-blockingz--disable-prompt-on-repostz!--disable-ipc-flooding-protectionz--disable-syncz--metrics-recording-onlyz--password-store=basicz--use-mock-keychainz--no-default-browser-checkz--no-process-per-sitez--disable-web-securityz--remote-debugging-pipeTz--remote-debugging-io-pipes=,zReturning cli: )r:r;r>sys executable_chromium_wrapper_pathrrappendr r!extendr#rTrSrmsvcrtZ get_osfhandler^r_rZset_handle_inheritabler'r()rDcliZr_handleZw_handleZ _inheritablerrrget_clisj        zChromium.get_clizMutableMapping[str, str]cCstj}|S)z#Return the env needed for chromium.)renvironcopy)rDenvrrrget_env;s zChromium.get_envcCst|dr|jdS)z4Clean up any leftovers form browser, like tmp files.r#N)hasattrr#cleanrXrrrrp@s zChromium.cleancCs |dS)z,Delete the temporary file and run `clean()`.N)rprXrrr__del__EszChromium.__del__)N)rrrr__annotations__ classmethodr,r3rGrWrYrZrarjrnrprqrrrrr4s, +;  @r)+r __future__rrr:rer<rcpathlibrtypingrZlogistror;rhZchoreographer.channelsrZchoreographer.utilsrrZ_chrome_constantsr loggingr r r rZ&choreographer.channels._interface_typer__file__resolveparentreZ_packaged_chromium_libs getLoggerrr'rcompiler0rNrrrrrrs4