a 5j5@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 PE:\soccer\ml_env\Lib\site-packages\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)%FirefoxProfileNcCsi|_|rLtjtd}tj||tdddd||_ t |j dnt|_ t j st tjtjttdd}t|t _ Wd n1s0Ytt 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-8encodingNZmutablefrozen)_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|dddB}|j D]&\}}| d|dt |dqLWd n1s0Yd 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 Csnt|tstdz$t|}|dks,|dkr4tdWnttfyTtdYn0||_|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|dtjddb}t|jd}t |jD]6\}}}|D]&}tj ||}| |||dqRqDWdn1s0Yt |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 encodeds6zFirefoxProfile.encodedc Cstd}t|ddx}|D]b}||}z t|d|j|d<Wqty|t d|d|dYq0qWdn1s0YdS)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_patternr9usrmatchesr r r r6s    z$FirefoxProfile._read_existing_userjsTc Csd}d}|dr tjdtj|dd}t|d}|D]}|dr~tj tj ||s|t tj ||qDtj tj tj ||st tj tj ||| |}ttj ||d}||WdqD1s0YqD|}|}||} | d } | s2Jd |tj |jd } tj | | } |s| d s|rtj| st | 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_detailsgetrexistsr rr(rrmtree) r+ZaddonroZtmpdirZxpifilecompressed_filenamedatar9Z addon_detailsZaddon_idZextensions_dir addon_pathr r r r1s<     *    z!FirefoxProfile._install_extensionc Csddddd}dd}dd}dd }tj|s@td |z&t|rt|d D}d |vr||d WdWS|d }Wdn1s0Yntj |rXtj |d }tj|rt |dd} || WdWS1s 0Yt tj |d dd} | }Wdn1sL0Yntd|Wn@tt fy} z"t t| tdWYd} ~ n d} ~ 00zt|} || d} || d} | | dd}|s| dd}|jD]0}|j| d}||vr||||iq|dsxt|jjD]2}|j|}|j| dkrD|d|jiqDWn<ty} z"t t| tdWYd} ~ n d} ~ 00t |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)rmror|versioncSs^|jj}d}t|jD]B}||j|krd||jvr||jddd}qZq|S)N:r?)ZdocumentElement attributesrangelengthitemr-r|rr)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_textcSsdt|}z|ddd}Wn.tyL|dddd|d}Yn0||d|dd d S) z[Extracts the details from the contents of a WebExtensions `manifest.json` file.Z applicationsZgeckormr| r@rF)rmrr|ro)r&r`KeyErrorreplace)contentmanifestrmr r r parse_manifest_jsons  "z:FirefoxProfile._addon_details..parse_manifest_jsonzAdd-on path does not exist: rjz 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# DescriptionrrrmrotruezAdd-on id could not be found.)"rrryOSErrorrM is_zipfilerNrsrvrtrr"rrstrsysexc_inforZ parseStringZgetElementsByTagNamerrZnodeNamerupdaterxrrrr|r-rbr@lower)r+r~detailsrrrr{rZmanifest_json_filenamer9erZemZrdf descriptionrentryr attributer r r rwsZ    *.*,       , zFirefoxProfile._addon_details)N)N)T)rr r r!r.r0rr3r:propertyrr=setterrErKrr[r6r1rwr r r r r+sB         .r)rRr(r&rr]rrrrcrMiorZxml.domrZtyping_extensionsrZselenium.common.exceptionsrr%rbrrr r r r s"