$)fddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZej(eZGd d e Z y) N)contextmanager)BytesIO) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)Servicec eZdZdZdZdZ ddedededdffd Z dfd Z dd Z e d Z ddefd ZddZdefdZdefdZdefdZdefdZxZS)rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNoptionsservice keep_alivereturnc|r|n t|_|r|n t}tj|j||j_|jj t|jj||j} t|-||d|_y#t$r|jwxYw)aCreates a new instance of the Firefox driver. Starts the service and then creates new instance of Firefox driver. :Args: - options - Instance of ``options.Options``. - service - (Optional) service instance for managing the starting and stopping of the driver. - keep_alive - Whether to configure remote_connection.RemoteConnection to use HTTP keep-alive. )remote_server_addrr ignore_proxy)command_executorrFN)r rrrget_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfrrrexecutor __class__s a/var/www/html/flask-app/venv/lib/python3.12/site-packages/selenium/webdriver/firefox/webdriver.pyrzWebDriver.__init__)s#*wwy $'')(11$,,H  *#||77! 44    G h  H     IIK  s B22C c t||jj y#t$rY&wxYw#|jj wxYw)z=Closes the browser and shuts down the GeckoDriver executable.N)rrrrstop)r r"s r#rzWebDriver.quitLsJ GLN LL        LL   s, 8;8;Ac,|jdd|iy)N SET_CONTEXTcontextexecute)r r(s r# set_contextzWebDriver.set_contextVs ]Y$89c#K|jdjd}|j| d|j|y#|j|wxYww)aSets 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. :param context: Context, may be one of the class properties `CONTEXT_CHROME` or `CONTEXT_CONTENT`. Usage example:: with selenium.context(selenium.CONTEXT_CHROME): # chrome scope ... do stuff ... GET_CONTEXTvalueN)r*popr+)r r(initial_contexts r#r(zWebDriver.contextYsS,,}599'B ! .    _ -D  _ -s2A!A A! AA!c tjj|rt}tjj |}t |dz}t j|dt j5}tj|D]B\}}}|D]7} tjj|| } |j| | |d9D dddtj|jjd} nGt!|d5} tj| j#jd} ddd |d} |j%d| dS#1swYxYw#1swY/xYw) aInstalls Firefox addon. Returns identifier of installed addon. This identifier can later be used to uninstall addon. :param temporary: allows you to load browser extensions temporarily during a session :param path: Absolute path to the addon that will be installed. :Usage: :: driver.install_addon('/path/to/firebug.xpi') rwNzUTF-8rb)addon temporary INSTALL_ADDONr/)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr*)r rr6fp path_rootzippedbase_filesfylefilenamer5filepayloads r# install_addonzWebDriver.install_addonosQ 77== B77##D)DD A IS'*>*>? E6&(ggdmEND!U %E#%77<<d#; Xx /CDEE E $$R[[]3::7CEdD! FT((5<> E E F Fs>r,cftj|jjdS)zGets the full document screenshot of the current window as a binary data. :Usage: :: driver.get_full_page_screenshot_as_png() ascii)rB b64decode"get_full_page_screenshot_as_base64encoder s r#r_z)WebDriver.get_full_page_screenshot_as_pngs* G G I P PQX YZZr,c*|jddS)zGets the full document screenshot of the current window as a base64 encoded string which is useful in embedded images in HTML. :Usage: :: driver.get_full_page_screenshot_as_base64() FULL_PAGE_SCREENSHOTr/r)rks r#riz,WebDriver.get_full_page_screenshot_as_base64s||23G<rsK" %@L6   8 $y=y=r,