o ҷh7@sddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZdZdZdZGdd d eZGd d d ZdS) N)BytesIO)minidom)WebDriverExceptionz webdriver.xpizwebdriver_prefs.jsonzfxdriver@googlecode.comc@seZdZdZdS)AddonFormatErrorz4Exception for not well-formed add-on manifest files.N)__name__ __module__ __qualname____doc__r r ]/var/www/html/venv/lib/python3.10/site-packages/selenium/webdriver/firefox/firefox_profile.pyr&src@seZdZdZdZd$ddZddZefddZd d Z e d d Z e d dZ e j d%ddZ e ddZej d%ddZe ddZej d%ddZe defddZddZddZddZd&d d!Zd"d#ZdS)'FirefoxProfileWEBDRIVER_ANONYMOUS_PROFILENcCs(tjs(ttjtjttdd}t |t_Wdn1s#wYt tjd|_ ||_d|_|js@||_n0t|_tj|jd}tj|j|tdddd ||_t|jd |tj|jd tj|jd |_tj|jd |_tj|jrt|jd dSdS)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. utf-8encodingNmutablezwebdriver-py-profilecopyz parent.locklockz .parentlock)ignorezuser.js extensionsi)r DEFAULT_PREFERENCESopenospathjoindirname__file__WEBDRIVER_PREFERENCESjsonloadcopydeepcopydefault_preferences profile_dir tempfolder_create_tempfoldertempfilemkdtempshutilcopytreeignore_patternschmod_read_existing_userjs extensionsDir userPrefsisfile)selfprofile_directory default_prefsnewprofr r r __init__.s2   zFirefoxProfile.__init__cCs||j|<dS)z0Sets the preference that we want in the profile.Nr"r0keyvaluer r r set_preferenceRszFirefoxProfile.set_preferencecCs||dSN)_install_extension)r0 extensionr r r add_extensionVszFirefoxProfile.add_extensioncCs<tjdD]\}}||jvr||j|<q||jdS)Nfrozen)r ritemsr"_write_user_prefsr6r r r update_preferencesYs   z!FirefoxProfile.update_preferencescC|jS)z8Gets the profile directory that is currently being used.)r#r0r r r rdzFirefoxProfile.pathcCrB)z+Gets the port that WebDriver is working on.)_portrCr r r portirDzFirefoxProfile.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 TypeErrorrEr9)r0rFr r r rFns cC |jdS)N webdriver_accept_untrusted_certsr5rCr r r accept_untrusted_certs| z%FirefoxProfile.accept_untrusted_certscC"t|ts td|d|dS)N%Please pass in a Boolean to this callrOrJboolrr9r0r8r r r rPs cCrN)N!webdriver_assume_untrusted_issuerr5rCr r r assume_untrusted_cert_issuerrQz+FirefoxProfile.assume_untrusted_cert_issuercCrR)NrSrWrTrVr r r rXs c Cs|t}t|dtj1}t|jd}t|jD]\}}}|D]}tj ||}| |||dq$qWdn1sCwYt | dS)znA zipped, base64 encoded string of profile directory for use with remote WebDriver JSON wire protocol.wrHNzUTF-8)rArzipfileZipFile ZIP_DEFLATEDlenrrwalkrwritebase64 b64encodegetvaluedecode) r0fpzipped path_rootbase_filesfylefilenamer r r encodedszFirefoxProfile.encodedcCstS)z9Creates a temp folder to store User.js and the extension.)r&r'rCr r r r%sz!FirefoxProfile._create_tempfolderc Csft|jddd!}|D]\}}|d|dt|dq WddS1s,wYdS)z1Writes the current user prefs dictionary to disk.rYrrz user_pref("z", z); N)rr.r?r_rdumps)r0 user_prefsfr7r8r r r r@s  "z FirefoxProfile._write_user_prefsc Cstd}zNt|dd=}|D]0}||}zt|d|j|d<Wqty?t d|d|dYqwWdWdS1sLwYWdSty]YdSw)Nzuser_pref\("(.*)",\s(.*)\)rrrHz5(skipping) failed to json.loads existing preference: ) recompilersearchrloadsgroupr" Exceptionwarningswarn)r0userjs pref_patternrousrmatchesr r r r,s"    & z$FirefoxProfile._read_existing_userjsTc Cs|tkrtjtjtt}d}d}|drtjdtj |dd}t |d}| D]^}|drMtj tj||sLttj||q0tj tjtj||sjttjtj||||}ttj||d }||Wdn1swYq0|}|}||} | d } | sJd |tj|j| } |s| d s|rtj|jst|jt|jd t|| dntj| stj|| d d|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: unpackrT)symlinks) WEBDRIVER_EXTrrrrrendswithr&r'splitrZr[namelistisdirmakedirsreadrr__addon_detailsgetr-existsr+r(r r)rmtree) r0addonrtmpdirxpifilecompressed_filenamedataro addon_detailsaddon_id addon_pathr r r r;sF          z!FirefoxProfile._install_extensionc Csddddd}dd}dd}dd }tj|s td |zt|rPz$t|d }d |vr@||d W| WS|d }W| n\| wtj |rtj |d }tj|rt |dd} || WdWS1s{wYt tj |d dd } | }Wdn1swYntd|Wntt fy} z tt| tdd} ~ wwzet|} || d} || d} | | dd}|s| dd}|jD]}|j| d}||vr||||iq|ds&t|jjD]}|j|}|j| dkr$|d|jiq Wnt y?} z tt| tdd} ~ wwt!|dtrR|d"dk|d<|ds\td|S)aReturns a dictionary of details about the addon. :param addon_path: path to the add-on directory or XPI Returns:: {'id': u'rainbow@colors.org', # id of the addon 'version': u'1.4', # version of the addon 'name': u'Rainbow', # name of the addon 'unpack': False } # whether to unpack the addon NF)rrrversioncSs`|jj}d}t|jD]"}||j|kr-d||jvr-||jddd}|Sq |S)N:rH)documentElement attributesrangelengthitemr8rr)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. applicationsgeckorr r@rF)rrrr)rrtKeyErrorreplace)contentmanifestrr r r parse_manifest_json s   z:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: rz manifest.jsonz install.rdfrrz/Add-on path is neither an XPI nor a directory: rpz#http://www.mozilla.org/2004/em-rdf#z+http://www.w3.org/1999/02/22-rdf-syntax-ns# DescriptionrrrrtruezAdd-on id could not be found.)#rrrOSErrorrZ is_zipfiler[rrcloserrrrrstrsysexc_infor parseStringgetElementsByTagNamerrnodeNamerupdaterrrrrr8rvrJlower)r0rdetailsrrrrrmanifest_json_filenameroeremrdf descriptionrentryr attributer r r rst           "        zFirefoxProfile._addon_detailsr:)rGN)T)rrrANONYMOUS_PROFILE_NAMErr4r9rr=rApropertyrrFsetterrPrXrrlr%r@r,r;rr r r r r *s8 $           0r )r`r rrrqr(rr&rwrZiorxml.domrselenium.common.exceptionsrrrEXTENSION_NAMErvrr r r r r s$