U g@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z eeZGdddZdS)N)Path)List)Optional)WebDriverExceptionc@s`eZdZdZeedddZeedddZ eee eddd Z eeed d d Z d S)SeleniumManagerzWrapper for getting information from the Selenium Manager binaries. This implementation is still in beta, and may change. )argsreturncCs\t|g|}ttjkr*|d|d|d|d|d||S)zDetermines the locations of the requested assets. :Args: - args: the commands to send to the selenium manager binary. :Returns: dictionary of assets and their path z--debugz--language-bindingpythonz--outputjson)str _get_binaryloggergetEffectiveLevelloggingDEBUGappend_run)selfrrN/tmp/pip-unpacked-wheel-3ysq0mf8/selenium/webdriver/common/selenium_manager.py binary_paths's     zSeleniumManager.binary_paths)rcCsttjd}td}|dk r,||}d}td}dk rXt d|t|}n| rf|}n~ddddddd }t j d krt nd }t j d krt d t j |t j |f}|dkrtdt j d|ttj|}|dks|std|t d||S)zDetermines the path of the correct Selenium Manager binary. :Returns: The Selenium Manager executable location :Raises: WebDriverException if the platform is unsupported zselenium-managerEXENZSE_MANAGER_PATHz2Selenium Manager set by env SE_MANAGER_PATH to: %szmacos/selenium-managerzwindows/selenium-manager.exezlinux/selenium-manager))darwinany)win32r)cygwinr)linuxx86_64)freebsdr)openbsdr)rrrr)rrzFSelenium Manager binary may not be compatible with %s; verify settingsz/Unsupported platform/architecture combination: /z2Unable to obtain working Selenium Manager binary; z$Selenium Manager binary found at: %s)r__file__parentjoinpath sysconfigget_config_var with_suffixosgetenvr debugexistssysplatformmachinewarninggetris_file)Z compiled_pathZexepathZenv_pathallowedarchlocationrrrr 9s8        zSeleniumManager._get_binaryc Csd|}td|zptjdkr6tj|dtjd}ntj|dd}|j d d}|j d d}|d krzt |ngid }Wn2tk r}ztd ||W5d }~XYnXt|d |d}|jrtd |d|jd|d||S)zExecutes the Selenium Manager Binary. :Args: - args: the components of the command being executed. :Returns: The log string containing the driver location.  zExecuting process: %srT)capture_output creationflags)r6zutf-8 )logsresultzUnsuccessful command executed: Nr:r;z; code: )joinr r)r+r, subprocessrunZCREATE_NO_WINDOWstdoutdecoderstripstderrr loads Exceptionrr _process_logs returncode)rcommandZcompleted_procr?rBoutputerrr;rrrris$    "zSeleniumManager._run) log_itemscCsD|D]:}|ddkr$t|dq|ddkrt|dqdS)NlevelWARNmessage)rINFO)r r.r))rJitemrrrrEs   zSeleniumManager._process_logsN) __name__ __module__ __qualname____doc__rdictr staticmethodrr r rrErrrrr!s/r)r rr'r,r=r+r$pathlibrtypingrrZselenium.commonr getLoggerrPr rrrrrs