U
    Æ½„bb(  ã                   @   s°   d Z ddlmZ ddlmZ ddlmZ ddlm	Z	 dd„ Z
dd	„ ZG d
d„ dƒZG dd„ dƒZG dd„ dƒZG dd„ dƒZG dd„ deƒZG dd„ deƒZG dd„ deƒZdS )aK  
We perform uniqueness checks explicitly on the serializer class, rather
the using Django's `.full_clean()`.

This gives us better separation of concerns, allows us to use single-step
object creation, and makes it possible to switch between using the implicit
`ModelSerializer` class and an equivalent explicit `Serializer` class.
é    )Ú	DataError)Úgettext_lazy)ÚValidationError)Ú
smart_reprc              
   C   s,   z
|   ¡ W S  tttfk
r&   Y dS X d S )NF)ÚexistsÚ	TypeErrorÚ
ValueErrorr   )Úqueryset© r
   ú=/tmp/pip-unpacked-wheel-iohwuy7z/rest_framework/validators.pyÚ	qs_exists   s    
r   c              
   K   s6   z| j f |ŽW S  tttfk
r0   |  ¡  Y S X d S ©N)Úfilterr   r   r   Únone)r	   Úkwargsr
   r
   r   Ú	qs_filter   s    r   c                   @   sF   e Zd ZdZedƒZdZddd„Zdd	„ Zd
d„ Z	dd„ Z
dd„ ZdS )ÚUniqueValidatorzˆ
    Validator that corresponds to `unique=True` on a model field.

    Should be applied to an individual field on the serializer.
    zThis field must be unique.TNÚexactc                 C   s   || _ |p| j| _|| _d S r   )r	   ÚmessageÚlookup)Úselfr	   r   r   r
   r
   r   Ú__init__*   s    zUniqueValidator.__init__c                 C   s   d|| j f |i}t|f|ŽS )zT
        Filter the queryset to all instances matching the given attribute.
        z%s__%s)r   r   )r   Úvaluer	   Ú
field_nameÚfilter_kwargsr
   r
   r   Úfilter_queryset/   s    zUniqueValidator.filter_querysetc                 C   s   |dk	r|j |jd�S |S ©z}
        If an instance is being updated, then do not include
        that instance itself as a uniqueness conflict.
        N)Úpk©Úexcluder   )r   r	   Úinstancer
   r
   r   Úexclude_current_instance6   s    z(UniqueValidator.exclude_current_instancec                 C   sR   |j d }t|jdd ƒ}| j}|  |||¡}|  ||¡}t|ƒrNt| jdd�‚d S )Néÿÿÿÿr    Úunique©Úcode)	Úsource_attrsÚgetattrÚparentr	   r   r!   r   r   r   )r   r   Zserializer_fieldr   r    r	   r
   r
   r   Ú__call__?   s    
zUniqueValidator.__call__c                 C   s   d| j jt| jƒf S )Nz<%s(queryset=%s)>)Ú	__class__Ú__name__r   r	   ©r   r
   r
   r   Ú__repr__L   s    þzUniqueValidator.__repr__)Nr   )r+   Ú
__module__Ú__qualname__Ú__doc__Ú_r   Úrequires_contextr   r   r!   r)   r-   r
   r
   r
   r   r   !   s   
	r   c                   @   sV   e Zd ZdZedƒZedƒZdZddd„Zdd	„ Z	d
d„ Z
dd„ Zdd„ Zdd„ ZdS )ÚUniqueTogetherValidatorzŸ
    Validator that corresponds to `unique_together = (...)` on a model class.

    Should be applied to the serializer class, not to an individual field.
    z0The fields {field_names} must make a unique set.úThis field is required.TNc                 C   s   || _ || _|p| j| _d S r   )r	   Úfieldsr   )r   r	   r5   r   r
   r
   r   r   ]   s    z UniqueTogetherValidator.__init__c                    s:   ˆj dk	rdS ‡ ‡‡fdd„ˆjD ƒ}|r6t|dd�‚dS )z~
        The `UniqueTogetherValidator` always forces an implied 'required'
        state on the fields it applies to.
        Nc                    s$   i | ]}ˆj | jˆ kr|ˆj“qS r
   )r5   ÚsourceÚmissing_message©Ú.0r   ©Úattrsr   Ú
