U
    g                     @  s6  d dl mZ d dlZd dlZd dlmZ d dlmZmZ d dl	Z	erPd dl
mZ dddd	d
dddZdddd	d
dddZddddZdddddZdddddZG dd deZedddd	dddd Zedddd	ddd!d"Zd#ejkr2d dlZeefD ]Zejeje	jd$e_qdS )%    )annotationsN)contextmanager)TYPE_CHECKINGNoReturn)	GeneratorFshieldfloatboolztrio.CancelScope)deadliner   returnc                C  s   t j| |dS )a>  Use as a context manager to create a cancel scope with the given
    absolute deadline.

    Args:
      deadline (float): The deadline.
      shield (bool): Initial value for the `~trio.CancelScope.shield` attribute
          of the newly created cancel scope.

    Raises:
      ValueError: if deadline is NaN.

    r   r   )trioCancelScoper    r   2/tmp/pip-unpacked-wheel-ks04xdmi/trio/_timeouts.py
move_on_at   s    r   )secondsr   r   c                C  s0   | dk rt dt| r"t dtj|| dS )a  Use as a context manager to create a cancel scope whose deadline is
    set to now + *seconds*.

    The deadline of the cancel scope is calculated upon entering.

    Args:
      seconds (float): The timeout.
      shield (bool): Initial value for the `~trio.CancelScope.shield` attribute
          of the newly created cancel scope.

    Raises:
      ValueError: if ``seconds`` is less than zero or NaN.

    r   `seconds` must be non-negativez`seconds` must not be NaN)r   Zrelative_deadline)
ValueErrormathisnanr   r   )r   r   r   r   r   move_on_after   s    
r   r   )r   c                     s"   t jdd I dH  tddS )z}Pause execution of the current task forever (or until cancelled).

    Equivalent to calling ``await sleep(math.inf)``.

    c                 S  s
   t jjjS )N)r   lowlevelZAbortZ	SUCCEEDED)_r   r   r   <lambda>C       zsleep_forever.<locals>.<lambda>Nz#Should never have been rescheduled!)r   r   Zwait_task_rescheduledRuntimeErrorr   r   r   r   sleep_forever=   s    r   None)r   r   c              	     s$   t |  t I dH  W 5 Q R X dS )a  Pause execution of the current task until the given time.

    The difference between :func:`sleep` and :func:`sleep_until` is that the
    former takes a relative time and the latter takes an absolute time
    according to Trio's internal clock (as returned by :func:`current_time`).

    Args:
        deadline (float): The time at which we should wake up again. May be in
            the past, in which case this function executes a checkpoint but
            does not block.

    Raises:
      ValueError: if deadline is NaN.

    N)r   r   )r   r   r   r   sleep_untilG   s    
r    )r   r   c                   sD   | dk rt d| dkr*tj I dH  ntt |  I dH  dS )a  Pause execution of the current task for the given number of seconds.

    Args:
        seconds (float): The number of seconds to sleep. May be zero to
            insert a checkpoint without actually blocking.

    Raises:
        ValueError: if *seconds* is negative or NaN.

    r   r   N)r   r   r   
checkpointr    current_time)r   r   r   r   sleep[   s
    r#   c                   @  s   e Zd ZdZdS )TooSlowErrorzRRaised by :func:`fail_after` and :func:`fail_at` if the timeout
    expires.

    N)__name__
__module____qualname____doc__r   r   r   r   r$   n   s   r$   z'Generator[trio.CancelScope, None, None]c             	   c  s,   t | |d}|V  W 5 Q R X |jr(tdS )a  Creates a cancel scope with the given deadline, and raises an error if it
    is actually cancelled.

    This function and :func:`move_on_at` are similar in that both create a
    cancel scope with a given absolute deadline, and if the deadline expires
    then both will cause :exc:`Cancelled` to be raised within the scope. The
    difference is that when the :exc:`Cancelled` exception reaches
    :func:`move_on_at`, it's caught and discarded. When it reaches
    :func:`fail_at`, then it's caught and :exc:`TooSlowError` is raised in its
    place.

    Args:
      deadline (float): The deadline.
      shield (bool): Initial value for the `~trio.CancelScope.shield` attribute
          of the newly created cancel scope.

    Raises:
      TooSlowError: if a :exc:`Cancelled` exception is raised in this scope
        and caught by the context manager.
      ValueError: if deadline is NaN.

    r   N)r   cancelled_caughtr$   )r   r   scoper   r   r   fail_atu   s    r+   c             	   c  s,   t | |d}|V  W 5 Q R X |jr(tdS )a  Creates a cancel scope with the given timeout, and raises an error if
    it is actually cancelled.

    This function and :func:`move_on_after` are similar in that both create a
    cancel scope with a given timeout, and if the timeout expires then both
    will cause :exc:`Cancelled` to be raised within the scope. The difference
    is that when the :exc:`Cancelled` exception reaches :func:`move_on_after`,
    it's caught and discarded. When it reaches :func:`fail_after`, then it's
    caught and :exc:`TooSlowError` is raised in its place.

    The deadline of the cancel scope is calculated upon entering.

    Args:
      seconds (float): The timeout.
      shield (bool): Initial value for the `~trio.CancelScope.shield` attribute
          of the newly created cancel scope.

    Raises:
      TooSlowError: if a :exc:`Cancelled` exception is raised in this scope
        and caught by the context manager.
      ValueError: if *seconds* is less than zero or NaN.

    r   N)r   r)   r$   )r   r   r*   r   r   r   
fail_after   s    r,   Zsphinx)return_annotation)
__future__r   r   sys
contextlibr   typingr   r   r   collections.abcr   r   r   r   r    r#   	Exceptionr$   r+   r,   modulesinspectc	Signaturefrom_callablereplacer   __signature__r   r   r   r   <module>   s0   
!%