B
    CVvgLM                 @   s  d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZmZ d d	lmZmZ d d
lmZ d dlmZ d dlmZ d dlmZmZ dd dD Z dd e ! D Z"edd#dd e"D  Z$G dd de%Z&dd Z'dd Z(dd Z)dd Z*d d! Z+d"d# Z,d$d% Z-G d&d' d'eZ.d(d) Z/dEd*d+Z0dFd,d-Z1d.d/ Z2dGd1d2Z3d3d4 Z4d5d6 Z5dHd7d8Z6G d9d: d:e%Z7d;d< Z8d=d> Z9d?d@ Z:dAdB Z;dCdD Z<dS )I    N)defaultdict)FieldDoesNotExist)modelsrouter)
LOOKUP_SEP)	Collector)pretty_name)NoReverseMatchreverse)formatstimezone)format_html)_lazy_re_compile)capfirst)ngettextoverridec             C   s   i | ]}d | |qS )z_%02X ).0ir   r   >/tmp/pip-install-o3oxmrkh/Django/django/contrib/admin/utils.py
<dictcomp>   s    r   s   ":/_#?;@&=+$,"[]<>%
\c             C   s   i | ]\}}t ||qS r   )chr)r   kvr   r   r   r      s    z_(?:%s)|c             C   s   g | ]}|d d qS )   Nr   )r   xr   r   r   
<listcomp>   s    r   c               @   s   e Zd ZdZdS )FieldIsAForeignKeyColumnNamez/A field is a foreign key attname, i.e. <FK>_id.N)__name__
__module____qualname____doc__r   r   r   r   r      s   r   c          	   C   s   | t}xt|D ]l}|dkr$| jj}y| |}W n tk
rH   wY qX t|dr| }|d j} t	dd |D rdS qW dS )zT
    Return True if 'distinct()' should be used to query the given lookup path.
    pkget_path_infoc             s   s   | ]}|j V  qd S )N)Zm2m)r   pathr   r   r   	<genexpr>/   s    z(lookup_needs_distinct.<locals>.<genexpr>TF)
splitr   r#   name	get_fieldr   hasattrr$   to_optsany)optsZlookup_pathZlookup_fields
field_namefieldZ	path_infor   r   r   lookup_needs_distinct   s    



r1   c             C   s0   |  dr|d}n|  dr,| dk}|S )zJ
    Return a lookup value prepared to be used in queryset filtering.
    Z__in,Z__isnull) false0)endswithr(   lower)keyvaluer   r   r   prepare_lookup_value5   s
    

r:   c             C   s   t | tr| tS | S )a*  
    Ensure that primary key values do not confuse the admin URLs by escaping
    any '/', '_' and ':' and similarly problematic characters.
    Similar to urllib.parse.quote(), except that the quoting is slightly
    different so that it doesn't get automatically unquoted by the Web browser.
    )
isinstancestr	translate	QUOTE_MAP)sr   r   r   quoteB   s    r@   c             C   s   t dd | S )zUndo the effects of quote().c             S   s   t | d  S )Nr   )UNQUOTE_MAP)mr   r   r   <lambda>N       zunquote.<locals>.<lambda>)
UNQUOTE_REsub)r?   r   r   r   unquoteL   s    rG   c             C   s:   g }x0| D ](}t |ttfr(|| q
|| q
W |S )zS
    Return a list which is a single level of flattening of the original list.
    )r;   listtupleextendappend)fieldsZflatr0   r   r   r   flattenQ   s    
rM   c             C   s,   g }x"| D ]\}}| t|d  q
W |S )z?Return a list of field names from an admin fieldsets structure.rL   )rJ   rM   )Z	fieldsetsfield_namesr)   r.   r   r   r   flatten_fieldsets^   s
    rO   c       	         s   y| d }W n t k
