a ¾Ê.c’8ã@s¢ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZdZdZdZGdd „d eƒZGd d „d ƒZdS) éN)ÚBytesIO)Úminidom)ÚWebDriverExceptionz webdriver.xpizwebdriver_prefs.jsonzfxdriver@googlecode.comc@seZdZdZdS)ÚAddonFormatErrorz3Exception for not well-formed add-on manifest filesN)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úd/var/www/brookimports/venv/lib/python3.9/site-packages/selenium/webdriver/firefox/firefox_profile.pyr'src@sÊeZdZdZdZd#dd„Zdd„Zefdd„Zd d „Z e d d „ƒZ e d d„ƒZ e j dd„ƒZ e dd„ƒZej dd„ƒZe dd„ƒZej dd„ƒZe dd„ƒZdd„Zdd„Zdd„Zd$dd „Zd!d"„ZdS)%ÚFirefoxProfileZWEBDRIVER_ANONYMOUS_PROFILENcCs2tjdtddtjs\ttj tj  t ¡t ¡ƒ}t   |¡t_Wdƒn1sR0Yt tjd¡|_||_d|_|jsŒ| ¡|_n`t ¡|_tj |jd¡}tj|j|t ddd ¡d ||_t |jd ¡| tj |jd ¡¡tj |jd ¡|_tj |jd ¡|_tj |j¡r.t |jd¡dS)a‰ Initialises 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. zAfirefox_profile has been deprecated, please use an Options objecté)Ú stacklevelNZmutablezwebdriver-py-profilecopyz parent.lockÚlockz .parentlock)Úignoreéízuser.jsÚ extensionsi¤)ÚwarningsÚwarnÚDeprecationWarningr ÚDEFAULT_PREFERENCESÚopenÚosÚpathÚjoinÚdirnameÚ__file__ÚWEBDRIVER_PREFERENCESÚjsonÚloadÚcopyÚdeepcopyÚdefault_preferencesÚ profile_dirZ tempfolderÚ_create_tempfolderÚtempfileÚmkdtempÚshutilÚcopytreeÚignore_patternsÚchmodÚ_read_existing_userjsÚ extensionsDirÚ userPrefsÚisfile)ÚselfZprofile_directoryZ default_prefsZnewprofr r r Ú__init__/s8 ÿÿ*ÿ    ÿzFirefoxProfile.__init__cCs||j|<dS)zB sets the preference that we want in the profile. N©r"©r/ÚkeyÚvaluer r r Úset_preferenceUszFirefoxProfile.set_preferencecCs| |¡dS)N)Ú_install_extension)r/Ú extensionr r r Ú add_extension[szFirefoxProfile.add_extensioncCs<tjd ¡D]\}}||jvr||j|<q| |j¡dS)NÚfrozen)r rÚitemsr"Ú_write_user_prefsr2r r r Úupdate_preferences^s  z!FirefoxProfile.update_preferencescCs|jS)zI Gets the profile directory that is currently being used )r#©r/r r r riszFirefoxProfile.pathcCs|jS)z< Gets the port that WebDriver is working on )Ú_portr=r r r ÚportpszFirefoxProfile.portc Csnt|tƒstdƒ‚z$t|ƒ}|dks,|dkr4tdƒ‚WnttfyTtdƒ‚Yn0||_| d|j¡dS)zA Sets the port that WebDriver will be running on zPort needs to be an integeréiÿÿz)Port number must be in the range 1..65535Zwebdriver_firefox_portN)Ú isinstanceÚintrÚ ValueErrorÚ TypeErrorr>r5)r/r?r r r r?ws  cCs |jdS)NÚ webdriver_accept_untrusted_certsr1r=r r r Úaccept_untrusted_certs‡sz%FirefoxProfile.accept_untrusted_certscCs |dvrtdƒ‚| d|¡dS)N©TFú%Please pass in a Boolean to this callrE©rr5©r/r4r r r rF‹scCs |jdS)NÚ!webdriver_assume_untrusted_issuerr1r=r r r Úassume_untrusted_cert_issuer‘sz+FirefoxProfile.assume_untrusted_cert_issuercCs |dvrtdƒ‚| d|¡dS)NrGrHrKrIrJr r r rL•sc CsŒ| ¡tƒ}t |dtj¡}t|jƒd}t |j¡D]6\}}}|D]&}tj  ||¡}|  |||d…¡qFq8|  ¡t   | ¡¡ d¡S)z A zipped, base64 encoded string of profile directory for use with remote WebDriver JSON wire protocol Úwr@NzUTF-8)r<rÚzipfileÚZipFileÚ ZIP_DEFLATEDÚlenrrÚwalkrÚwriteÚcloseÚbase64Ú b64encodeÚgetvalueÚdecode) r/ÚfpZzippedZ path_rootÚbaseÚdirsÚfilesZfyleÚfilenamer r r ÚencodedœszFirefoxProfile.encodedcCst ¡S)zJ Creates a temp folder to store User.js and the extension )r%r&r=r r r r$­sz!FirefoxProfile._create_tempfolderc Cs`t|jdƒ@}| ¡D]&\}}| d|›dt |¡›d¡qWdƒn1sR0YdS)zB writes the current user prefs dictionary to disk rMz user_pref("z", z); N)rr-r:rSrÚdumps)r/Z user_prefsÚfr3r4r r r r;³sz FirefoxProfile._write_user_prefsc Cs¾ddl}t d¡}z”t|ƒx}|D]b}t ||¡}z t | d¡¡|j| d¡<Wq"t y‚|  d| d¡| d¡¡Yq"0q"Wdƒn1sš0YWnt y¸Yn0dS)Nrzuser_pref\("(.*)",\s(.*)\)r r@z7(skipping) failed to json.loads existing preference: %s) rÚreÚcompilerÚsearchrÚloadsÚgroupr"Ú Exceptionr)r/ZuserjsrZPREF_REr`ÚusrÚmatchesr r r r+»s      ÿÿ0 z$FirefoxProfile._read_existing_userjsTc Csä|tkrtj tj t¡t¡}d}d}| d¡r*tjdtj  |¡dd}t   |d¡}|  ¡D]¾}| d¡rœtj  tj ||¡¡sšt tj ||¡¡qbtj  tj tj ||¡¡¡sÖt tj tj ||¡¡¡| |¡}ttj ||¡dƒ}| |¡Wdƒqb1s0Yqb|}|}| |¡} |  d ¡} | sPJd |ƒ‚tj |j| ¡} |s²| d s²|r²tj |j¡s t |j¡t |jd ¡t || d¡ntj | ¡sÐtj|| d d|ràt |¡dS)a  Installs 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Ú.éÿÿÿÿ)ÚsuffixÚrú/ÚwbÚidz#The addon id could not be found: %sÚunpackrT)Úsymlinks)Ú WEBDRIVER_EXTrrrrrÚendswithr%r&ÚsplitrNrOÚnamelistÚisdirÚmakedirsÚreadrrSÚ_addon_detailsÚgetr,Úexistsr*r'r r(Úrmtree) r/ZaddonrpZtmpdirZxpifileÚcompressed_fileÚnameÚdatar`Z addon_detailsZaddon_idÚ addon_pathr r r r6Ìs>     ,   z!FirefoxProfile._install_extensionc Csàdddddœ}dd„}dd„}dd „}tj |¡s>td |ƒ‚zt |¡r¢zHt |d ¡}d | ¡vr€|| d ¡ƒW|  ¡WS| d ¡}W|  ¡n |  ¡0n¬tj  |¡rBtj  |d ¡}tj |¡rt |ƒ} ||  ¡ƒWdƒWS1sú0Yt tj  |d ¡ƒ} |  ¡}Wdƒn1s60Yn td|ƒ‚Wn@tt fy} z"tt| ƒt ¡dƒ‚WYd} ~ n d} ~ 00zÔt |¡} || dƒ} || dƒ} |  | d¡ d¡}|sÜ|  d¡ d¡}|jD]4}|j | d¡}|| ¡vrâ| |||ƒi¡qâ| d¡sdt|jjƒD]2}|j |¡}|j| dkr0| d|j i¡q0Wn<t!y¢} z"tt| ƒt ¡dƒ‚WYd} ~ n d} ~ 00t"|dtƒrÈ|d #¡dk|d<| d¡sÜtdƒ‚|S)a­ Returns 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)rorpr~ÚversioncSs^|jj}d}t|jƒD]B}| |¡j|krd| |¡jvr| |¡j d¡dd}qZq|S)NÚú:r@)ZdocumentElementÚ attributesÚrangeÚlengthÚitemr4r~rt)ÚdocÚurlr„Ú namespaceÚir r r Úget_namespace_idsz7FirefoxProfile._addon_details..get_namespace_idcSs6g}|jD]}|j|jkr | |j¡q d |¡ ¡S)z'Retrieve the text value of a given noder‚)Ú childNodesZnodeTypeZ TEXT_NODEÚappendrrÚstrip)ÚelementÚrcÚnoder r r Úget_texts   z/FirefoxProfile._addon_details..get_textcSsdt |¡}z|ddd}Wn.tyL|d dd¡d|d}Yn0||d|dd d œS) zOExtracts the details from the contents of a WebExtensions `manifest.json` file.Z applicationsZgeckoror~ú r‚ú@rF)rorr~rp)rrdÚKeyErrorÚreplace)ÚcontentÚmanifestror r r Úparse_manifest_json$s  "üz:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: %srlz manifest.jsonz install.rdfz1Add-on path is neither an XPI nor a directory: %sr z#http://www.mozilla.org/2004/em-rdf#z+http://www.w3.org/1999/02/22-rdf-syntax-ns#Ú Descriptionrr‚rorpÚtruezAdd-on id could not be found.)$rrr{ÚOSErrorrNÚ is_zipfilerOrurxrTrvrrr–rÚstrÚsysÚexc_inforÚ parseStringZgetElementsByTagNamer‡rZnodeNamer—ÚkeysÚupdaterzr…r„r†r~r4rfrAÚlower)r/r€ÚdetailsrŒr“ršr}r™Zmanifest_json_filenamer`ÚerˆZemÚrdfÚ descriptionr’Úentryr‹Ú attributer r r ryüsjü      ü  .*,      , zFirefoxProfile._addon_details)N)T)rrrZANONYMOUS_PROFILE_NAMErr0r5rrr8r<Úpropertyrr?ÚsetterrFrLr^r$r;r+r6ryr r r r r +s6 &           0r )rUr rrrar'r r%rrNÚiorZxml.domrZselenium.common.exceptionsrrrrZEXTENSION_NAMErfrr r r r r Ús"