a &hm@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m ZddlmZdd lmZdd lmZGd d d eZ dS) N)contextmanager)BytesIO)Optional) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)ServicecseZdZdZdZdZd!eeeee ddfdd Z dd fd d Z dd d d Z e ddZd"ed ddZdd ddZe d ddZe d ddZed ddZed ddZddZdd ZZS)#rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNT)optionsservice keep_alivereturncs|r|nt|_|r|nt}t|j|}|rB||_d|_|jpR||j_ |j t |jj ||j d}ztj||dWnty|Yn0d|_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_executorr F)r rrrget_browser_pathbinary_locationbrowser_versionenv_pathget_driver_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfr rrfinderexecutor __class__i/var/www/html/swiplay.fr/scambot/venv/lib/python3.9/site-packages/selenium/webdriver/firefox/webdriver.pyr's&    zWebDriver.__init__)rcsBz0ztWnty"Yn0W|jn |j0dS)z=Closes the browser and shuts down the GeckoDriver executable.N)rr rrstopr"r%r'r(r Os  zWebDriver.quitcCs|dd|idS)NZ SET_CONTEXTcontextexecute)r"r+r'r'r( set_contextYszWebDriver.set_contextc cs@|dd}||zdVW||n ||0dS)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.)r"r+Zinitial_contextr'r'r(r+\s  zWebDriver.contextFc Cstj|rt}tj|}t|d}tj|dtjddR}t |D]6\}}}|D]&} tj || } | | | |dqXqJWdn1s0Yt |d} n>t|d$} t | d} Wdn1s0Y| |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 temporaryZ INSTALL_ADDONr/)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr-)r"rr5fp path_rootzippedbase_filesfylefilenamer4filepayloadr'r'r( install_addonrs   6 2 zWebDriver.install_addoncCs|dd|idS)zUninstalls Firefox addon using its identifier. :Usage: :: driver.uninstall_addon("addon@foo.com") ZUNINSTALL_ADDONidNr,)r"Z identifierr'r'r(uninstall_addonszWebDriver.uninstall_addoncCs|dstdt|}zXz8t|d}||Wdn1sP0YWntytYW~dS0W~n~0dS)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_pngrDr?OSError)r"rMpngfr'r'r( 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") )r])r"rMr'r'r(save_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)r@ b64decode"get_full_page_screenshot_as_base64encoder*r'r'r(rYs 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/r,r*r'r'r(ras z,WebDriver.get_full_page_screenshot_as_base64cOstdSNNotImplementedErrorr"argskwargsr'r'r( download_fileszWebDriver.download_filecOstdSrcrdrfr'r'r(get_downloadable_filessz WebDriver.get_downloadable_files)NNT)F)__name__ __module__ __qualname____doc__ZCONTEXT_CHROMEZCONTEXT_CONTENTrrr boolrr r.rr+strrPrRr]r^bytesrYrarirj __classcell__r'r'r%r(r!s0(  "   r)r@r6rVr: contextlibriortypingr'selenium.webdriver.common.driver_finderr#selenium.webdriver.remote.webdriverrRemoteWebDriverr rremote_connectionr rr r'r'r'r(s