r*   g i t g fS X t|jj}t|d}||  t  fdd|}fdd|j	D }dd |j
 D }|||fS )	a  
    Find all objects related to ``objs`` that should also be deleted. ``objs``
    must be a homogeneous iterable of objects (e.g. a QuerySet).

    Return a nested list of strings suitable for display in the
    template with the ``unordered_list`` filter.
    r   )usingc                s   | j }| jk}| j}dt|j| f }|r j| | sJ|j y(td j|j	|j
f d t| jf}W n tk
r   |S X tdt|j|| S |S d S )Nz%s: %sz%s:%s_%s_changez{}: <a href="{}">{}</a>)	__class__	_registry_metar   verbose_nameZhas_delete_permissionaddr
   r)   	app_label
model_namer@   r#   r	   r   )objmodelZ	has_adminr.   Zno_edit_linkZ	admin_url)
admin_siteperms_neededrequestr   r   format_callbackz   s(    
z,get_deleted_objects.<locals>.format_callbackc                s   g | ]} |qS r   r   )r   rX   )r]   r   r   r      s    z'get_deleted_objects.<locals>.<listcomp>c             S   s   i | ]\}}t ||jjqS r   )lenrS   verbose_name_plural)r   rY   objsr   r   r   r      s    z'get_deleted_objects.<locals>.<dictcomp>)
IndexErrorsetr   Zdb_for_writerS   rY   NestedObjectscollectnested	protected
model_objsitems)	r`   r\   rZ   rX   rP   Z	collectorZ	to_deleterf   Zmodel_countr   )rZ   r]   r[   r\   r   get_deleted_objectsh   s    


ri   c                   sX   e Zd Z fddZdd Zd fdd	Z fdd	Zd
d ZdddZdd Z	  Z
S )rc   c                s*   t  j|| i | _t | _tt| _d S )N)super__init__edgesrb   rf   r   rg   )selfargskwargs)rQ   r   r   rk      s    zNestedObjects.__init__c             C   s   | j |g | d S )N)rl   
setdefaultrK   )rm   sourcetargetr   r   r   add_edge   s    zNestedObjects.add_edgeNc          
      s   xd|D ]\}|rB| dsB||jj|jjd }| t||| n| d | | j|jj | qW yt	 j
|fd|i|S  tjk
r } z| j|j W d d }~X Y n2 tjk
r } z| j|j W d d }~X Y nX d S )N+)classrV   source_attr)r6   rS   rW   rV   rs   getattrrg   rY   rU   rj   rd   r   ZProtectedErrorrf   updateZprotected_objectsZRestrictedErrorZrestricted_objects)rm   r`   rq   rv   ro   rX   related_namee)rQ   r   r   rd      s    
zNestedObjects.collectc                s$   t  |||}|jdd |D  S )Nc             S   s   g | ]
}|j qS r   )r)   )r   Zrelated_fieldr   r   r   r      s    z1NestedObjects.related_objects.<locals>.<listcomp>)rj   related_objectsZselect_related)rm   related_modelZrelated_fieldsr`   qs)rQ   r   r   r{      s    zNestedObjects.related_objectsc             C   sn   ||krg S | | g }x*| j|dD ]}|| ||| q*W |rV||g}n|g}|rj|| |S )Nr   )rU   rl   getrJ   _nestedrK   )rm   rX   seenr]   childrenchildretr   r   r   r      s    

zNestedObjects._nestedc             C   s:   t  }g }x*| jddD ]}|| ||| qW |S )z4
        Return the graph as a nested list.
        Nr   )rb   rl   r~   rJ   r   )rm   r]   r   rootsrootr   r   r   re      s
    zNestedObjects.nestedc             O   s   dS )z
        We always want to load the objects into memory so that we can display
        them to the user in confirm page.
        Fr   )rm   rn   ro   r   r   r   can_fast_delete   s    zNestedObjects.can_fast_delete)NN)N)r   r    r!   rk   rs   rd   r{   r   re   r   __classcell__r   r   )rQ   r   rc      s   

rc   c             C   sF   t | tjtjjfr| j}nt | tjjr4| jj}n| }|j	|j
dS )z
    Return a `dict` with keys 'verbose_name' and 'verbose_name_plural',
    typically for use with string formatting.

    `obj` may be a `Model` instance, `Model` subclass, or `QuerySet` instance.
    )rT   r_   )r;   r   ZModelbaseZ	ModelBaserS   queryQuerySetrY   rT   r_   )rX   r.   r   r   r   model_format_dict   s    
r   c             C   sN   t | tjjr$|dkr|  }| j} t| }|d |d  }}t|||pJdS )a-  
    Return the appropriate `verbose_name` or `verbose_name_plural` value for
    `obj` depending on the count `n`.

    `obj` may be a `Model` instance, `Model` subclass, or `QuerySet` instance.
    If `obj` is a `QuerySet` instance, `n` is optional and the length of the
    `QuerySet` is used.
    NrT   r_   r   )r;   r   r   r   countrY   r   r   )rX   ndZsingularpluralr   r   r   model_ngettext   s    	r   c          	   C   s   |j }yt|| }W nv ttfk
r   t| r>| }||}nDt|| rd| dkrdt|| }||}nt|| }t|r~| }n|}d }Y nX d }t|| }|||fS )N__str__)rS   _get_non_gfk_fieldr   r   callabler+   rw   )r)   rX   model_adminr.   fattrr9   r   r   r   lookup_field  s$    





r   c             C   sR   |  |}|jr(|jr|jr"|jr(t |jrN|jsNt|drN|j|krNt	 |S )a  
    For historical reasons, the admin app relies on GenericForeignKeys as being
    "not found" by get_field(). This could likely be cleaned up.

    Reverse relations should also be excluded as these aren't attributes of the
    model (rather something like `foo_set`).
    attname)
r*   is_relationZmany_to_oner|   Zone_to_manyr   Zmany_to_manyr+   r   r   )r.   r)   r0   r   r   r   r      s    
 r   Fc       	      C   s  d}y<t |j| }y
|j}W n tk
r:   |jjj}Y nX W nZ tk
rz   | dkrnt|jj}t}nt| r|| }nt|| rt	|| }npt|| rt	|| }nZ|r| |j
kr|j
|  }n@d| |jjf }|r|d|jj 7 }|r|d|jj 7 }t|t|dr|j}n`t|trFt|drFt|jdrF|jj}n0t|rn|jdkrbd}n
t|j}nt| }Y n" tk
r   t| }| }Y nX |r||fS |S dS )	ak  
    Return a sensible label for a field name. The name can be a callable,
    property (but not created with @property decorator), or the name of an
    object's attribute, as well as a model field. If return_attr is True, also
    return the resolved attribute (which could be a callable). This will be
    None if (and only if) the name refers to a field.
    Nr   zUnable to lookup '%s' on %sz or %sshort_descriptionfgetz<lambda>z--)r   rS   rT   AttributeErrorr|   r   r<   r   r+   rw   rL   Zobject_namerQ   r   r   r;   propertyr   r   r   )	r)   rY   r   Zreturn_attrformr   r0   labelmessager   r   r   label_for_field5  sT    





