U Pf5@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 N/tmp/pip-unpacked-wheel-ds5t_0qj/selenium/webdriver/firefox/firefox_profile.pyr&src@seZdZdZd"ddZddZedd#ddZd d Ze d d Z e ed ddZ e j ed ddddZ e edddZ e j edddddZ e edddZej edddddZe edddZddZedd$ddZedd d!ZdS)%FirefoxProfileNc Csi|_|rLtjtd}tj||tdddd||_ t |j dnzt|_ t j st tjtjttdd}t|t _ W5QRXtt j d |_t j d D]\}}||j|<qd 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-8encodingZmutablefrozenN)_desired_preferencesospathjointempfilemkdtempshutilcopytreeignore_patterns _profile_dirchmodrDEFAULT_PREFERENCESopendirname__file__WEBDRIVER_PREFERENCESjsonloadcopydeepcopyitems)selfZprofile_directoryZnewprofZ default_prefskeyvaluer r r __init__.s*   zFirefoxProfile.__init__cCs||j|<dS)z0Sets the preference that we want in the profile.Nr)r+r,r-r r r set_preferenceMszFirefoxProfile.set_preferencercCs||dS)N)_install_extension)r+ extensionr r r add_extensionQszFirefoxProfile.add_extensionc Cstj|jd}tj|r2t|d||t|ddd8}|j D]&\}}| d|dt |dqLW5QRXd S) z&Writes the desired user prefs to disk.zuser.jsiwrrz user_pref("z", z); N) rrrrisfiler _read_existing_userjsr"rr*writer&dumps)r+Z user_prefsfr,r-r r r update_preferencesUs   z!FirefoxProfile.update_preferencescCs|jS)z8Gets the profile directory that is currently being used.)rr+r r r raszFirefoxProfile.pathz'The port is stored in the Service classcCs|jS)z+Gets the port that WebDriver is working on.)_portr;r r r portfszFirefoxProfile.port)returnc Cspt|tstdz$t|}|dks,|dkr4tdWn ttfk rVtdYnX||_|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..65535Zwebdriver_firefox_portN) isinstanceintr ValueError TypeErrorr<r0)r+r=r r r r=ls  z8Allowing untrusted certs is toggled in the Options classcCs |jdS)N webdriver_accept_untrusted_certsr/r;r r r accept_untrusted_certs{sz%FirefoxProfile.accept_untrusted_certscCs"t|tstd|d|dS)N%Please pass in a Boolean to this callrDr@boolrr0r+r-r r r rEs cCs |jdS)N!webdriver_assume_untrusted_issuerr/r;r r r assume_untrusted_cert_issuersz+FirefoxProfile.assume_untrusted_cert_issuercCs"t|tstd|d|dS)NrFrJrGrIr r r rKs c Cs|jr|t}tj|dtjddX}t|jd}t |jD]6\}}}|D]&}tj ||}| |||dqRqDW5QRXt |dS)z]Updates preferences and creates a zipped, base64 encoded string of profile directory.r4F)strict_timestampsr?NzUTF-8)rr:rzipfileZipFile ZIP_DEFLATEDlenrrwalkrr7base64 b64encodegetvaluedecode) r+fpZzippedZ path_rootbase_filesZfylefilenamer r r encodeds"zFirefoxProfile.encodedc Cstd}t|ddp}|D]d}||}z t|d|j|d<Wqtk r~t d|d|dYqXqW5QRXdS)zRReads existing preferences and adds them to desired preference dictionary.zuser_pref\("(.*)",\s(.*)\)rrr?z5(skipping) failed to json.loads existing preference: N) recompiler"searchr&loadsgroupr Exceptionwarningswarn)r+ZuserjsZ pref_patternr9Zusrmatchesr r r r6s   z$FirefoxProfile._read_existing_userjsTc Csd}d}|drtjdtj|dd}t|d}|D]}|dr|tj tj ||st tj ||qBtj tj tj ||st tj tj ||| |}ttj ||d}||W5QRXqB|}|}||} | d } | std |tj |jd } tj | | } |s| d s|rtj| stt | t| d t|| dntj| stj|| dd|rt|dS)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)endswithrrrrsplitrMrNnamelistisdirrmakedirsr#readr"r7_addon_detailsgetAssertionErrorrexistsr rr(rrmtree) r+ZaddonrnZtmpdirZxpifilecompressed_filenamedatar9Z addon_detailsZaddon_idZextensions_dir addon_pathr r r r1s<         z!FirefoxProfile._install_extensionc Csddddd}dd}dd}dd }tj|s@td |zt|rt|d 8}d |kr||d W5QRWS|d }W5QRXntj |rtj |d }tj|rt |dd} || W5QRWSQRXt tj |d dd} | }W5QRXntd|Wn>tt fk rh} zt t| tdW5d} ~ XYnXzt|} || d} || d} | | dd}|s| dd}|jD]0}|j| d}||kr||||iq|ds8t|jjD]2}|j|}|j| dkr|d|jiqWn:tk rt} zt t| tdW5d} ~ XYnXt |dtr|d!dk|d<|dst d|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)rlrnr|versioncSs^|jj}d}t|jD]B}||j|krd||jkr||jddd}qZq|S)N:r?)ZdocumentElement attributesrangelengthitemr-r|rq)docurlr namespaceir r r get_namespace_idsz7FirefoxProfile._addon_details..get_namespace_idcSs6g}|jD]}|j|jkr ||jq d|S)z(Retrieve the text value of a given node.r) childNodesZnodeTypeZ TEXT_NODEappendr}rstrip)elementrcnoder r r get_texts   z/FirefoxProfile._addon_details..get_textcSsft|}z|ddd}Wn0tk rN|dddd|d}YnX||d|dd d S) z[Extracts the details from the contents of a WebExtensions `manifest.json` file.Z applicationsZgeckorlr| r@rF)rlrr|rn)r&r`KeyErrorreplace)contentmanifestrlr r r parse_manifest_jsons "z:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: riz 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# DescriptionrrrlrntruezAdd-on id could not be found.)"rrryOSErrorrM is_zipfilerNrrrursrr"rrstrsysexc_infor parseStringZgetElementsByTagNamerrnodeNamerupdaterwrrrr|r-rbr@lower)r+r~detailsrrrr{rZmanifest_json_filenamer9eremrdf descriptionrentryr attributer r r rvsZ      (       ( zFirefoxProfile._addon_details)N)N)T)rr r r!r.r0rr3r:propertyrr=setterrErKrr[r6r1rvr r r r r+sB         .r)rRr(r&rr]rrrrcrMiorxml.domrZtyping_extensionsrZselenium.common.exceptionsrr%rbrrr r r r s"