U
    ‰dü,  ã                   @   s¦   d 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 ejejejejfZG dd	„ d	ƒZd
d„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ Zdd„ ZdS )z”
Helper functions for mapping model fields to a dictionary of default
keyword arguments that should be used for their equivalent serializer fields.
é    N)Ú
validators)Úmodels)Úcapfirst)Úpostgres_fields)ÚUniqueValidatorc                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	ÚClassLookupDicta
  
    Takes a dictionary with classes as keys.
    Lookups against this object will traverses the object's inheritance
    hierarchy in method resolution order, and returns the first matching value
    from the dictionary or raises a KeyError if nothing matches.
    c                 C   s
   || _ d S ©N©Úmapping)Úselfr
   © r   úF/tmp/pip-unpacked-wheel-quko9yh2/rest_framework/utils/field_mapping.pyÚ__init__   s    zClassLookupDict.__init__c                 C   sR   t |dƒr|j}n|j}t |¡D ]}|| jkr"| j|   S q"td|j ƒ‚d S )NÚ_proxy_classzClass %s not found in lookup.)Úhasattrr   Ú	__class__ÚinspectÚgetmror
   ÚKeyErrorÚ__name__)r   ÚkeyZ
base_classÚclsr   r   r   Ú__getitem__   s    

zClassLookupDict.__getitem__c                 C   s   || j |< d S r   r	   )r   r   Úvaluer   r   r   Ú__setitem__*   s    zClassLookupDict.__setitem__N)r   Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r      s   r   c                 C   s   |  dd¡ ¡ }t| jƒ|kS )z•
    Returns `True` if the label based on the model's verbose name
    is not equal to the default label it would have based on it's field name.
    Ú_ú )ÚreplaceÚ
capitalizer   Úverbose_name)Úmodel_fieldÚ
field_nameZdefault_labelr   r   r   Úneeds_label.   s    r%   c                 C   s   dd| j j ¡ i S )zz
    Given a model class, return the view name to use for URL relationships
    that refer to instances of the model.
    z%(model_name)s-detailÚ
