8hddlZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddlmZddlmZdd lmZej(eZGd d e Z y) N)contextmanager)BytesIO) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)ServiceceZdZdZdZdZ ddededdffd Zdfd Z dd Z e d Z dde fd Zdd ZdefdZdefdZdefdZde fdZxZS)rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNoptionsservicereturncb|r|n t|_|r|n t}tj|j||j_|jj t|jj|j|}t|-||d|_ y)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_addr ignore_proxy keep_alive)command_executorrFN) r rrrget_pathpathstartr service_url_ignore_local_proxysuper__init__ _is_remote)selfrrrexecutor __class__s {/home/stella.sw7ft.com/public_html/site_rebuilder/venv/lib/python3.12/site-packages/selenium/webdriver/firefox/webdriver.pyrzWebDriver.__init__)s#*wwy $'')(11$,,H  *#||77 44!  (GDcv t||jj y#t$rY&wxYw)z3Quits the driver and close every associated window.N)rquit Exceptionrstop)rr s r!r$zWebDriver.quitGs7  GLN     s , 88c,|jdd|iy)N SET_CONTEXTcontextexecute)rr)s r! set_contextzWebDriver.set_contextQs ]Y$89r"c#K|jdjd}|j| d|j|y#|j|wxYww)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_contexts r!r)zWebDriver.contextTsS,,}599'B ! .    _ -D  _ -s2A!A A! AA!c tjj|rt}t |dz}t j |dt j5}tj|D]B\}}}|D]7} tjj|| } |j| | |d9D dddtj|jjd} nGt|d5} tj| j!jd} ddd |d} |j#d| dS#1swYxYw#1swY/xYw) 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+)rrr6fp path_rootzippedbase_filesfylefilenamer5filepayloads r! install_addonzWebDriver.install_addonjs= 77== BD A IS'*>*>? E6&(ggdmEND!U %E#%77<<d#; Xx /CDEE E $$R[[]3::7CEdD! FT((5<> E E F FsAE3E!E!E*c,|jdd|iy)zUninstalls Firefox addon using its identifier. :Usage: :: driver.uninstall_addon('addon@foo.com') UNINSTALL_ADDONidNr*)r identifiers r!uninstall_addonzWebDriver.uninstall_addons &z(:;r"c*|jjdstjdt|j } t |d5}|j|ddd~y#1swY xYw#t$rY~ywxYw#~wxYw)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_pngrEr@OSError)rrNpngfs r! get_full_page_screenshot_as_filez*WebDriver.get_full_page_screenshot_as_files~~((0 MMp 224 h%         s< A?A3)A?3A<8A?? B B B  BBc$|j|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') )rb)rrNs r!save_full_page_screenshotz#WebDriver.save_full_page_screenshots44X>>r"cftj|jjdS)zGets the full document screenshot of the current window as a binary data. :Usage: :: driver.get_full_page_screenshot_as_png() ascii)rA b64decode"get_full_page_screenshot_as_base64encoders r!r^z)WebDriver.get_full_page_screenshot_as_pngs* G G I P PQX YZZr"c*|jddS)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*rjs r!rhz,WebDriver.get_full_page_screenshot_as_base64s||23G<<D:?T? [ [ =C =r"r)rAloggingr8r[r; contextlibrior'selenium.webdriver.common.driver_finderr#selenium.webdriver.remote.webdriverrRemoteWebDriverrrremote_connectionr rr getLoggerrmloggerr"r!rsK" %@L6   8 $q=q=r"