U
|
\ ±d 8 ã @ s¢ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
|
m Z d dl m Z d dlmZ dZdZdZG dd d eZG d
|
d d ZdS ) é N)ÚBytesIO)Úminidom)ÚWebDriverExceptionz webdriver.xpizwebdriver_prefs.jsonzfxdriver@googlecode.comc @ s e Zd ZdZdS )ÚAddonFormatErrorz4Exception for not well-formed add-on manifest files.N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__© r
|
r
|
úad:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\selenium/webdriver/firefox/firefox_profile.pyr & s r c @ sâ e Zd ZdZdZd$ddZdd ZefddZd d
|
Z e
|
d d Z e
|
d d Z e j ddddZ e
|
dd Zej ddddZe
|
dd Zej ddddZe
|
edddZdd Zdd Zd d Zd%d d!Zd"d# ZdS )&ÚFirefoxProfileZWEBDRIVER_ANONYMOUS_PROFILENc C s" t jdtdd tjsLttj tj t
|
¡t ¡dd}t |¡t_W 5 Q R X t tjd ¡| _|| _d| _| js|| ¡ | _n`t ¡ | _tj | jd¡}tj| j|t d d
|
d ¡d || _t | jd ¡ | tj | jd¡¡ tj | jd¡| _tj | jd¡| _ tj | j ¡r t | j d¡ 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é )Ú
|
stacklevelúutf-8©ÚencodingZmutableNzwebdriver-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__. s>
|
ÿ ÿ
|
ÿzFirefoxProfile.__init__c C s || j |< dS )z0sets the preference that we want in the profile.N©r% ©r2 ÚkeyÚvaluer
|
r
|
r Úset_preferenceU s z FirefoxProfile.set_preferencec C s | |¡ d S )N)Ú_install_extension)r2 Ú extensionr
|
r
|
r Ú add_extensionY s z FirefoxProfile.add_extensionc C s< t jd ¡ D ] \}}|| jkr|| j|< q| | j¡ d S )NÚfrozen)r r Úitemsr% Ú_write_user_prefsr5 r
|
r
|
r Úupdate_preferences\ s
|
z!FirefoxProfile.update_preferencesc C s | j S )z8Gets the profile directory that is currently being used.)r& ©r2 r
|
r
|
r r g s zFirefoxProfile.pathc C s | j S )z+Gets the port that WebDriver is working on.)Ú_portr@ r
|
r
|
r Úportl s zFirefoxProfile.port)Úreturnc C sp t |tstdz$t|}|dk s,|dkr4tdW n ttfk
|
rV tdY nX || _| 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Ú TypeErrorrA r8 )r2 rB r
|
r
|
r rB q s
|
c C s
|
| j d S )NÚ webdriver_accept_untrusted_certsr4 r@ r
|
r
|
r Úaccept_untrusted_certs s z%FirefoxProfile.accept_untrusted_certsc C s |dkrt d| d|¡ d S )N©TFú%Please pass in a Boolean to this callrI ©r r8 ©r2 r7 r
|
r
|
r rJ s c C s
|
| j d S )NÚ!webdriver_assume_untrusted_issuerr4 r@ r
|
r
|
r Ú assume_untrusted_cert_issuer s z+FirefoxProfile.assume_untrusted_cert_issuerc C s |dkrt d| d|¡ d S )NrK rL rO rM rN r
|
r
|
r rP s c C s | ¡ t }t |dtj¡X}t| jd }t | j¡D ]6\}}}|D ]&}tj ||¡}|
|
|||d
¡ qHq:W 5 Q R X t | ¡ ¡ d¡S )znA zipped, base64 encoded string of profile directory for use with
|
remote WebDriver JSON wire protocol.ÚwrD NzUTF-8)r? r ÚzipfileÚZipFileÚ ZIP_DEFLATEDÚlenr r Úwalkr ÚwriteÚbase64Ú b64encodeÚgetvalueÚdecode) r2 ÚfpZzippedZ path_rootÚbaseÚ_ÚfilesZfyleÚfilenamer
|
r
|
r Úencoded s "zFirefoxProfile.encodedc C s t ¡ S )z9Creates a temp folder to store User.js and the extension.)r( r) r@ r
|
r
|
r r' ¢ s z!FirefoxProfile._create_tempfolderc
|
C sP t | jddd6}| ¡ D ]&\}}| d| dt |¡ d¡ qW 5 Q R X dS )z1writes the current user prefs dictionary to disk.rQ r r z user_pref("z", z);
|
N)r r0 r= rW r! Údumps)r2 Z
|
user_prefsÚfr6 r7 r
|
r
|
r r> ¦ s z FirefoxProfile._write_user_prefsc C sª t d¡}zt|ddp}|D ]d}| |¡}z t | d¡¡| j| d¡< W q tk
|
r t
|
d| d¡| d¡ ¡ Y q X q W 5 Q R X W n tk
|
r¤ Y nX d S )Nzuser_pref\("(.*)",\s(.*)\)r r r rD z5(skipping) failed to json.loads existing preference: ) ÚreÚcompiler Úsearchr! ÚloadsÚgroupr% Ú Exceptionr r )r2 ZuserjsZ pref_patternrc ZusrÚmatchesr
|
r
|
r r. ¬ s
|
|
ÿz$FirefoxProfile._read_existing_userjsTc
|
C sÐ |t kr tj tj t¡t ¡}d}d}| d¡rtjdtj |¡d d}t
|
|d¡}| ¡ D ]¨}| d¡rtj tj ||¡¡st tj ||¡¡ qbtj tj tj ||¡¡¡sÖt tj tj ||¡¡¡ | |¡}ttj ||¡d}| |¡ W 5 Q R X qb|}|}| |¡} | d ¡}
|
|
|
s<td
|
| tj | j|
|
¡} |s| d s|rtj | j¡st | j¡ t | jd ¡ t || d ¡ n tj | ¡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: Úunpackr T)Úsymlinks) Ú WEBDRIVER_EXTr r r r r Úendswithr( r) ÚsplitrR rS ÚnamelistÚisdirÚmakedirsÚreadr rW Ú_addon_detailsÚgetÚAssertionErrorr/ Úexistsr- r* r# r+ Úrmtree) r2 Zaddonrr ZtmpdirZxpifileÚcompressed_fileÚnameÚdatarc Z addon_detailsZaddon_idÚ
|
addon_pathr
|
r
|
r r9 ¼ s>
|
|
|
|
z!FirefoxProfile._install_extensionc
|
C s® ddddd}dd }dd }dd }t j |¡s@td
|
| zæt |¡rz:t |d ¡}d | ¡ krz|| d ¡W ¢W S | d ¡}W 5 | ¡ X nt j
|
|¡rt j |d ¡}t j |¡rêt |dd } || ¡ W 5 Q R £ W S Q R X t t j |d ¡dd} | ¡ }W 5 Q R X ntd| W n> tt fk
|
rd }
|
ztt|
|
t ¡ d W 5 d}
|
~
|
X Y nX zÐt |¡} || d} || d} | | d ¡ d¡}|s°| d¡ d¡}|jD ]0}|j | d¡}||kr¶| |||i¡ q¶| d¡s4t|j j D ]2}|j |¡}|j | d kr | d|j i¡ q W n: t k
|
rp }
|
ztt|
|
t ¡ d W 5 d}
|
~
|
X Y nX t!|d tr|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)rq rr r Úversionc S s^ | j j}d}t|jD ]B}| |¡j|krd| |¡jkr| |¡j d¡d d } qZq|S )NÚ ú:rD )ZdocumentElementÚ
|
attributesÚrangeÚlengthÚitemr7 r rv )ÚdocÚurlr Ú namespaceÚir
|
r
|
r Úget_namespace_idû s z7FirefoxProfile._addon_details.<locals>.get_namespace_idc S s6 g }| j D ] }|j|jkr
|
| |j¡ q
|
d |¡ ¡ S )z(Retrieve the text value of a given node.r
)Ú
|
childNodesZnodeTypeZ TEXT_NODEÚappendr r Ústrip)ÚelementÚrcÚnoder
|
r
|
r Úget_text s
|
|
z/FirefoxProfile._addon_details.<locals>.get_textc S sf t | ¡}z|d d d }W n0 tk
|
rN |d dd¡d |d }Y nX ||d |d d d
|
S ) z[Extracts the details from the contents of a WebExtensions
|
`manifest.json` file.Z applicationsZgeckorq r ú r
ú@r F)rq r r rr )r! rg ÚKeyErrorÚreplace)ÚcontentÚmanifestrq r
|
r
|
r Úparse_manifest_json s
|
"üz:FirefoxProfile._addon_details.<locals>.parse_manifest_jsonz Add-on path does not exist: rn z manifest.jsonz install.rdfr r z/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#Ú Descriptionr r
rq rr Útruez Add-on id could not be found.)#r r r~ ÚOSErrorrR Ú
|
is_zipfileÚcloserS rw rz rx r r r r ÚstrÚsysÚexc_infor Ú parseStringZgetElementsByTagNamer r ÚnodeNamer Úupdater| r r r r r7 ri rE Úlower)r2 r Údetailsr r r r r Zmanifest_json_filenamerc Úer ÚemÚrdfÚ descriptionr Úentryr Ú attributer
|
r
|
r r{ ì s^
|
(
|
|
|
|
|
( z FirefoxProfile._addon_details)N)T)r r r ZANONYMOUS_PROFILE_NAMEr r3 r8 rt r; r? Úpropertyr rB ÚsetterrJ rP r£ ra r' r> r. r9 r{ r
|
r
|
r
|
r r * s6
|
'
|
|
|
|
|
0r )rX r# r! r rd r* r¤ r( r rR Úior Úxml.domr Zselenium.common.exceptionsr rt r ZEXTENSION_NAMEri r r r
|
r
|
r
|
r Ú<module> s"
|