serializerr
   r   Ú
<dictcomp>j   s   þ zCUniqueTogetherValidator.enforce_required_fields.<locals>.<dictcomp>Úrequiredr$   )r    r5   r   )r   r;   r<   Úmissing_itemsr
   r:   r   Úenforce_required_fieldsb   s    
þz/UniqueTogetherValidator.enforce_required_fieldsc                    s^   ‡fdd„| j D ƒ}ˆjdk	r@|D ]}|ˆ kr"tˆj|ƒˆ |< q"‡ fdd„|D ƒ}t|f|ŽS )zU
        Filter the queryset to all instances matching the given attributes.
        c                    s   g | ]}ˆ j | j‘qS r
   )r5   r6   r8   )r<   r
   r   Ú
<listcomp>w   s   ÿz;UniqueTogetherValidator.filter_queryset.<locals>.<listcomp>Nc                    s   i | ]}|ˆ | “qS r
   r
   )r9   r6   )r;   r
   r   r=   „   s   ÿ z;UniqueTogetherValidator.filter_queryset.<locals>.<dictcomp>)r5   r    r'   r   )r   r;   r	   r<   Úsourcesr6   r   r
   )r;   r<   r   r   r   s    
þ

þz'UniqueTogetherValidator.filter_querysetc                 C   s   |dk	r|j |jd�S |S r   r   ©r   r;   r	   r    r
   r
   r   r!   Š   s    z0UniqueTogetherValidator.exclude_current_instancec                    s€   ˆ   ||¡ ˆ j}ˆ  |||¡}ˆ  |||j¡}‡ fdd„| ¡ D ƒ}d |kr|t|ƒr|d ˆ j¡}ˆ j	j
|d�}t|dd�‚d S )Nc                    s   g | ]\}}|ˆ j kr|‘qS r
   )r5   )r9   Úfieldr   r,   r
   r   rA   š   s    
 z4UniqueTogetherValidator.__call__.<locals>.<listcomp>z, )Úfield_namesr#   r$   )r@   r	   r   r!   r    Úitemsr   Újoinr5   r   Úformatr   )r   r;   r<   r	   Zchecked_valuesrE   r   r
   r,   r   r)   “   s    
ÿz UniqueTogetherValidator.__call__c                 C   s   d| j jt| jƒt| jƒf S )Nz<%s(queryset=%s, fields=%s)>)r*   r+   r   r	   r5   r,   r
   r
   r   r-   ¢   s
    ýz UniqueTogetherValidator.__repr__)N)r+   r.   r/   r0   r1   r   r7   r2   r   r@   r   r!   r)   r-   r
   r
   r
   r   r3   S   s   
	r3   c                   @   s    e Zd ZedƒZdZdd„ ZdS )Ú$ProhibitSurrogateCharactersValidatorz7Surrogate characters are not allowed: U+{code_point:X}.Z surrogate_characters_not_allowedc                 C   s<   dd„ t |ƒD ƒD ]$}| jjt|ƒd�}t|| jd�‚qd S )Nc                 s   s.   | ]&}d t |ƒ  krdkrn q|V  qdS )i Ø  iÿß  N)Úord)r9   Úchr
   r
   r   Ú	<genexpr>¯   s
      ÿz@ProhibitSurrogateCharactersValidator.__call__.<locals>.<genexpr>)Ú
code_pointr$   )Ústrr   rH   rJ   r   r%   )r   r   Zsurrogate_characterr   r
   r
   r   r)   ®   s    z-ProhibitSurrogateCharactersValidator.__call__N)r+   r.   r/   r1   r   r%   r)   r
   r
   r
   r   rI   ª   s   rI   c                   @   sN   e Zd ZdZedƒZdZddd„Zdd„ Zdd	„ Z	d
