`1hmddlZddlZddlZddlZddlmZddlmZddlm Z ddl m Z ddl m ZddlmZdd lmZdd lmZGd d eZ y) N)contextmanager)BytesIO)Optional) DriverFinder) WebDriver)Options)FirefoxRemoteConnection)Servicec eZdZdZdZdZ ddeedeede ddffd Z dfd Z dd Z e d Zddefd ZddZde fdZde fdZdefdZdefdZdZdZxZS)rz=Controls the GeckoDriver and allows you to drive the browser.chromecontentNoptionsservice keep_alivereturnc<|r|n t|_|r|n t}t|j|}|j r|j |_d|_|jjxs|j|j_ |jjt|jj||j} t|=||d|_y#t $r|j#wxYw)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_executorrF)r rr rget_browser_pathbinary_locationbrowser_versionenv_pathget_driver_pathpathstartr service_url_ignore_local_proxysuper__init__ Exceptionquit _is_remote)selfrrrfinderexecutor __class__s b/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/selenium/webdriver/firefox/webdriver.pyr!zWebDriver.__init__'s#*wwy $'')dllG4  " " $&,&=&=&?G #&*G # LL113Ov7M7M7O  *#||77! 44    G h  H     IIK  s 'DDc t||jj y#t$rY&wxYw#|jj wxYw)z=Closes the browser and shuts down the GeckoDriver executable.N)r r#r"rstop)r%r(s r)r#zWebDriver.quitOsJ GLN LL        LL   s, 8;8;Ac,|jdd|iy)N SET_CONTEXTcontextexecute)r%r.s r) set_contextzWebDriver.set_contextYs ]Y$89c#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)r0popr1)r%r.initial_contexts r)r.zWebDriver.context\sS,,}599'B ! .    _ -D  _ -s2A!A A! AA!c tjj|rt}tjj |}t |dz}t j|dt jd5}tj|D]B\}}}|D]7} tjj|| } |j| | |d9D dddtj|jjd} nGt!|d5} tj| j#jd} ddd |d} |j%d | d S#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") rwF)strict_timestampsNzUTF-8rb)addon temporary INSTALL_ADDONr5)osrisdirrnormpathlenzipfileZipFile ZIP_DEFLATEDwalkjoinwritebase64 b64encodegetvaluedecodeopenreadr0)r%rr=fp path_rootzippedbase_filesfylefilenamer<filepayloads r) install_addonzWebDriver.install_addonrsU 77== B77##D)DD A IS'*>*>RWX E\b&(ggdmEND!U %E#%77<<d#; Xx /CDEE E $$R[[]3::7CEdD! FT((5<> E E F Fs>AE6!3F6E?F 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(:;r2c*|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_pngrMrHOSError)r%rVpngfs 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") )rj)r%rVs r)save_full_page_screenshotz#WebDriver.save_full_page_screenshots44X>>r2cftj|jjdS)zGets the full document screenshot of the current window as a binary data. :Usage: :: driver.get_full_page_screenshot_as_png() ascii)rI b64decode"get_full_page_screenshot_as_base64encoder%s r)rfz)WebDriver.get_full_page_screenshot_as_pngs* G G I P PQX YZZr2c*|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_SCREENSHOTr5r/rrs r)rpz,WebDriver.get_full_page_screenshot_as_base64s||23G<rs9" %@L6D"D"r2