U
    ꥡcA                     @   st  d Z ddlZddlZddlmZ ddlmZmZ ddlm	Z	m
Z
mZmZ ddlmZ z
ejZW n" ek
r   eedZY nX d	d
ddddddddddddddddgZdd Zdd Zedd Zedddd G d!d" d"Zd#d Zedddd$G d%d& d&ZdFd'dZedddd G d(d) d)Zd*d Zedddd G d+d, d,Zd-d Zedddd G d.d/ d/Zd0d Z edddd G d1d2 d2Z!d3d Z"edddd G d4d5 d5Z#dGd6d
Z$edddd G d7d8 d8Z%dHd9dZ&edddd$G d:d; d;Z'd<d Z(d=d Z)d>d Z*d?d Z+edddd$G d@dA dAZ,dBd Z-edddd$G dCdD dDZ.dEd Z/dS )Iz
Commonly useful validators.
    N)contextmanager   )get_run_validatorsset_run_validators)_AndValidatorand_attribattrs)NotCallableError r   deep_iterabledeep_mappingdisabledgeget_disabledgtin_instance_ofis_callablelelt
matches_remax_lenmin_lenoptionalprovidesset_disabledc                 C   s   t |   dS )a  
    Globally disable or enable running validators.

    By default, they are run.

    :param disabled: If ``True``, disable running all validators.
    :type disabled: bool

    .. warning::

        This function is not thread-safe!

    .. versionadded:: 21.3.0
    Nr   )r    r   3/tmp/pip-unpacked-wheel-9v97kvxj/attr/validators.pyr   .   s    c                   C   s   t   S )z
    Return a bool indicating whether validators are currently disabled or not.

    :return: ``True`` if validators are currently disabled.
    :rtype: bool

    .. versionadded:: 21.3.0
    )r   r   r   r   r   r   @   s    	c                   c   s"   t d z
dV  W 5 t d X dS )z
    Context manager that disables running validators within its context.

    .. warning::

        This context manager is not thread-safe!

    .. versionadded:: 21.3.0
    FTNr   r   r   r   r   r   L   s    
FT)reprslotshashc                   @   s"   e Zd Ze Zdd Zdd ZdS )_InstanceOfValidatorc                 C   s4   t || js0tdj|j| j|j|d|| j|dS )P
        We use a callable class to be able to change the ``__repr__``.
        z?'{name}' must be {type!r} (got {value!r} that is a {actual!r}).)nametypeactualvalueN)
isinstancer&   	TypeErrorformatr%   	__class__selfinstattrr(   r   r   r   __call__b   s    z_InstanceOfValidator.__call__c                 C   s   dj | jdS )Nz)<instance_of validator for type {type!r}>r&   )r+   r&   r.   r   r   r   __repr__t   s    z_InstanceOfValidator.__repr__N)__name__
__module____qualname__r   r&   r1   r4   r   r   r   r   r#   ^   s   r#   c                 C   s   t | S )a  
    A validator that raises a `TypeError` if the initializer is called
    with a wrong type for this particular attribute (checks are performed using
    `isinstance` therefore it's also valid to pass a tuple of types).

    :param type: The type to check for.
    :type type: type or tuple of types

    :raises TypeError: With a human readable error message, the attribute
        (of type `attrs.Attribute`), the expected type, and the value it
        got.
    )r#   r2   r   r   r   r   z   s    )r    frozenr!   c                   @   s(   e Zd Ze Ze Zdd Zdd ZdS )_MatchesReValidatorc                 C   s0   |  |s,tdj|j| jj|d|| j|dS )r$   z9'{name}' must match regex {pattern!r} ({value!r} doesn't))r%   patternr(   N)
match_func
ValueErrorr+   r%   r:   r-   r   r   r   r1      s    
  z_MatchesReValidator.__call__c                 C   s   dj | jdS )Nz.<matches_re validator for pattern {pattern!r}>)r:   )r+   r:   r3   r   r   r   r4      s    z_MatchesReValidator.__repr__N)r5   r6   r7   r   r:   r;   r1   r4   r   r   r   r   r9      s   r9   c              	   C   s   t jdt jt jf}||kr@tddtdd t|D t	| t
