o 5@Hh@s|ddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z ddl m Z eeZGdddZdS)N)Path)ListOptional)WebDriverExceptionc@sdeZdZdZdedefddZedefddZ edee defdd Z ed eefd 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)selfrrj/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/selenium/webdriver/common/selenium_manager.py binary_paths&s      zSeleniumManager.binary_pathscCsttjd}td}|dur||}d}td}dur,t d|t|}nF| r3|}n?ddddddd }t j d vrEt nd }t j d vrSt d t j |t j |f}|durjtdt j d|ttj|}|dusz|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-managerEXENSE_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)rrr r)rr zFSelenium 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) compiled_pathexepathenv_pathallowedarchlocationrrrr 8s8      zSeleniumManager._get_binaryc Csd|}td|z8tjdkrtj|dtjd}ntj|dd}|j d d}|j d d}|d kr=t |ngid }WntyW}ztd ||d }~wwt|d |d}|jrwtd |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)r:zutf-8 )logsresultzUnsuccessful command executed: Nr>r?z; code: )joinr r*r,r- subprocessrunCREATE_NO_WINDOWstdoutdecoderstripstderrr loads Exceptionrr _process_logs returncode)rcommandcompleted_procrDrGoutputerrr?rrrrhs(    zSeleniumManager._run log_itemscCsD|D]}|ddkrt|dq|ddvrt|dqdS)NlevelWARNmessage)rINFO)r r/r*)rPitemrrrrJs  zSeleniumManager._process_logsN) __name__ __module__ __qualname____doc__rdictr staticmethodrr r rrJrrrrr s/r)r rr(r-rAr,r%pathlibrtypingrrselenium.commonr getLoggerrVr rrrrrs