a Šxd8ã@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)ÚAddonFormatErrorz4Exception for not well-formed add-on manifest files.N)Ú__name__Ú __module__Ú __qualname__Ú__doc__©r r úb/var/www/html/Ranjet/env/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œdd„ƒZ e dd„ƒZej ddœdd„ƒZe dd„ƒZej ddœdd„ƒZe edœdd„ƒZdd„Zdd„Zdd„Zd%d d!„Zd"d#„ZdS)&ÚFirefoxProfileZWEBDRIVER_ANONYMOUS_PROFILENcCs6tjdtddtjs`ttj tj  t ¡t ¡dd}t   |¡t_Wdƒn1sV0Yt tjd¡|_||_d|_|js| ¡|_n`t ¡|_tj |jd¡}tj|j|t d d d ¡d ||_t |jd ¡| tj |jd¡¡tj |jd¡|_tj |jd¡|_tj |j¡r2t |jd¡dS)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. zAfirefox_profile has been deprecated, please use an Options objecté)Ú stacklevelúutf-8©ÚencodingNZmutablezwebdriver-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__.s4 ÿÿ*  ÿzFirefoxProfile.__init__cCs||j|<dS)z0sets the preference that we want in the profile.N©r%©r2ÚkeyÚvaluer r r Úset_preferenceUszFirefoxProfile.set_preferencecCs| |¡dS)N)Ú_install_extension)r2Ú extensionr r r Ú add_extensionYszFirefoxProfile.add_extensioncCs<tjd ¡D]\}}||jvr||j|<q| |j¡dS)NÚfrozen)r rÚitemsr%Ú_write_user_prefsr5r r r Úupdate_preferences\s  z!FirefoxProfile.update_preferencescCs|jS)z8Gets the profile directory that is currently being used.)r&©r2r r r rgszFirefoxProfile.pathcCs|jS)z+Gets the port that WebDriver is working on.)Ú_portr@r r r ÚportlszFirefoxProfile.port)Úreturnc Csnt|tƒstdƒ‚z$t|ƒ}|dks,|dkr4tdƒ‚WnttfyTtdƒ‚Yn0||_| d|j¡dS)z0Sets 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Ú TypeErrorrAr8)r2rBr r r rBqs  cCs |jdS)NÚ webdriver_accept_untrusted_certsr4r@r r r Úaccept_untrusted_certssz%FirefoxProfile.accept_untrusted_certscCs |dvrtdƒ‚| d|¡dS)N©TFú%Please pass in a Boolean to this callrI©rr8©r2r7r r r rJƒscCs |jdS)NÚ!webdriver_assume_untrusted_issuerr4r@r r r Úassume_untrusted_cert_issuer‰sz+FirefoxProfile.assume_untrusted_cert_issuercCs |dvrtdƒ‚| d|¡dS)NrKrLrOrMrNr r r rPsc Cs¤| ¡tƒ}t |dtj¡b}t|jƒd}t |j¡D]6\}}}|D]&}tj  ||¡}|  |||d…¡qHq:Wdƒn1s†0Yt   |  ¡¡ d¡S)znA zipped, base64 encoded string of profile directory for use with remote WebDriver JSON wire protocol.ÚwrDNzUTF-8)r?rÚzipfileÚZipFileÚ ZIP_DEFLATEDÚlenrrÚwalkrÚwriteÚbase64Ú b64encodeÚgetvalueÚdecode) r2ÚfpZzippedZ path_rootÚbaseÚ_ÚfilesZfyleÚfilenamer r r Úencoded”s6zFirefoxProfile.encodedcCst ¡S)z9Creates a temp folder to store User.js and the extension.)r(r)r@r r r r'¢sz!FirefoxProfile._create_tempfolderc Csdt|jddd@}| ¡D]&\}}| d|›dt |¡›d¡qWdƒn1sV0YdS)z1writes the current user prefs dictionary to disk.rQrrz user_pref("z", z); N)rr0r=rWr!Údumps)r2Z user_prefsÚfr6r7r r r r>¦sz FirefoxProfile._write_user_prefsc Csºt d¡}z˜t|ddx}|D]b}| |¡}z t | d¡¡|j| d¡<Wqty~t   d| d¡| d¡›¡Yq0qWdƒn1s–0YWnty´Yn0dS)Nzuser_pref\("(.*)",\s(.*)\)rrr rDz5(skipping) failed to json.loads existing preference: ) ÚreÚcompilerÚsearchr!ÚloadsÚgroupr%Ú Exceptionrr)r2ZuserjsZ pref_patternrcZusrÚmatchesr r r r.¬s    ÿ. 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 ¡} | sRJd |›ƒ‚tj |j| ¡} |s´| d s´|r´tj |j¡s¢t |j¡t |jd ¡t || d¡ntj | ¡sÒtj|| d d|rât |¡dS)zñ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: ÚunpackrT)Úsymlinks)Ú WEBDRIVER_EXTrrrrrÚendswithr(r)ÚsplitrRrSÚnamelistÚisdirÚmakedirsÚreadrrWÚ_addon_detailsÚgetr/Úexistsr-r*r#r+Úrmtree) r2ZaddonrrZtmpdirZxpifileÚcompressed_fileÚnameÚdatarcZ addon_detailsZaddon_idÚ addon_pathr r r r9¼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  |¡rLtj  |d ¡}tj |¡r t |dd} ||  ¡ƒWdƒWS1s0Yt tj  |d ¡dd} |  ¡}Wdƒn1s@0Yntd|›ƒ‚Wn@tt fyœ} z"tt| ƒt ¡dƒ‚WYd} ~ n d} ~ 00zÐt |¡} || dƒ} || dƒ} |  | d¡ d¡}|sè|  d¡ d¡}|jD]0}|j | d¡}||vrî| |||ƒi¡qî| d¡slt|jjƒD]2}|j |¡}|j| dkr8| d|ji¡q8Wn<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)rqrrr€ÚversioncSs^|jj}d}t|jƒD]B}| |¡j|krd| |¡jvr| |¡j d¡dd}qZq|S)NÚú:rD)ZdocumentElementÚ attributesÚrangeÚlengthÚitemr7r€rv)ÚdocÚurlr†Ú namespaceÚir r r Úget_namespace_idûsz7FirefoxProfile._addon_details..get_namespace_idcSs6g}|jD]}|j|jkr | |j¡q d |¡ ¡S)z(Retrieve the text value of a given node.r„)Ú 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) z[Extracts the details from the contents of a WebExtensions `manifest.json` file.Z applicationsZgeckorqr€ú r„ú@rƒF)rqrƒr€rr)r!rgÚKeyErrorÚreplace)ÚcontentÚmanifestrqr r r Úparse_manifest_jsons  "üz:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: rnz manifest.jsonz install.rdfrrz/Add-on path is neither an XPI nor a directory: r z#http://www.mozilla.org/2004/em-rdf#z+http://www.w3.org/1999/02/22-rdf-syntax-ns#Ú Descriptionrr„rqrrÚtruezAdd-on id could not be found.)#rrr}ÚOSErrorrRÚ is_zipfilerSrwrzÚcloserxrrr˜rÚstrÚsysÚexc_inforÚ parseStringZgetElementsByTagNamer‰rÚnodeNamer™Úupdater|r‡r†rˆr€r7rirEÚlower)r2r‚ÚdetailsrŽr•rœrr›Zmanifest_json_filenamercÚerŠÚemÚrdfÚ descriptionr”ÚentryrÚ attributer r r r{ìsb      ü .*,       , zFirefoxProfile._addon_details)N)T)rrrZANONYMOUS_PROFILE_NAMErr3r8rtr;r?ÚpropertyrrBÚsetterrJrPr¢rar'r>r.r9r{r r r r r *s6 '         0r )rXr#r!rrdr*r£r(rrRÚiorÚxml.domrZselenium.common.exceptionsrrtr ZEXTENSION_NAMErirr r r r r Ús"