U
|
\ ±d # ã @ s^ d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dl m
|
Z
|
G dd dZ dS ) é N)Úsystem)ÚSTDOUT)ÚPopen)ÚWebDriverException)Úutilsc @ sv e Zd ZdZd ddZdd Zd dd Zd
|
d Zd d Zd ddZ dd Z
|
dd Z dd Z dd Z dd Zdd ZdS ) Ú FirefoxBinaryzx_ignore_nofocus.soNc C sz || _ |pttjd| _d| _t ¡ | _| j s:| ¡ | _ | j
|
¡ sLt dtj ¡ | _d| jd< d| jd< d| jd< dS )a¾ Creates a new instance of Firefox binary.
|
|
:Args:
|
- firefox_path - Path to the Firefox executable. By default, it will be detected from the standard locations.
|
- log_file - A file object to redirect the firefox process output to. It can be sys.stdout.
|
Please note that with parallel run the output won't be synchronous.
|
By default, it will be redirected to /dev/null.
|
ÚwbNzûFailed to find firefox binary. You can set it by specifying the path to 'firefox_binary':
|
|
from selenium.webdriver.firefox.firefox_binary import FirefoxBinary
|
|
binary = FirefoxBinary('/path/to/binary')
|
driver = webdriver.Firefox(firefox_binary=binary)Ú1ZMOZ_CRASHREPORTER_DISABLEZ MOZ_NO_REMOTEZ NO_EM_RESTART)Ú
|
_start_cmdÚopenÚosÚdevnullÚ _log_fileÚ command_liner ÚlowerÚplatformÚ_get_firefox_start_cmdÚstripr ÚenvironÚcopyÚ _firefox_env)ÚselfZ firefox_pathZlog_file© r ú`d:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\selenium/webdriver/firefox/firefox_binary.pyÚ__init__ s
|
|
ÿ
|
|
|
zFirefoxBinary.__init__c G s
|
|| _ d S )N)r )r Úargsr r r Úadd_command_line_optionsA s z&FirefoxBinary.add_command_line_optionsé c C s$ || _ | | j j¡ | j|d dS )zlLaunches the browser for the given profile name.
|
|
It is assumed the profile already exists.
|
)ÚtimeoutN)ÚprofileÚ_start_from_profile_pathÚpathÚ_wait_until_connectable)r r r r r r Úlaunch_browserD s z FirefoxBinary.launch_browserc C s | j r| j ¡ | j ¡ dS )zMKill the browser.
|
|
This is useful when the browser is stuck.
|
N)ÚprocessÚkillÚwait)r r r r r% N s
|
zFirefoxBinary.killc C s\ || j d< | jdkr | ¡ | jdg}| jrB| jD ]}| |¡ q2t|| jt| j d| _ d S )NZXRE_PROFILE_PATHÚlinuxz -foreground)ÚstdoutÚstderrÚenv)
|
r r Ú_modify_link_library_pathr
|
r Úappendr r r r$ )r r! ÚcommandÚclir r r r W s
|
|
|
|
z&FirefoxBinary._start_from_profile_pathc C s` d}t | jj¡s\| j ¡ r$td||krH| ¡ td| jj d|d7 }t
|
d¡ qdS )z9Blocks until the extension is connectable in the firefox.r zThe browser appears to have exited before we could connect. If you specified a log_file in the FirefoxBinary constructor, check it for details.zzCan't load the profile. Possible firefox version mismatch. You must use GeckoDriver instead for Firefox 48+. Profile Dir: zT If you specified a log_file in the FirefoxBinary constructor, check it for details.é T) r Zis_connectabler Úportr$ Úpollr r% r! ÚtimeÚsleep)r r Úcountr r r r" b s
|
ÿÿ z%FirefoxBinary._wait_until_connectablec
|
C s z4ddl m} ddl m} ddl m} ddl m} W n, tk
|
r` ddlm}m}m}m} Y nX dd l}d}d}|D ]n}z |||} || d}W qêW qv tk
|
râ z |||} || d}W Y qêW n tk
|
rÜ Y nX Y qvX qvdS |sòdS | |¡d S ) Nr )ÚHKEY_CURRENT_USER)ÚHKEY_LOCAL_MACHINE)ÚOpenKey)Ú
|
QueryValue)r7 r8 r6 r5 )z/SOFTWARE\Classes\FirefoxHTML\shell\open\commandz<SOFTWARE\Classes\Applications\firefox.exe\shell\open\commandÚ )
|
Ú_winregr5 r6 r7 r8 Ú ImportErrorÚwinregÚshlexÚOSErrorÚsplit)
|
r r5 r6 r7 r8 r= Úkeysr- r! Úkeyr r r Ú_find_exe_in_registryy s4
|
|
|
|
|
z#FirefoxBinary._find_exe_in_registryc C s¤ d}| j dkrBd}| |¡}|s$d}tj |¡s tj d¡| }n^| j dkr^| ¡ pZ| ¡ }nB| j dkr|tjdkr|| ¡ }n$d D ]}| |¡}|r q qt d
|
|S ) z$Return the command to start firefox.r9 ÚdarwinÚfirefoxz4/Applications/Firefox.app/Contents/MacOS/firefox-binú~ZwindowsÚjavaÚnt)rD Z iceweaselziCould not find firefox in your system PATH. Please specify the firefox binary location or install firefox)
|
r Úwhichr r! ÚexistsÚ
|
expanduserrB Ú_default_windows_locationÚnameÚ RuntimeError)r Z start_cmdZffnamer r r r s(
|
|
|
|
|
ÿz$FirefoxBinary._get_firefox_start_cmdc C sJ t dd¡t dd¡g}|D ](}t j |d¡}t |t j¡r | S q dS )NZ PROGRAMFILESzC:\Program FileszPROGRAMFILES(X86)zC:\Program Files (x86)zMozilla Firefox\firefox.exer9 )r Úgetenvr! ÚjoinÚaccessÚX_OK)r Z program_filesr! Z binary_pathr r r rK ¹ s
|
|
þ
|
z'FirefoxBinary._default_windows_locationc C s@ t j dd¡}| | jdd¡}||7 }|| jd< | j| jd< d S )NZLD_LIBRARY_PATHr9 Úx86Úamd64Z
|
LD_PRELOAD)r r ÚgetÚ_extract_and_checkr r ÚNO_FOCUS_LIBRARY_NAME)r Zexisting_ld_lib_pathZnew_ld_lib_pathr r r r+ Ä s
|
|
z'FirefoxBinary._modify_link_library_pathc C sv ||g}d}|D ]`}t j |j|¡}t j |¡s:t |¡ dd l}| t j t j t¡|| j ¡|¡ ||d 7 }q|S )Nr9 r ú:)
|
r r! rO rI ÚmakedirsÚshutilr ÚdirnameÚ__file__rV ) r r rR rS ÚpathsZ
|
built_pathr! Z library_pathrY r r r rU Î s
|
"z FirefoxBinary._extract_and_checkc C sL t jd t j¡D ]4}t j ||¡}t |t j¡rt j |¡s| S qdS )zMReturns the fully qualified path by searching Path of the given
|
name.ÚPATHN) r r r? Úpathsepr! rO rP rQ Úisdir)r ÚfnameÚpeZ checknamer r r rH Ü s
|
|
zFirefoxBinary.which)NN)r )r )Ú__name__Ú
|
__module__Ú __qualname__rV r r r# r% r r" rB r rK r+ rU rH r r r r r s
|
!
|
|
|
#
|
r ) r r2 r r Ú
|
subprocessr r Zselenium.common.exceptionsr Zselenium.webdriver.commonr r r r r r Ú<module> s
|
