o 5@Hh5@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlmZdZedGdd d eZGd d d ZdS) N)BytesIO)minidom) deprecated)WebDriverExceptionzwebdriver_prefs.json/Addons must be added after starting the sessionc@seZdZdZdS)AddonFormatErrorz4Exception for not well-formed add-on manifest files.N)__name__ __module__ __qualname____doc__r r j/home/air/sanwanet/gpt-api/venv/lib/python3.10/site-packages/selenium/webdriver/firefox/firefox_profile.pyr&src@seZdZdZd"ddZddZedd"ddZd d Ze d d Z e ed ddZ e j ed d#ddZ e edddZ e j edd#ddZ e edddZej edd#ddZe defddZddZedd$ddZedd d!ZdS)%FirefoxProfileNcCsi|_|r'tjtd}tj||tdddd||_ t |j dd St|_ t j sTt tjtjttdd}t|t _ Wd n1sOwYtt j d |_t j d D] \}}||j|<qdd S) aInitialises a new instance of a Firefox Profile. :args: - profile_directory: Directory of profile that you want to use. If a directory is passed in it will be cloned and the cloned directory will be used by the driver when instantiated. This defaults to None and will create a new directory when object is created. zwebdriver-py-profilecopyz parent.locklockz .parentlock)ignoreutf-8encodingNmutablefrozen)_desired_preferencesospathjointempfilemkdtempshutilcopytreeignore_patterns _profile_dirchmodrDEFAULT_PREFERENCESopendirname__file__WEBDRIVER_PREFERENCESjsonloadcopydeepcopyitems)selfprofile_directorynewprof default_prefskeyvaluer r r __init__.s(   zFirefoxProfile.__init__cCs||j|<dS)z0Sets the preference that we want in the profile.Nr)r,r0r1r r r set_preferenceMzFirefoxProfile.set_preferencercCs||dSN)_install_extension)r, extensionr r r add_extensionQr5zFirefoxProfile.add_extensionc Cstj|jd}tj|rt|d||t|ddd"}|j D]\}}| d|dt |dq&Wd d S1sEwYd S) z&Writes the desired user prefs to disk.zuser.jsiwrrz user_pref("z", z); N) rrrr isfiler!_read_existing_userjsr#rr+writer'dumps)r, user_prefsfr0r1r r r update_preferencesUs    "z!FirefoxProfile.update_preferencescC|jS)z8Gets the profile directory that is currently being used.)r r,r r r raszFirefoxProfile.pathz'The port is stored in the Service classcCrB)z+Gets the port that WebDriver is working on.)_portrCr r r portfszFirefoxProfile.portreturnc Csjt|ts tdzt|}|dks|dkrtdWn ttfy(tdw||_|d|jdS)z0Sets the port that WebDriver will be running on.zPort needs to be an integeriz)Port number must be in the range 1..65535webdriver_firefox_portN) isinstanceintr ValueError TypeErrorrDr4)r,rEr r r rEls z8Allowing untrusted certs is toggled in the Options classcC |jdS)N webdriver_accept_untrusted_certsr3rCr r r accept_untrusted_certs{ z%FirefoxProfile.accept_untrusted_certscC"t|ts td|d|dS)N%Please pass in a Boolean to this callrNrIboolrr4r,r1r r r rOs cCrM)N!webdriver_assume_untrusted_issuerr3rCr r r assume_untrusted_cert_issuerrPz+FirefoxProfile.assume_untrusted_cert_issuercCrQ)NrRrVrSrUr r r rWs c Cs|jr|t}tj|dtjdd1}t|jd}t |jD]\}}}|D]}tj ||}| |||dq)q"Wdn1sHwYt |dS)z]Updates preferences and creates a zipped, base64 encoded string of profile directory.r:F)strict_timestampsrGNzUTF-8)rrArzipfileZipFile ZIP_DEFLATEDlenrrwalkrr=base64 b64encodegetvaluedecode) r,fpzipped path_rootbase_filesfylefilenamer r r encodedszFirefoxProfile.encodedc Cstd}t|dd<}|D]0}||}zt|d|j|d<Wqty>t d|d|dYqwWddS1sJwYdS)zRReads existing preferences and adds them to desired preference dictionary.zuser_pref\("(.*)",\s(.*)\)rrrGz5(skipping) failed to json.loads existing preference: N) recompiler#searchr'loadsgroupr Exceptionwarningswarn)r,userjs pref_patternr@usrmatchesr r r r<s    "z$FirefoxProfile._read_existing_userjsTc Csd}d}|drtjdtj|dd}t|d}|D]^}|dr>tj tj ||s=t tj ||q!tj tj tj ||s[t tj tj ||| |}ttj ||d }||Wdn1szwYq!|}|}||} | d } | sJd |tj |jd } tj | | } |s| d s|rtj| st | t| d t|| dntj| stj|| dd|rt|dSdS)zInstalls addon from a filepath, url or directory of addons in the profile. - path: url, absolute path to .xpi, or directory of addons - unpack: whether to unpack unless specified otherwise in the install.rdf Nz.xpi.)suffixr/wbidz!The addon id could not be found: extensionsunpackrT)symlinks)endswithrrrrsplitrYrZnamelistisdirrmakedirsr$readr#r=_addon_detailsgetr existsr!rr)rrmtree) r,addonrtmpdirxpifilecompressed_filenamedatar@ addon_detailsaddon_idextensions_dir addon_pathr r r r7sD            z!FirefoxProfile._install_extensionc Csddddd}dd}dd}dd }tj|s td |zt|rWt|d "}d |vrB||d WdWS|d }Wdn1sQwYnWtj |rtj |d }tj|rt |dd} || WdWS1swYt tj |d dd } | }Wdn1swYntd|Wntt fy} z t t| tdd} ~ wwzft|} || d} || d} | | dd}|s| dd}|jD]}|j| d}||vr||||iq|ds.t|jjD]}|j|}|j| dkr,|d|jiqWntyG} z t t| tdd} ~ wwt |dtrZ|d!dk|d<|dsdt d|S)aReturns a dictionary of details about the addon. :param addon_path: path to the add-on directory or XPI Returns:: { "id": "rainbow@colors.org", # id of the addon "version": "1.4", # version of the addon "name": "Rainbow", # name of the addon "unpack": False, } # whether to unpack the addon NF)r~rrversioncSs`|jj}d}t|jD]"}||j|kr-d||jvr-||jddd}|Sq |S)N:rG)documentElement attributesrangelengthitemr1rr)docurlr namespaceir r r get_namespace_idsz7FirefoxProfile._addon_details..get_namespace_idcSs6g}|jD]}|j|jkr||jqd|S)z(Retrieve the text value of a given node.r) childNodesnodeType TEXT_NODEappendrrstrip)elementrcnoder r r get_texts    z/FirefoxProfile._addon_details..get_textcSsdt|}z |ddd}Wnty&|dddd|d}Ynw||d|dd d S) z[Extracts the details from the contents of a WebExtensions `manifest.json` file. applicationsgeckor~r r@rF)r~rrr)r'roKeyErrorreplace)contentmanifestr~r r r parse_manifest_jsons   z:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: r{z manifest.jsonz install.rdfrrz/Add-on path is neither an XPI nor a directory: rkz#http://www.mozilla.org/2004/em-rdf#z+http://www.w3.org/1999/02/22-rdf-syntax-ns# Descriptionrrr~rtruezAdd-on id could not be found.)"rrrOSErrorrY is_zipfilerZrrrrr#rrstrsysexc_infor parseStringgetElementsByTagNamerrnodeNamerupdaterrrrrr1rqrIlower)r,rdetailsrrrrrmanifest_json_filenamer@eremrdf descriptionrentryr attributer r r rst         "         zFirefoxProfile._addon_detailsr6)rFN)T)rr r r"r2r4rr9rApropertyrrEsetterrOrWrrjr<r7rr r r r r+sD        .r)r^r)r'rrlrrrrrrYiorxml.domrtyping_extensionsrselenium.common.exceptionsrr&rqrrr r r r s$