r   c          	   C   sB   d}yt |j| }W n ttfk
r,   Y nX t|dr>|j}|S )Nr3   	help_text)r   rS   r   r   r+   r   )r)   rY   r   r0   r   r   r   help_text_for_fieldq  s    
r   c             C   s*  ddl m} t|dd r*t|j| |S t|tjr>|| S | d krJ|S t|tj	rft
t| S t|tjtjfrt
| S t|tjrt
| |jS t|tjtjfrt
| S t|tjr| rtd| j| S t|tjr| rytj| d|jdS  tk
r   t| |S X n
t| |S d S )Nr   )_boolean_iconflatchoicesz<a href="{}">{}</a>F)ensure_asciicls),django.contrib.admin.templatetags.admin_listr   rw   dictr   r~   r;   r   ZBooleanFieldZDateTimeFieldr   localizer   template_localtimeZ	DateFieldZ	TimeFieldZDecimalFieldnumber_formatZdecimal_placesZIntegerFieldZ
FloatFieldZ	FileFieldr   urlZ	JSONFieldjsondumpsencoder	TypeErrordisplay_for_value)r9   r0   empty_value_displayr   r   r   r   display_for_field}  s.    

r   c             C   s   ddl m} |r|| S | d kr$|S t| tr6t| S t| tjrRtt	| S t| tj
tjfrnt| S t| ttjtfrt| S t| ttfrddd | D S t| S d S )Nr   )r   z, c             s   s   | ]}t |V  qd S )N)r<   )r   r   r   r   r   r'     s    z$display_for_value.<locals>.<genexpr>)r   r   r;   boolr<   datetimer   r   r   r   datetimeintdecimalDecimalfloatr   rH   rI   join)r9   r   booleanr   r   r   r   r     s     