r\|rVtd| }nt | |}|t jkrz|j}n|t jkr|j}n|j}t||S )a  
    A validator that raises `ValueError` if the initializer is called
    with a string that doesn't match *regex*.

    :param regex: a regex string or precompiled pattern to match against
    :param int flags: flags that will be passed to the underlying re function
        (default 0)
    :param callable func: which underlying `re` function to call. Valid options
        are `re.fullmatch`, `re.search`, and `re.match`; the default ``None``
        means `re.fullmatch`. For performance reasons, the pattern is always
        precompiled using `re.compile`.

    .. versionadded:: 19.2.0
    .. versionchanged:: 21.3.0 *regex* can be a pre-compiled pattern.
    Nz'func' must be one of {}.z, c                 s   s   | ]}|r|j pd V  qdS )NoneN)r5   ).0er   r   r   	<genexpr>   s    zmatches_re.<locals>.<genexpr>zR'flags' can only be used with a string pattern; pass flags to re.compile() instead)re	fullmatchsearchmatchr<   r+   joinsortedsetr)   Patternr*   compiler9   )regexflagsfuncZvalid_funcsr:   r;   r   r   r   r      s0    



c                   @   s"   e Zd Ze Zdd Zdd ZdS )_ProvidesValidatorc                 C   s0   | j |s,tdj|j| j |d|| j |dS )r$   z<'{name}' must provide {interface!r} which {value!r} doesn't.)r%   	interfacer(   N)rN   Z
providedByr*   r+   r%   r-   r   r   r   r1      s      z_ProvidesValidator.__call__c                 C   s   dj | jdS )Nz0<provides validator for interface {interface!r}>rN   )r+   rN   r3   r   r   r   r4      s    z_ProvidesValidator.__repr__N)r5   r6   r7   r   rN   r1   r4   r   r   r   r   rM      s   rM   c                 C   s   t | S )a)  
    A validator that raises a `TypeError` if the initializer is called
    with an object that does not provide the requested *interface* (checks are
    performed using ``interface.providedBy(value)`` (see `zope.interface
    <https://zopeinterface.readthedocs.io/en/latest/>`_).

    :param interface: The interface to check for.
    :type interface: ``zope.interface.Interface``

    :raises TypeError: With a human readable error message, the attribute
        (of type `attrs.Attribute`), the expected interface, and the
        value it got.
    )rM   rO   r   r   r   r      s    c                   @   s"   e Zd Ze Zdd Zdd ZdS )_OptionalValidatorc                 C   s   |d krd S |  ||| d S )N	validatorr-   r   r   r   r1     s    z_OptionalValidator.__call__c                 C   s   dj t| jdS )Nz'<optional validator for {what} or None>)what)r+   r    rR   r3   r   r   r   r4     s    z_OptionalValidator.__repr__N)r5   r6   r7   r   rR   r1   r4   r   r   r   r   rP      s   rP   c                 C   s   t | trtt| S t| S )a  
    A validator that makes an attribute optional.  An optional attribute is one
    which can be set to ``None`` in addition to satisfying the requirements of
    the sub-validator.

    :param validator: A validator (or a list of validators) that is used for
        non-``None`` values.
    :type validator: callable or `list` of callables.

    .. versionadded:: 15.1.0
    .. versionchanged:: 17.1.0 *validator* can be a list of validators.
    )r)   listrP   r   rQ   r   r   r   r     s    
