U
|
\ ±d<$ ã @ s6 d dl mZ d dl mZ d dlmZ G dd dZdS )é )ÚNoSuchElementException)ÚUnexpectedTagNameException)ÚByc @ s¸ e Zd ZddddZedd Zedd Zed d
|
Zd d Zd d Z dd Z
|
ddddZ dd Z dd Z dd ZddddZdddd Zeed d d Zeed d d!ZdS )"ÚSelectN)Úreturnc C sB |j ¡ dkr td|j || _| j d¡}|o:|dk| _dS )a| Constructor. A check is made that the given element is, indeed, a
|
SELECT tag. If it is not, then an UnexpectedTagNameException is thrown.
|
|
:Args:
|
- webelement - SELECT element to wrap
|
|
Example:
|
from selenium.webdriver.support.ui import Select
|
|
Select(driver.find_element(By.TAG_NAME, "select")).select_by_index(2)
|
Úselectz/Select only works on <select> elements, not on ÚmultipleÚfalseN)Ztag_nameÚlowerr Ú_elZget_dom_attributeÚ is_multiple)ÚselfZ
|
webelementZmulti© r úXd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\selenium/webdriver/support/select.pyÚ__init__ s
|
zSelect.__init__c C s | j tjd¡S )z;Returns a list of all options belonging to this select tag.Úoption)r Ú find_elementsr ZTAG_NAME©r r r r Úoptions) s zSelect.optionsc C s dd | j D S )zLReturns a list of all selected options belonging to this select
|
tag.c S s g | ]}| ¡ r|qS r )Ú is_selected)Ú.0Úoptr r r Ú
|
<listcomp>2 s z/Select.all_selected_options.<locals>.<listcomp>)r r r r r Úall_selected_options. s zSelect.all_selected_optionsc C s( | j D ]}| ¡ r| S qtddS )zjThe first selected option in this select tag (or the currently
|
selected option in a normal select)zNo options are selectedN)r r r ©r r r r r Úfirst_selected_option4 s
|
|
z Select.first_selected_optionc C s` d| |¡ d}| j tj|¡}d}|D ] }| |¡ | jsD dS d}q*|s\td| dS )aL Select all options that have a value matching the argument. That is,
|
when given "foo" this would select an option like:
|
|
<option value="foo">Bar</option>
|
|
:Args:
|
- value - The value to match against
|
|
throws NoSuchElementException If there is no option with specified value in SELECT
|
zoption[value =ú]FNTz!Cannot locate option with value: )Ú_escape_stringr r r Ú CSS_SELECTORÚ _set_selectedr r )r ÚvalueÚcssÚoptsÚmatchedr r r r Úselect_by_value= s
|
zSelect.select_by_valuec C sD t |}| jD ]"}| d¡|kr| |¡ dS qtd| dS )a: Select the option at the given index. This is done by examining the
|
"index" attribute of an element, and not merely by counting.
|
|
:Args:
|
- index - The option at this index will be selected
|
|
throws NoSuchElementException If there is no option with specified index in SELECT
|
ÚindexNú$Could not locate element with index )Ústrr Ú get_attributer r )r r% Úmatchr r r r Úselect_by_indexS s
|
|
zSelect.select_by_indexc C sÞ d| |¡ d}| j tj|¡}d}|D ] }| |¡ | jsD dS d}q*t|dkrÈd|krÈ| |¡}|dkrx| j }n"d | |¡ d
|
}| j tj|¡}|D ](}||j
|
kr| |¡ | js dS d}q|sÚt d | dS ) aS Select all options that display text matching the argument. That is,
|
when given "Bar" this would select an option like:
|
|
<option value="foo">Bar</option>
|
|
:Args:
|
- text - The visible text to match against
|
|
throws NoSuchElementException If there is no option with specified text in SELECT
|
ú .//option[normalize-space(.) = r FNTr ú Ú z.//option[contains(.,z)]ú,Could not locate element with visible text: ) r r r r ÚXPATHr r ÚlenÚ_get_longest_tokenr Útextr ) r r2 Úxpathr" r# r Zsub_string_without_spaceÚ
|
candidatesÚ candidater r r Úselect_by_visible_textc s,
|
|
|
|
z Select.select_by_visible_textc C s( | j std| jD ]}| |¡ qdS )zÌClear all selected entries.
|
|
This is only valid when the SELECT supports multiple selections.
|
throws NotImplementedError If the SELECT does not support
|
multiple selections
|
z3You may only deselect all options of a multi-selectN)r ÚNotImplementedErrorr Ú_unset_selectedr r r r Ú deselect_all s
|
zSelect.deselect_allc C sb | j stdd}d| |¡ d}| j tj|¡}|D ]}| |¡ d}q8|s^td| dS )aR Deselect all options that have a value matching the argument. That
|
is, when given "foo" this would deselect an option like:
|
|
<option value="foo">Bar</option>
|
|
:Args:
|
- value - The value to match against
|
|
throws NoSuchElementException If there is no option with specified value in SELECT
|
ú/You may only deselect options of a multi-selectFzoption[value = r Tz%Could not locate element with value: N) r r7 r r r r r r8 r )r r r# r! r" r r r r Údeselect_by_value s
|
zSelect.deselect_by_valuec C sN | j std| jD ]&}| d¡t|kr| |¡ dS qtd| dS )a? Deselect the option at the given index. This is done by examining
|
the "index" attribute of an element, and not merely by counting.
|
|
:Args:
|
- index - The option at this index will be deselected
|
|
throws NoSuchElementException If there is no option with specified index in SELECT
|
r: r% Nr& )r r7 r r( r' r8 r )r r% r r r r Údeselect_by_indexª s
|
|
zSelect.deselect_by_indexc C sb | j stdd}d| |¡ d}| j tj|¡}|D ]}| |¡ d}q8|s^td| dS )zúDeselect all options that display text matching the argument. That
|
is, when given "Bar" this would deselect an option like:
|
|
<option value="foo">Bar</option>
|
|
:Args:
|
- text - The visible text to match against
|
r: Fr+ r Tr. N) r r7 r r r r r/ r8 r )r r2 r# r3 r" r r r r Údeselect_by_visible_text» s
|
z Select.deselect_by_visible_textc C s$ | ¡ s | ¡ std| ¡ d S )Nz$You may not select a disabled option)r Ú
|
is_enabledr7 Úclick©r r r r r r Ï s zSelect._set_selectedc C s | ¡ r| ¡ d S )N)r r? r@ r r r r8 Õ s zSelect._unset_selected)r r c C s d|krtd|krt| d¡}dg}|D ] }| d| d¡ | d¡ q$|dd
}| d¡rf| d¡ d |¡d S d|krd| dS d| dS )
|
Nú"ú'zconcat(z, '"', r éÿÿÿÿz, '"'r- ú))ÚsplitÚappendÚendswithÚjoin)r r Z
|
substringsÚresultZ substringr r r r Ù s
|
|
|
zSelect._escape_stringc C s0 | d¡}d}|D ]}t|t|kr|}q|S )Nr, r- )rE r0 )r r ÚitemsÚlongestÚitemr r r r1 ê s
|
zSelect._get_longest_token)Ú__name__Ú
|
__module__Ú __qualname__r Úpropertyr r r r$ r* r6 r9 r; r< r= r r8 r' r r1 r r r r r s$
|
|
|
% r N)Zselenium.common.exceptionsr r Z selenium.webdriver.common.byr r r r r r Ú<module> s
|
