gddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl mZddlmZdd lmZGd d e Z dS) N)contextmanager)BytesIO) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)Servicec eZdZdZdZdZ ddededed dffd Z dfd Z dd Z e d Z dd efdZddZd efdZd efdZd efdZd efdZxZS)rz=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)N SET_CONTEXTcontextexecute)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. :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/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. :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') rwF)strict_timestampsNzUTF-8rb)addon temporary INSTALL_ADDONr4)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr/)r$rr<fp path_rootzippedbase_filesfylefilenamer;filepayloads r( install_addonzWebDriver.install_addonqs# 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. :Usage: :: 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. :Usage: :: 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_pngrLrGOSError)r$rUpngfs 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. :Usage: :: driver.save_full_page_screenshot('/Screenshots/foo.png') )ri)r$rUs 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. :Usage: :: driver.get_full_page_screenshot_as_png() ascii)rH b64decode"get_full_page_screenshot_as_base64encoder$s r(rez)WebDriver.get_full_page_screenshot_as_pngs0 G G I I P PQX Y YZZZr1c8|ddS)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_SCREENSHOTr4r.rqs r(roz,WebDriver.get_full_page_screenshot_as_base64s||233G<rbrB contextlibrior'selenium.webdriver.common.driver_finderr#selenium.webdriver.remote.webdriverrRemoteWebDriverrrremote_connectionr rr r1r(rs" %%%%%%@@@@@@LLLLLL666666~=~=~=~=~=~=~=~=~=~=r1