o ѷ6ib@sHddlZddlZddlmZddlmZddlmZGdddeZdS)N)BinaryIO)DesiredCapabilities) ArgOptionscs6eZdZdZd fdd ZedefddZejdeddfd dZededBfd d Z e jdeddfd d Z ede efd dZ deddfddZ deddfddZ edefddZdedeeBeBe eBddfddZedefddZejdeddfddZdefddZedefddZZS)!ChromiumOptionszgoog:chromeOptionsreturnNcs2td|_g|_g|_i|_d|_d|_dS)z1Initialize ChromiumOptions with default settings.NF)super__init___binary_location_extension_files _extensions_experimental_options_debugger_address_enable_webextensionsself __class__h/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/selenium/webdriver/chromium/options.pyr s  zChromiumOptions.__init__cC|jS)z>Returns the location of the binary, otherwise an empty string.)r rrrrbinary_location'zChromiumOptions.binary_locationvaluecCst|ts t|j||_dS)zzAllows you to set where the chromium binary lives. Args: value: Path to the Chromium binary. N) isinstancestr TypeErrorBINARY_LOCATION_ERRORr rrrrrr,s   cCr)z4Returns the address of the remote devtools instance.)rrrrrdebugger_address7rz ChromiumOptions.debugger_addresscCst|ts td||_dS)zSet the address of the remote devtools instance for active wait connection. Args: value: Address of remote devtools instance if any (hostname[:port]). z!Debugger Address must be a stringN)rrrrrrrrr<s  c Csddtdtfdd}g}|jD]}t|d}|||Wdn1s'wYq||jS)z9Returns a list of encoded extensions that will be loaded. file_datarcSst|dS)Nzutf-8)base64 b64encodereaddecode)r rrr_decodeKsz+ChromiumOptions.extensions.._decoderbN)rrr openappendr )rr%encoded_extensions extensionfrrr extensionsGs   zChromiumOptions.extensionsr*cCsD|rtjtj|}tj|r|j|dStdtd)zAdd the path to an extension to be extracted to ChromeDriver. Args: extension: Path to the *.crx file. z#Path to the extension doesn't existargument can not be nullN) ospathabspath expanduserexistsr r(OSError ValueError)rr*extension_to_addrrr add_extensionXs  zChromiumOptions.add_extensioncCs|r |j|dStd)zAdd Base64-encoded string with extension data to be extracted to ChromeDriver. Args: extension: Base64 encoded string with extension data. r-N)r r(r4)rr*rrradd_encoded_extensiongsz%ChromiumOptions.add_encoded_extensioncCr)z:Returns a dictionary of experimental options for chromium.r rrrrexperimental_optionsrrz$ChromiumOptions.experimental_optionsnamecCs||j|<dS)zAdds an experimental option which is passed to chromium. Args: name: The experimental option name. value: The option value. Nr8)rr:rrrradd_experimental_optionwsz'ChromiumOptions.add_experimental_optioncCr)zKReturn whether webextension support is enabled for Chromium-based browsers.)rrrrrenable_webextensionsrz$ChromiumOptions.enable_webextensionscCs`||_|rddg}|D] }||jvr||q dSddg}|D] }||jvr-|j|q dS)aEnables or disables webextension support for Chromium-based browsers. Args: value: True to enable webextension support, False to disable. Notes: - When enabled, this automatically adds the required Chromium flags: - --enable-unsafe-extension-debugging - --remote-debugging-pipe - When disabled, this removes BOTH flags listed above, even if they were manually added via add_argument() before enabling webextensions. - Enabling --remote-debugging-pipe makes the connection b/w chromedriver and the browser use a pipe instead of a port, disabling many CDP functionalities like devtools z#--enable-unsafe-extension-debuggingz--remote-debugging-pipeN)r _arguments add_argumentremove)rrrequired_flagsflagflags_to_removerrrr<s    cCsd|j}|j}|jr||j|j|d<|jr|j|d<|j|d<|jr+|j|d<|||j <|S)zCreates a capabilities with all the options that have been set. Returns: A dictionary with all set options. r,binaryargsdebuggerAddress) _capsr9copymobile_optionsupdater,rr=rKEY)rcapschrome_optionsrrrto_capabilitiess       zChromiumOptions.to_capabilitiescCs tjS)N)rCHROMErGrrrrdefault_capabilitiess z$ChromiumOptions.default_capabilities)rN)__name__ __module__ __qualname__rJr propertyrrsetterrlistr,r6r7dictr9intr;boolr<rMrO __classcell__rrrrrs2    & r) r!r.typingr.selenium.webdriver.common.desired_capabilitiesr!selenium.webdriver.common.optionsrrrrrrs