i"ddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZdd lmZGd d eZdS) N)contextmanager)BytesIO)Optional) DriverFinder)Options)FirefoxRemoteConnection)Service) WebDriverc eZdZdZdZdZ ddeedeede d dffd Z dfd Z dd Z e d Zdd efdZddZd e fdZd e fdZd efdZd efdZdZdZdZxZS)r z=Controls the GeckoDriver and allows you to drive the browser.chromecontentNToptionsservice keep_alivereturncl|r|n t|_|r|n t}t|j|}|r ||_d|_|jp||j_ |j t|jj ||j } t||n##t $r|wxYwd|_dS)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. N)remote_server_addrr ignore_proxy)command_executorrF)r rrrget_browser_pathbinary_locationbrowser_versionenv_pathget_driver_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfrrrfinderexecutor __class__s p/var/www/html/nodeJS/PythonScripts/venv3.11/lib/python3.11/site-packages/selenium/webdriver/firefox/webdriver.pyr zWebDriver.__init__&s+#*8wwwyy $3'''))dlG44  " " $ $ +&,&=&=&?&?G #&*G # L1133Ov7M7M7O7O  *#|7! 4      GG  h  H H H H    IIKKK   s &#D D*c tn#t$rYnwxYw|jdS#|jwxYw)z=Closes the browser and shuts down the GeckoDriver executable.N)rr"r!rstop)r$r's r(r"zWebDriver.quitNsq GGLLNNNN    D  L       DL      s! $A 1A1AA*c6|dd|idS)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_contextXs# ]Y$899999c#K|dd}|| dV||dS#||wxYw)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. 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/popr0)r$r-initial_contexts r(r-zWebDriver.context`s},,}5599'BB !!! . EEE   _ - - - - -D  _ - - - -s AA3Fc jtj|rt}tj|}t |dz}t j|dt jd5}tj |D]I\}}}|D]@} tj || } | | | |dAJ dddn #1swxYwYtj |d} nat!|d5} tj | d} dddn #1swxYwY| |d} |d | d S) 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_ADDONr4)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr/)r$rr=fp path_rootzippedbase_filesfylefilenamer<filepayloads r( install_addonzWebDriver.install_addonvs#" 7==   FB7##D))DD A IS'*>RWXXX E\b&(gdmmEEND!U %EE#%7<<d#;#; Xx /CDDDDEE E E E E E E E E E E E E E E E $R[[]]33::7CCEEdD!! FT(55<>s%AC..C25C2:F  F F c6|dd|idS)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(:;;;;;r1cb|dstjdt|} t |d5}||dddn #1swxYwYn#t$rY~dSwxYw~n#~wxYwdS)aSaves a full document screenshot of the current window to a PNG image file. Returns False if there is any IOError, else returns True. Use full paths in your filename. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. 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_pngrMrHOSError)r$rVpngfs r( get_full_page_screenshot_as_filez*WebDriver.get_full_page_screenshot_as_files~~((00  Mp   2244 h%%                     GGGGtsNB'B = B B  BB BB) B$B)#B$$B))B,c,||S)aSaves a full document screenshot of the current window to a PNG image file. Returns False if there is any IOError, else returns True. Use full paths in your filename. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Example: driver.save_full_page_screenshot("/Screenshots/foo.png") )rj)r$rVs r(save_full_page_screenshotz#WebDriver.save_full_page_screenshots44X>>>r1cttj|dS)zGets the full document screenshot of the current window as a binary data. Returns: Binary data of the screenshot. Example: driver.get_full_page_screenshot_as_png() ascii)rI b64decode"get_full_page_screenshot_as_base64encoder$s r(rfz)WebDriver.get_full_page_screenshot_as_pngs0 G G I I P PQX Y YZZZr1c8|ddS)a Gets the full document screenshot of the current window as a base64 encoded string which is useful in embedded images in HTML. Returns: Base64 encoded string of the screenshot. Example: driver.get_full_page_screenshot_as_base64() FULL_PAGE_SCREENSHOTr4r.rrs r(rpz,WebDriver.get_full_page_screenshot_as_base64s||233G<rs" %%%%%%@@@@@@666666PPPPPP666666LLLLLLP"P"P"P"P"P"P"P"P"P"r1