o ѷ6i"@sddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m Z ddlmZddlmZGd d d eZdS) N)contextmanager)BytesIO) DriverFinder)Options)FirefoxRemoteConnection)Service) WebDriverc seZdZdZdZdZ   d%dedBdedBded dffd d Z d&fd d Z d&ddZ e ddZ d'd efddZd&ddZd efddZd efddZd efddZd efddZdd Zd!d"Zd#d$ZZS)(rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNToptionsservice keep_alivereturncs|r|nt|_|r |nt}t|j|}|r!||_d|_|jp)||j_ |j t |jj ||j d}z tj||dWn tyQ|wd|_dS)auCreate a new instance of the Firefox driver, start the service, and create new instance. 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)rr rrget_browser_pathbinary_locationbrowser_versionenv_pathget_driver_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfr r r finderexecutor __class__i/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/selenium/webdriver/firefox/webdriver.pyr%s(      zWebDriver.__init__csPz!ztWn tyYn wW|jdSW|jdS|jw)z=Closes the browser and shuts down the GeckoDriver executable.N)rrrr stopr r#r%r&rKs zWebDriver.quitcC|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+r%r%r& set_contextUszWebDriver.set_contextc cs@|dd}||z dVW||dS||w)aSet 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-popr.)r r+initial_contextr%r%r&r+]s  zWebDriver.contextFc Cstj|r[t}tj|}t|d}tj|dtjdd)}t |D]\}}}|D]} tj || } | | | |dq,q%Wdn1sKwYt |d} nt|d} t | d} Wdn1suwY| |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_ADDONr0)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr-)r rr8fp path_rootzippedbase_filesfylefilenamer7filepayloadr%r%r& install_addonts$     zWebDriver.install_addoncCr))zUninstalls Firefox addon using its identifier. Args: identifier: The addon identifier to uninstall. Example: driver.uninstall_addon("addon@foo.com") UNINSTALL_ADDONidNr,)r identifierr%r%r&uninstall_addon zWebDriver.uninstall_addoncCs|ds tdt|}z1z$t|d }||Wdn 1s(wYWW~dSWW~dStyBYW~dSw~w)aSave a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Returns: False if there is any IOError, else returns True. Use full paths in your filename. 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_pngrHrCOSError)r rQpngfr%r%r& get_full_page_screenshot_as_files(     z*WebDriver.get_full_page_screenshot_as_filecCs ||S)aSave a full document screenshot of the current window to a PNG image file. Args: filename: The full path you wish to save your screenshot to. This should end with a `.png` extension. Returns: False if there is any IOError, else returns True. Use full paths in your filename. Example: driver.save_full_page_screenshot("/Screenshots/foo.png") )rd)r rQr%r%r&save_full_page_screenshots z#WebDriver.save_full_page_screenshotcCst|dS)zGet the full document screenshot of the current window as binary data. Returns: Binary data of the screenshot. Example: driver.get_full_page_screenshot_as_png() ascii)rD b64decode"get_full_page_screenshot_as_base64encoder(r%r%r&r`rYz)WebDriver.get_full_page_screenshot_as_pngcCs|ddS)zGet the full document screenshot of the current window as a base64-encoded string. Returns: Base64 encoded string of the screenshot. Example: driver.get_full_page_screenshot_as_base64() FULL_PAGE_SCREENSHOTr0r,r(r%r%r&rhs z,WebDriver.get_full_page_screenshot_as_base64cOt)zBDownload file functionality is not implemented for Firefox driver.NotImplementedErrorr argskwargsr%r%r& download_filezWebDriver.download_filecOrk)zKGet downloadable files functionality is not implemented for Firefox driver.rlrnr%r%r&get_downloadable_filesrrz WebDriver.get_downloadable_filescOrk)zNDelete downloadable files functionality is not implemented for Firefox driver.rlrnr%r%r&delete_downloadable_filesrrz#WebDriver.delete_downloadable_files)NNT)rN)F)__name__ __module__ __qualname____doc__CONTEXT_CHROMECONTEXT_CONTENTrrboolrrr.rr+strrTrXrdrebytesr`rhrqrsrt __classcell__r%r%r#r&rs:&   #   r)rDr:r]r> contextlibrior'selenium.webdriver.common.driver_finderr"selenium.webdriver.firefox.optionsr,selenium.webdriver.firefox.remote_connectionr"selenium.webdriver.firefox.servicer#selenium.webdriver.remote.webdriverrRemoteWebDriverr%r%r%r&s