cQbjddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZGd d e Zy) N)contextmanager)BytesIO) DriverFinder)LocalWebDriver)Options)FirefoxRemoteConnection)Servicec eZdZdZdZdZ ddedzdedzdeddffd Z dd Z e d Z dde fd Zdd ZdefdZdefdZdefdZde fdZxZS) WebDriverz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNoptionsservice keep_alivereturnc|r|n t|_|r|n t|_t |j|j}|j r0|j |j_d|j_|jjxs|j|j_ |jjt|jj||jj} t|A||jy#t"$r|j%wxYw)a^Create a new instance of the Firefox driver, start the service, and create new instance. Args: options: Instance of Options. service: Service object for handling the browser driver if you need to pass extra details. keep_alive: Whether to configure FirefoxRemoteConnection to use HTTP keep-alive. N)remote_server_addrr ignore_proxy)command_executorr)r rrrrget_browser_pathbinary_locationbrowser_versionenv_pathget_driver_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit)selfrrrfinderexecutor __class__s W/var/www/html/venv/lib/python3.12/site-packages/selenium/webdriver/firefox/webdriver.pyr zWebDriver.__init__&s#*wwy ")wwy dllDLL9  " " $+1+B+B+DDLL (+/DLL ( LL113Ov7M7M7O  *#||77!99    G h  M  IIK  s D00E c,|jdd|iy)zSets the context that Selenium commands are running in. Args: context: Context to set, should be one of CONTEXT_CHROME or CONTEXT_CONTENT. SET_CONTEXTcontextNexecute)r#r*s r' set_contextzWebDriver.set_contextJs ]Y$89c#K|jdjd}|j| d|j|y#|j|wxYww)aSet the context that Selenium commands are running in using a `with` statement. The state of the context on the server is saved before entering the block, and restored upon exiting it. Args: context: Context, may be one of the class properties `CONTEXT_CHROME` or `CONTEXT_CONTENT`. Example: with selenium.context(selenium.CONTEXT_CHROME): # chrome scope ... do stuff ... GET_CONTEXTvalueN)r,popr-)r#r*initial_contexts r'r*zWebDriver.contextRsS ,,}599'B ! .    _ -D  _ -s2A!A A! AA!c tjj|rt}tjj |}t |dz}t j|dt jd5}tj|D]B\}}}|D]7} tjj|| } |j| | |d9D dddtj|jjd} nGt!|d5} tj| j#jd} ddd |d} |j%d | d S#1swYxYw#1swY/xYw) aInstalls Firefox addon. Returns identifier of installed addon. This identifier can later be used to uninstall addon. Args: path: Absolute path to the addon that will be installed. temporary: Allows you to load browser extensions temporarily during a session. Returns: Identifier of installed addon. Example: driver.install_addon("/path/to/firebug.xpi") wF)strict_timestampsNzUTF-8rb)addon temporary INSTALL_ADDONr1)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr,)r#rr:fp path_rootzippedbase_filesfylefilenamer9filepayloads r' install_addonzWebDriver.install_addonisU 77== B77##D)DD A IS'*>*>RWX E\b&(ggdmEND!U %E#%77<<d#; Xx /CDEE E $$R[[]3::7CEdD! FT((5<> E E F Fs>AE6!3F6E?F c,|jdd|iy)zUninstalls Firefox addon using its identifier. Args: identifier: The addon identifier to uninstall. Example: driver.uninstall_addon("addon@foo.com") UNINSTALL_ADDONidNr+)r# identifiers r'uninstall_addonzWebDriver.uninstall_addons &z(:;r.c*|jjdstjdt|j } t |d5}|j|ddd~y#1swY xYw#t$rY~ywxYw#~wxYw)aSave a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Returns: False if there is any IOError, else returns True. Use full paths in your filename. Example: driver.get_full_page_screenshot_as_file("/Screenshots/foo.png") z.pngz^name used for saved screenshot does not match file type. It should end with a `.png` extensionwbNFT) lowerendswithwarningswarn UserWarningget_full_page_screenshot_as_pngrJrEOSError)r#rSpngfs r' get_full_page_screenshot_as_filez*WebDriver.get_full_page_screenshot_as_files~~((0 MMp 224 h%         s< A?A3)A?3A<8A?? B B B  BBc$|j|S)aSave a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Returns: False if there is any IOError, else returns True. Use full paths in your filename. Example: driver.save_full_page_screenshot("/Screenshots/foo.png") )rg)r#rSs r'save_full_page_screenshotz#WebDriver.save_full_page_screenshots44X>>r.cftj|jjdS)zGet the full document screenshot of the current window as binary data. Returns: Binary data of the screenshot. Example: driver.get_full_page_screenshot_as_png() ascii)rF b64decode"get_full_page_screenshot_as_base64encoder#s r'rcz)WebDriver.get_full_page_screenshot_as_pngs* G G I P PQX YZZr.c*|jddS)zGet the full document screenshot of the current window as a base64-encoded string. Returns: Base64 encoded string of the screenshot. Example: driver.get_full_page_screenshot_as_base64() FULL_PAGE_SCREENSHOTr1r+ros r'rmz,WebDriver.get_full_page_screenshot_as_base64s||23G<rs6$ %@>6P6v=v=r.