o
    Vh                     @   s   U d dl Z d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm	Z	 d d	l
mZ d d
lmZ d dlmZ dZeed< efZejeje  ed< edeeef dZedZG dd dee ZdS )    N)Callable)Generic)Literal)TypeVar)Union)NoSuchElementException)TimeoutException)WaitExcTypes)	WebDriver)
WebElementg      ?POLL_FREQUENCYIGNORED_EXCEPTIONSD)boundTc                
   @   s   e Zd Zedfdedededeje fddZ	dd	 Z
ddeegeed ef f dedefddZddeegef dedeeed f fddZdS )WebDriverWaitNdrivertimeoutpoll_frequencyignored_exceptionsc                 C   sp   || _ t|| _|| _| jdkrt| _tt}|r1z	|t| W n t	y0   |
| Y nw t|| _dS )aV  Constructor, takes a WebDriver instance and timeout in seconds.

        :Args:
         - driver - Instance of WebDriver (Ie, Firefox, Chrome or Remote) or a WebElement
         - timeout - Number of seconds before timing out
         - poll_frequency - sleep interval between calls
           By default, it is 0.5 second.
         - ignored_exceptions - iterable structure of exception classes ignored during calls.
           By default, it contains NoSuchElementException only.

        Example::

         from selenium.webdriver.support.wait import WebDriverWait 

         element = WebDriverWait(driver, 10).until(lambda x: x.find_element(By.ID, "someId")) 

         is_disappeared = WebDriverWait(driver, 30, 1, (ElementNotVisibleException)).\ 

                     until_not(lambda x: x.find_element(By.ID, "someId").is_displayed())
        r   N)_driverfloat_timeout_pollr   listr   extenditer	TypeErrorappendtuple_ignored_exceptions)selfr   r   r   r   
exceptions r#   \/home/air/yokohama/back/venv/lib/python3.10/site-packages/selenium/webdriver/support/wait.py__init__(   s   

zWebDriverWait.__init__c                 C   s(   dt | j dt | j d| jj dS )N<.z (session="z")>)type
__module____name__r   
session_id)r!   r#   r#   r$   __repr__N   s   (zWebDriverWait.__repr__ methodFmessagereturnc              
   C   s   d}d}t  | j }	 z|| j}|r|W S W n | jy7 } zt|dd}t|dd}W Y d}~nd}~ww t | j t  |krEnqt|||)a  Calls the method provided with the driver as an argument until the         return value does not evaluate to ``False``.

        :param method: callable(WebDriver)
        :param message: optional message for :exc:`TimeoutException`
        :returns: the result of the last call to `method`
        :raises: :exc:`selenium.common.exceptions.TimeoutException` if timeout occurs
        NTscreen
stacktrace)	time	monotonicr   r   r    getattrsleepr   r   )r!   r.   r/   r1   r2   end_timevalueexcr#   r#   r$   untilQ   s&   	
zWebDriverWait.untilTc                 C   sd   t  | j }	 z|| j}|s|W S W n | jy   Y dS w t | j t  |kr1	 t|q)a  Calls the method provided with the driver as an argument until the         return value evaluates to ``False``.

        :param method: callable(WebDriver)
        :param message: optional message for :exc:`TimeoutException`
        :returns: the result of the last call to `method`, or
                  ``True`` if `method` has raised one of the ignored exceptions
        :raises: :exc:`selenium.common.exceptions.TimeoutException` if timeout occurs
        T)r3   r4   r   r   r    r6   r   r   )r!   r.   r/   r7   r8   r#   r#   r$   	until_notk   s   

zWebDriverWait.until_not)r-   )r*   r)   __qualname__r   r   r   typingOptionalr	   r%   r,   r   r   r   r   strr:   r;   r#   r#   r#   r$   r   '   s    
&.2r   )r3   r=   r   r   r   r   r   selenium.common.exceptionsr   r   selenium.typesr	   #selenium.webdriver.remote.webdriverr
   $selenium.webdriver.remote.webelementr   r   r   __annotations__r   TupleType	Exceptionr   r   r   r#   r#   r#   r$   <module>   s"   
