o VhU@sddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZeeZGd d d e Z dS) N)contextmanager)BytesIO) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)Servicec seZdZdZdZdZ   ddededed dffd d Z d fd d Z d ddZ e ddZ d!d efddZd ddZd efddZd efddZd efddZd efddZZS)"rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNToptionsservice keep_alivereturncs|r|nt|_|r |nt}t|j||j_|jt|jj||j d}z t j ||dWn t y=| wd|_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. )remote_server_addrr ignore_proxy)command_executorr FN)r r rrget_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfr r rexecutor __class__a/home/air/yokohama/back/venv/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 stoprrr!r"rLs zWebDriver.quitcCs|dd|idS)N SET_CONTEXTcontextexecute)rr&r!r!r" set_contextVszWebDriver.set_contextc cs@|dd}||z dVW||dS||w)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(popr))rr&initial_contextr!r!r"r&Ys  zWebDriver.contextFc Cstj|rSt}t|d}t|dtj)}t|D]\}}}|D]} tj || } | | | |dq$qWdn1sCwYt | d} nt|d} t | d} Wdn1smwY| |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 temporary INSTALL_ADDONr+)osrisdirrlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr()rrr1fp path_rootzippedbase_filesfylefilenamer0filepayloadr!r!r" install_addonos"    zWebDriver.install_addoncCs|dd|idS)zUninstalls Firefox addon using its identifier. :Usage: :: driver.uninstall_addon('addon@foo.com') UNINSTALL_ADDONidNr')r identifierr!r!r"uninstall_addonszWebDriver.uninstall_addoncCs|ds tdt|}z1z$t|d }||Wdn 1s(wYWW~dSWW~dStyBYW~dSw~w)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_pngr@r;OSError)rrIpngfr!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[)rrIr!r!r"save_full_page_screenshots z#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"rWs 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() FULL_PAGE_SCREENSHOTr+r'r$r!r!r"r_s z,WebDriver.get_full_page_screenshot_as_base64)NNT)rN)F)__name__ __module__ __qualname____doc__CONTEXT_CHROMECONTEXT_CONTENTrr boolrrr)rr&strrLrPr[r\bytesrWr_ __classcell__r!r!rr"r#s4#     r)r<loggingr3rTr6 contextlibrior'selenium.webdriver.common.driver_finderr#selenium.webdriver.remote.webdriverrRemoteWebDriverr rremote_connectionrr r getLoggerrbloggerr!r!r!r"s