U Pf^@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl mZddlmZdd lmZGd d d e Z dS) N)contextmanager)BytesIO) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)ServicecseZdZdZdZdZdeeeddfdd Z dd fd d Z dd d d Z e ddZ ded ddZdd ddZed ddZed ddZed ddZed ddZZS)rz=Controls the GeckoDriver and allows you to drive the browser.ZchromecontentNT)optionsservice keep_alivereturncs|r|nt|_|r|nt}t|j|}|rB||_d|_||j_|j t |jj ||j d}zt j||dWntk r|YnXd|_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)Zremote_server_addrr Z ignore_proxy)Zcommand_executorr F)r r rrZget_browser_pathZbinary_locationZbrowser_versionZget_driver_pathpathstartrZ service_urlZ_ignore_local_proxysuper__init__ ExceptionquitZ _is_remote)selfr r r finderexecutor __class__H/tmp/pip-unpacked-wheel-ds5t_0qj/selenium/webdriver/firefox/webdriver.pyr&s&    zWebDriver.__init__)rcs:z(ztWntk r$YnXW5|jXdS)z=Closes the browser and shuts down the GeckoDriver executable.N)r stoprrrrrrrrNs  zWebDriver.quitcCs|dd|idS)NZ SET_CONTEXTcontextexecute)rrrrr set_contextXszWebDriver.set_contextc cs6|dd}||z dVW5||XdS)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 ... Z GET_CONTEXTvalueN)r popr!)rrZinitial_contextrrrr[s   zWebDriver.contextFc Cstj|rt}tj|}t|d}t|dtjH}t |D]6\}}}|D]&} tj || } | | | |dqTqFW5QRXt |d} n*t|d} t | d} W5QRX| |d} |d| dS) 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 temporaryZ INSTALL_ADDONr")osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr )rrr'fpZ path_rootZzippedbase_filesZfylefilenamer&filepayloadrrr install_addonqs   "  zWebDriver.install_addoncCs|dd|idS)zUninstalls Firefox addon using its identifier. :Usage: :: driver.uninstall_addon('addon@foo.com') ZUNINSTALL_ADDONidNr)r identifierrrruninstall_addonszWebDriver.uninstall_addonc Csp|dstdt|}zDz$t|d}||W5QRXWntk rbYW dSXW5~XdS)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` extensionwbFT) lowerendswithwarningswarn UserWarningget_full_page_screenshot_as_pngr6r1OSError)rr<Zpngfrrr get_full_page_screenshot_as_files z*WebDriver.get_full_page_screenshot_as_filecCs ||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') )rL)rr<rrrsave_full_page_screenshotsz#WebDriver.save_full_page_screenshotcCst|dS)zGets the full document screenshot of the current window as a binary data. :Usage: :: driver.get_full_page_screenshot_as_png() ascii)r2 b64decode"get_full_page_screenshot_as_base64encoderrrrrIs z)WebDriver.get_full_page_screenshot_as_pngcCs|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() ZFULL_PAGE_SCREENSHOTr"rrrrrrPs z,WebDriver.get_full_page_screenshot_as_base64)NNT)F)__name__ __module__ __qualname____doc__ZCONTEXT_CHROMEZCONTEXT_CONTENTrr boolrrr!rrstrr?rBrLrMbytesrIrP __classcell__rrrrr s,(  "  r)r2r(rFr, contextlibriorZ'selenium.webdriver.common.driver_finderrZ#selenium.webdriver.remote.webdriverrZRemoteWebDriverr rZremote_connectionrr r rrrrs