c                   @   s"   e Zd Ze Zdd Zdd ZdS )_InValidatorc                 C   sP   z|| j k}W n tk
r&   d}Y nX |sLtdj|j| j |d|| j |d S )NFz/'{name}' must be in {options!r} (got {value!r}))r%   optionsr(   )rV   r*   r<   r+   r%   )r.   r/   r0   r(   Z
in_optionsr   r   r   r1   $  s    
  z_InValidator.__call__c                 C   s   dj | jdS )Nz(<in_ validator with options {options!r}>rV   )r+   rV   r3   r   r   r   r4   4  s    z_InValidator.__repr__N)r5   r6   r7   r   rV   r1   r4   r   r   r   r   rU      s   rU   c                 C   s   t | S )a  
    A validator that raises a `ValueError` if the initializer is called
    with a value that does not belong in the options provided.  The check is
    performed using ``value in options``.

    :param options: Allowed options.
    :type options: list, tuple, `enum.Enum`, ...

    :raises ValueError: With a human readable error message, the attribute (of
       type `attrs.Attribute`), the expected options, and the value it
       got.

    .. versionadded:: 17.1.0
    .. versionchanged:: 22.1.0
       The ValueError was incomplete until now and only contained the human
       readable error message. Now it contains all the information that has
       been promised since 17.1.0.
    )rU   rW   r   r   r   r   :  s    c                   @   s   e Zd Zdd Zdd ZdS )_IsCallableValidatorc                 C   s,   t |s(d}t|j|j||jd|ddS )r$   z?'{name}' must be callable (got {value!r} that is a {actual!r}).)r%   r(   r'   )msgr(   N)callabler
   r+   r%   r,   )r.   r/   r0   r(   messager   r   r   r1   R  s      z_IsCallableValidator.__call__c                 C   s   dS )Nz<is_callable validator>r   r3   r   r   r   r4   b  s    z_IsCallableValidator.__repr__N)r5   r6   r7   r1   r4   r   r   r   r   rX   P  s   rX   c                   C   s   t  S )ax  
    A validator that raises a `attr.exceptions.NotCallableError` if the
    initializer is called with a value for this particular attribute
    that is not callable.

    .. versionadded:: 19.1.0

    :raises `attr.exceptions.NotCallableError`: With a human readable error
        message containing the attribute (`attrs.Attribute`) name,
        and the value it got.
    )rX   r   r   r   r   r   f  s    c                   @   s:   e Zd Zee dZedee dZdd Zdd Z	dS )_DeepIterablerQ   NdefaultrR   c                 C   s4   | j dk	r|  ||| |D ]}| ||| qdS r$   N)iterable_validatormember_validator)r.   r/   r0   r(   memberr   r   r   r1   |  s    
z_DeepIterable.__call__c                 C   s,   | j d krdndj| j d}dj|| jdS )Nr   z {iterable!r})iterablezJ<deep_iterable validator for{iterable_identifier} iterables of {member!r}>)iterable_identifierrb   )r`   r+   ra   )r.   rd   r   r   r   r4     s    z_DeepIterable.__repr__)
r5   r6   r7   r   r   ra   r   r`   r1   r4   r   r   r   r   r\   u  s    
r\   c                 C   s    t | ttfrt|  } t| |S )a4  
    A validator that performs deep validation of an iterable.

    :param member_validator: Validator(s) to apply to iterable members
    :param iterable_validator: Validator to apply to iterable itself
        (optional)

    .. versionadded:: 19.1.0

    :raises TypeError: if any sub-validators fail
    )r)   rT   tupler   r\   )ra   r`   r   r   r   r     s    c                   @   sF   e Zd Zee dZee dZedee dZdd Z	dd Z
dS )_DeepMappingrQ   Nr]   c                 C   sF   | j dk	r|  ||| |D ]$}| ||| | ||||  qdS r_   )mapping_validatorkey_validatorvalue_validator)r.   r/   r0   r(   keyr   r   r   r1     s
    
z_DeepMapping.__call__c                 C   s   dj | j| jdS )NzA<deep_mapping validator for objects mapping {key!r} to {value!r}>)rj   r(   )r+   rh   ri   r3   r   r   r   r4     s
     z_DeepMapping.__repr__)r5   r6   r7   r   r   rh   ri   r   rg   r1   r4   r   r   r   r   rf     s
   rf   c                 C   s   t | ||S )a}  
    A validator that performs deep validation of a dictionary.

    :param key_validator: Validator to apply to dictionary keys
    :param value_validator: Validator to apply to dictionary values
    :param mapping_validator: Validator to apply to top-level mapping
        attribute (optional)

    .. versionadded:: 19.1.0

    :raises TypeError: if any sub-validators fail
    )rf   )rh   ri   rg   r   r   r   r     s    c                   @   s.   e Zd Ze Ze Ze Zdd Zdd ZdS )_NumberValidatorc                 C   s.   |  || js*tdj|j| j| j|ddS )r$   z&'{name}' must be {op} {bound}: {value})r%   opboundr(   N)compare_funcrm   r<   r+   r%   
compare_opr-   r   r   r   r1     s    z_NumberValidator.__call__c                 C   s   dj | j| jdS )Nz<Validator for x {op} {bound}>)rl   rm   )r+   ro   rm   r3   r   r   r   r4     s     z_NumberValidator.__repr__N)	r5   r6   r7   r   rm   ro   rn   r1   r4   r   r   r   r   rk     s
   rk   c                 C   s   t | dtjS )z
    A validator that raises `ValueError` if the initializer is called
    with a number larger or equal to *val*.

    :param val: Exclusive upper bound for values

    .. versionadded:: 21.3.0
    <)rk   operatorr   valr   r   r   r     s    	c                 C   s   t | dtjS )z
    A validator that raises `ValueError` if the initializer is called
    with a number greater than *val*.

    :param val: Inclusive upper bound for values

    .. versionadded:: 21.3.0
    z<=)rk   rq   r   rr   r   r   r   r     s    	c                 C   s   t | dtjS )z
    A validator that raises `ValueError` if the initializer is called
    with a number smaller than *val*.

    :param val: Inclusive lower bound for values

    .. versionadded:: 21.3.0
    z>=)rk   rq   r   rr   r   r   r   r     s    	c                 C   s   t | dtjS )z
    A validator that raises `ValueError` if the initializer is called
    with a number smaller or equal to *val*.

    :param val: Exclusive lower bound for values

    .. versionadded:: 21.3.0
    >)rk   rq   r   rr   r   r   r   r     s    	c                   @   s"   e Zd Ze Zdd Zdd ZdS )_MaxLengthValidatorc                 C   s.   t || jkr*tdj|j| jt |ddS )r$   z*Length of '{name}' must be <= {max}: {len})r%   maxlenN)rw   
max_lengthr<   r+   r%   r-   r   r   r   r1     s      z_MaxLengthValidator.__call__c                 C   s   dj | jdS )Nz<max_len validator for {max}>)rv   )r+   rx   r3   r   r   r   r4   &  s    z_MaxLengthValidator.__repr__N)r5   r6   r7   r   rx   r1   r4   r   r   r   r   ru     s   ru   c                 C   s   t | S )z
    A validator that raises `ValueError` if the initializer is called
    with a string or iterable that is longer than *length*.

    :param int length: Maximum length of the string or iterable

    .. versionadded:: 21.3.0
    )ru   lengthr   r   r   r   *  s    	c                   @   s"   e Zd Ze Zdd Zdd ZdS )_MinLengthValidatorc                 C   s.   t || jk r*tdj|j| jt |ddS )r$   z*Length of '{name}' must be => {min}: {len})r%   minrw   N)rw   
min_lengthr<   r+   r%   r-   r   r   r   r1   :  s      z_MinLengthValidator.__call__c                 C   s   dj | jdS )Nz<min_len validator for {min}>)r|   )r+   r}   r3   r   r   r   r4   E  s    z_MinLengthValidator.__repr__N)r5   r6   r7   r   r}   r1   r4   r   r   r   r   r{   6  s   r{   c                 C   s   t | S )z
    A validator that raises `ValueError` if the initializer is called
    with a string or iterable that is shorter than *length*.

    :param int length: Minimum length of the string or iterable

    .. versionadded:: 22.1.0
    )r{   ry   r   r   r   r   I  s    	)r   N)N)N)0__doc__rq   rA   
contextlibr   _configr   r   _maker   r   r   r	   
exceptionsr
   rH   AttributeErrorr&   rI   __all__r   r   r   r#   r   r9   r   rM   r   rP   r   rU   r   rX   r   r\   r   rf   r   rk   r   r   r   r   ru   r   r{   r   r   r   r   r   <module>   s   


0