r   c               @   s   e Zd ZdS )NotRelationFieldN)r   r    r!   r   r   r   r   r     s   r   c             C   s"   t | dr|  d jjS td S )Nr$   r%   )r+   r$   r,   rY   r   )r0   r   r   r   get_model_from_relation  s    
r   c          	   C   s   g }| }| t}x|D ]}|j|}t|t|d kr`yt| W n tk
r^   P Y nX |jr|jrr|j	r|
 }|jj}n|jj}|j}|d| qW |t|fS )z Create a reversed field path.

    E.g. Given (Order, "user__groups"),
    return (Group, "user__order").

    Final field must be a related model, not a data field.
    r   r   )r(   r   rS   r*   r^   r   r   r   Zauto_createdZconcreteZrelated_query_nameZremote_fieldrY   r0   r)   r|   insertr   )rY   r&   Zreversed_pathparentpiecespiecer0   ry   r   r   r   reverse_field_path  s"    


r   c             C   sH   | t}g }x4|D ],}|r*t|d }n| }||j| qW |S )a;   Return list of Fields given path relative to model.

    e.g. (ModelX, "user__groups__name") -> [
        <django.db.models.fields.related.ForeignKey object at 0x...>,
        <django.db.models.fields.related.ManyToManyField object at 0x...>,
        <django.db.models.fields.CharField object at 0x...>,
    ]
    r%   )r(   r   r   rK   rS   r*   )rY   r&   r   rL   r   r   r   r   r   get_fields_from_path  s    	

r   c             C   s"  | j }td t| |}W dQ R X g }|r<|di i n| j rT|dd|ii |rtd x|D ]}x.|jD ]$}|dt|jjt|di qvW x@|jD ]6\}}	|dt|jjt|t|j	d |	di qW x.|j
D ]$}
|dt|
jjt|
di qW qjW W dQ R X |S )	z
    Construct a JSON structure describing changes from a changed object.
    Translations are deactivated so that strings are stored untranslated.
    Translation happens later on LogEntry access.
    NaddedchangedrL   )r)   objectr   )r)   r   rL   Zdeleted)changed_datatranslation_override#_get_changed_field_labels_from_formrK   Znew_objectsr<   rS   rT   Zchanged_objectsZformsZdeleted_objects)r   ZformsetsrU   r   changed_field_labelsZchange_messageZformsetZadded_objectZchanged_objectZchanged_fieldsZdeleted_objectr   r   r   construct_change_message  s8    	





"r   c          	   C   sR   g }xH|D ]@}y| j | jp|}W n tk
r:   |}Y nX |t| q
W |S )N)rL   r   KeyErrorrK   r<   )r   r   r   r/   Zverbose_field_namer   r   r   r     s    

r   )N)N)NFN)F)=r   r   r   collectionsr   Zdjango.core.exceptionsr   Z	django.dbr   r   Zdjango.db.models.constantsr   Zdjango.db.models.deletionr   Zdjango.forms.utilsr   Zdjango.urlsr	   r
   Zdjango.utilsr   r   Zdjango.utils.htmlr   Zdjango.utils.regex_helperr   Zdjango.utils.textr   Zdjango.utils.translationr   r   r   r>   rh   rA   r   rE   	Exceptionr   r1   r:   r@   rG   rM   rO   ri   rc   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   <module>   sP   

8A


<
1