model_name)Ú_metaZobject_nameÚlower)Úmodelr   r   r   Úget_detail_view_name7   s     
ÿr*   c                 C   sÄ  i }t |jƒ}||d< |jr4t|| ƒr4t|jƒ|d< |jrD|j|d< t|ddƒ}|dk	r`||d< t|ddƒ}|dk	r|||d< t|tj	ƒr’|j
|d< t|tjƒr¤|jrÊtr´t|tjƒsÊttdƒrÖt|tjƒrÖd	d
i|d< |jräd|d< t|tjƒsø|jsd|d< |S | ¡ s|js|jr&d|d< |jrJt|tjtjfƒrJd|d< |jsntrnt|tjƒrnd|d< t|tjƒrÞ|j|d< |jdk	rœ|j|d< |jdk	r²|j|d< |jdk	rÈ|j|d< |jdk	rÞ|j|d< |jrô|j|d< ntdd„ |D ƒdƒ}|dk	r4t|tƒr4||d< dd„ |D ƒ}tdd„ |D ƒdƒ}|dk	rtt|tƒrt||d< d d„ |D ƒ}t|tjƒrd!d„ |D ƒ}t|tj ƒr¬d"d„ |D ƒ}t|tj	ƒrÈd#d„ |D ƒ}t|tj!ƒräd$d„ |D ƒ}t|tj"ƒr d%d„ |D ƒ}t|d&dƒ}|dk	rDt|tjtjtj#fƒrD||d&< d'd„ |D ƒ}td(d„ |D ƒdƒ}	|	dk	r†t|tjƒr†|	|d)< d*d„ |D ƒ}t|d+dƒr²t$|j%j&t'|ƒd,}
| (|
¡ |rÀ||d-< |S ).zE
    Creates a default instance of a basic non-relational field.
    r#   ÚlabelÚ	help_textÚ
max_digitsNÚdecimal_placesÚallow_unicodeÚ	JSONFieldZbase_templateztextarea.htmlÚstyleTÚ
allow_nullÚ	read_onlyFÚrequiredZallow_blankÚallow_emptyÚpathÚmatchÚ	recursiveÚallow_filesÚallow_foldersÚchoicesc                 s   s    | ]}t |tjƒr|jV  qd S r   )Ú
isinstancer   ÚMaxValueValidatorÚlimit_value©Ú.0Ú	validatorr   r   r   Ú	<genexpr>ˆ   s   ÿz#get_field_kwargs.<locals>.<genexpr>Ú	max_valuec                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   r=   r?   r   r   r   Ú
<listcomp>Ž   s   ÿz$get_field_kwargs.<locals>.<listcomp>c                 s   s    | ]}t |tjƒr|jV  qd S r   )r<   r   ÚMinValueValidatorr>   r?   r   r   r   rB   •   s   ÿÚ	min_valuec                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   rE   r?   r   r   r   rD   ›   s   ÿc                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   ZURLValidatorr?   r   r   r   rD   £   s   ÿc                 S   s   g | ]}|t jk	r|‘qS r   )r   Zvalidate_emailr?   r   r   r   rD   «   s   
ÿc                 S   s   g | ]}|t jk	r|‘qS r   )r   Zvalidate_slugr?   r   r   r   rD   ²   s   
ÿc                 S   s   g | ]}|t jk	r|‘qS r   )r   Zvalidate_ipv46_addressr?   r   r   r   rD   ¹   s   
ÿc                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   ZDecimalValidatorr?   r   r   r   rD   ¿   s   ÿÚ
max_lengthc                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   ZMaxLengthValidatorr?   r   r   r   rD   É   s   ÿc                 s   s    | ]}t |tjƒr|jV  qd S r   )r<   r   ÚMinLengthValidatorr>   r?   r   r   r   rB   Ð   s   ÿÚ
min_lengthc                 S   s   g | ]}t |tjƒs|‘qS r   )r<   r   rH   r?   r   r   r   rD   Ö   s   ÿÚunique©ÚquerysetÚmessager   ))Úlistr   r"   r%   r   r,   Úgetattrr<   r   Z	SlugFieldr/   Z	TextFieldr;   r   r0   r   ÚnullZ	AutoFieldÚeditableÚhas_defaultÚblankZ	CharFieldZ
ArrayFieldZFilePathFieldr6   r7   r8   r9   r:   ÚnextÚNUMERIC_FIELD_TYPESZURLFieldZ
EmailFieldZGenericIPAddressFieldÚDecimalFieldZ	FileFieldr   r)   Ú_default_managerÚget_unique_error_messageÚappend)r$   r#   ÚkwargsZvalidator_kwargr-   r.   rC   rF   rG   rI   rA   r   r   r   Úget_field_kwargsA   sâ    


ÿ
ÿþ
þ




ÿýÿÿýÿÿÿÿÿÿ"ÿÿýÿþ
r[   c                 C   s~  |\}}}}}}|j t|ƒdœ}|r,d|d< |r8||d< |oB| ¡ }	|	rrt|	tjƒs`tjf |	Ž}	|d  |	¡|d< |rŠd|d< | dd¡ |rz|jr®t	|| ƒr®t
|jƒ|d< |j}
|
rÀ|
|d	< |jsÚd|d< | dd¡ |jrèd|d
< | dd¡rø|S | ¡ s|js|jrd|d< |jr,|j|d< t|ddƒrdt|jj t|ƒd}| dg ¡|g |d< |rz|jszd|d< |S )z@
    Creates a default instance of a flat relational field.
    )rL   Ú	view_nameTÚmanyÚto_fieldrL   r3   Nr+   r,   r2   Fr4   r   rJ   rK   r5   )rW   r*   Zget_limit_choices_tor<   r   ÚQÚfilterÚpopr"   r%   r   r,   rQ   rP   ÚgetrR   rS   r   rO   r   r)   rX   )r$   Úrelation_infor#   Zrelated_modelÚto_manyr^   Zhas_through_modelÚreverserZ   Zlimit_choices_tor,   rA   r   r   r   Úget_relation_kwargsç   sT    þ
þrf   c                 C   s   ddi}| j rd|d< |S )Nr3   Tr]   )rd   )rc   rZ   r   r   r   Úget_nested_relation_kwargs   s    rg   c                 C   s   dt | ƒiS )Nr\   )r*   )r#   r   r   r   Úget_url_kwargs'  s     ÿrh   c                 C   s,   | j  dd ¡}|r(|| jjj| jdœ }|S )NrJ   )r&   Zfield_label)Zerror_messagesrb   r)   r'   r"   )r#   Zunique_error_messager   r   r   rX   -  s    þrX   )r   r   Zdjango.corer   Z	django.dbr   Zdjango.utils.textr   Zrest_framework.compatr   Zrest_framework.validatorsr   ZIntegerFieldZ
FloatFieldrV   ZDurationFieldrU   r   r%   r*   r[   rf   rg   rh   rX   r   r   r   r   Ú<module>   s(      ÿ	
 '9