d„ Z
dd„ Zdd„ ZdS )ÚBaseUniqueForValidatorNr4   Tc                 C   s"   || _ || _|| _|p| j| _d S r   )r	   rD   Ú
date_fieldr   )r   r	   rD   rP   r   r
   r
   r   r   º   s    zBaseUniqueForValidator.__init__c                    s0   ‡ ‡fdd„ˆj ˆjfD ƒ}|r,t|dd�‚dS )z�
        The `UniqueFor<Range>Validator` classes always force an implied
        'required' state on the fields they are applied to.
        c                    s   i | ]}|ˆ kr|ˆj “qS r
   )r7   r8   ©r;   r   r
   r   r=   Å   s   þ zBBaseUniqueForValidator.enforce_required_fields.<locals>.<dictcomp>r>   r$   N)rD   rP   r   )r   r;   r?   r
   rQ   r   r@   À   s
    
þz.BaseUniqueForValidator.enforce_required_fieldsc                 C   s   t dƒ‚d S )Nz&`filter_queryset` must be implemented.)ÚNotImplementedError)r   r;   r	   r   Údate_field_namer
   r
   r   r   Í   s    z&BaseUniqueForValidator.filter_querysetc                 C   s   |dk	r|j |jd�S |S r   r   rC   r
   r
   r   r!   Ð   s    z/BaseUniqueForValidator.exclude_current_instancec                 C   s‚   |j | j jd }|j | j jd }|  |¡ | j}|  ||||¡}|  |||j¡}t	|ƒr~| j
j| jd�}t| j|idd�‚d S )Nr"   )rP   r#   r$   )r5   rD   r&   rP   r@   r	   r   r!   r    r   r   rH   r   )r   r;   r<   r   rS   r	   r   r
   r
   r   r)   Ù   s    
 ÿþzBaseUniqueForValidator.__call__c                 C   s&   d| j jt| jƒt| jƒt| jƒf S )Nz*<%s(queryset=%s, field=%s, date_field=%s)>)r*   r+   r   r	   rD   rP   r,   r
   r
   r   r-   é   s    üzBaseUniqueForValidator.__repr__)N)r+   r.   r/   r   r1   r7   r2   r   r@   r   r!   r)   r-   r
   r
   r
   r   rO   µ   s   
	rO   c                   @   s   e Zd ZedƒZdd„ ZdS )ÚUniqueForDateValidatorz6This field must be unique for the "{date_field}" date.c                 C   sV   || j  }|| j }i }|||< |j|d| < |j|d| < |j|d| < t|f|ŽS )Nz%s__dayú	%s__monthú%s__year)rD   rP   ÚdayÚmonthÚyearr   ©r   r;   r	   r   rS   r   Údater   r
   r
   r   r   õ   s    

z&UniqueForDateValidator.filter_querysetN©r+   r.   r/   r1   r   r   r
   r
   r
   r   rT   ò   s   rT   c                   @   s   e Zd ZedƒZdd„ ZdS )ÚUniqueForMonthValidatorz7This field must be unique for the "{date_field}" month.c                 C   s:   || j  }|| j }i }|||< |j|d| < t|f|ŽS )NrU   )rD   rP   rX   r   rZ   r
   r
   r   r     s    

z'UniqueForMonthValidator.filter_querysetNr\   r
   r
   r
   r   r]     s   r]   c                   @   s   e Zd ZedƒZdd„ ZdS )ÚUniqueForYearValidatorz6This field must be unique for the "{date_field}" year.c                 C   s:   || j  }|| j }i }|||< |j|d| < t|f|ŽS )NrV   )rD   rP   rY   r   rZ   r
   r
   r   r     s    

z&UniqueForYearValidator.filter_querysetNr\   r
   r
   r
   r   r^     s   r^   N)r0   Z	django.dbr   Zdjango.utils.translationr   r1   Zrest_framework.exceptionsr   Z#rest_framework.utils.representationr   r   r   r   r3   rI   rO   rT   r]   r^   r
   r
   r
   r   Ú<module>   s   2W=