u^sK@sddlZddlZddlZddlmZddlmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZmZdd lmZmZdd lmZdd lmZdd lmZmZd ddDZddejDZ ej!ddj"dde DZ#Gddde$Z%ddZ&ddZ'ddZ(ddZ)dd Z*d!d"Z+d#d$Z,Gd%d&d&e Z-d'd(Z.dd)d*Z/dd+d,Z0d-d.Z1dd/dd0d1Z2d2d3Z3d4d5Z4d/d6d7Z5Gd8d9d9e$Z6d:d;Z7d<d=Z8d>d?Z9d@dAZ:dBdCZ;dS)DN) defaultdict)FieldDoesNotExist)modelsrouter) LOOKUP_SEP) Collector) pretty_name)NoReverseMatchreverse)formatstimezone) format_html)capfirst)ngettextoverridecCsi|]}d||qS)z_%02X).0irrs rs":/_#?;@&=+$,"[]<>% \cCs%i|]\}}t||qSr)chr)rkvrrrrs z_(?:%s)|cCs g|]}|ddqS)Nr)rxrrr s rc@seZdZdZdS)FieldIsAForeignKeyColumnNamez/A field is a foreign key attname, i.e. _id.N)__name__ __module__ __qualname____doc__rrrrrs rc Cs|jt}x|D]}|dkr4|jj}y|j|}Wntk r^wYqXt|dr|j}|dj}t dd|DrdSqWdS) zT Return True if 'distinct()' should be used to query the given lookup path. pk get_path_inforcss|]}|jVqdS)N)Zm2m)rpathrrr .sz(lookup_needs_distinct..TF) splitrr"name get_fieldrhasattrr#to_optsany)optsZ lookup_pathZ lookup_fields field_namefieldZ path_inforrrlookup_needs_distincts      r0cCsF|jdr!|jd}n!|jdrB|jdk}|S)zJ Return a lookup value prepared to be used in queryset filtering. Z__in,Z__isnullfalse0)r2zfalser4)endswithr'lower)keyvaluerrrprepare_lookup_value4s r9cCs t|tr|jtS|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. ) isinstancestr translate QUOTE_MAP)srrrquoteAsr?cCstjdd|S)zUndo the effects of quote().cSst|jdS)Nr) UNQUOTE_MAPgroup)mrrrMszunquote..) UNQUOTE_REsub)r>rrrunquoteKsrFcCsMg}x@|D]8}t|ttfr8|j|q |j|q W|S)zS Return a list which is a single level of flattening of the original list. )r:listtupleextendappend)fieldsZflatr/rrrflattenPs  rLcCs8g}x+|D]#\}}|jt|dq W|S)z?Return a list of field names from an admin fieldsets structure.rK)rIrL)Z fieldsets field_namesr(r-rrrflatten_fieldsets]s rNc sy|d}Wn%tk r5gitgfSYnXtj|jj}td|}|j|tfdd|j}fdd|j D}dd|j j D}|||fS) 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. rusingc s|j}|jk}|j}dt|j|f}|rj|j|sij|jy8tdj|j |j fdt |j f}Wnt k r|SYnXtdt|j||S|SdS)Nz%s: %sz%s:%s_%s_changez{}: {}) __class__ _registry_metar verbose_nameZhas_delete_permissionaddr r( app_label model_namer?r"r r )objmodelZ has_adminr-Z no_edit_linkZ admin_url) admin_site perms_neededrequestrrformat_callbackys(      z,get_deleted_objects..format_callbackcsg|]}|qSrr)rrW)r\rrrs z'get_deleted_objects..cSs+i|]!\}}t||jjqSr)lenrRverbose_name_plural)rrXobjsrrrrs z'get_deleted_objects..) IndexErrorsetrZ db_for_writerRrX NestedObjectscollectnested protected model_objsitems) r_r[rYrWrOZ collectorZ to_deletereZ model_countr)rYr\rZr[rget_deleted_objectsgs   rhcseZdZfddZddZddfddZfdd Zd d Zdd d ZddZ S)rbcs;tj||i|_t|_tt|_dS)N)super__init__edgesrarerrf)selfargskwargs)rPrrrjs  zNestedObjects.__init__cCs |jj|gj|dS)N)rk setdefaultrJ)rlsourcetargetrrradd_edgeszNestedObjects.add_edgeNc sx|D]}|ra|jd ra|d|jjd|jji}|jt|||n|jd||j|jjj|qWyt j |d||SWn;t j k r}z|j j|jWYdd}~XnXdS)N+classrU source_attr)r5rRrVrUrrgetattrrfrXrTrircrZProtectedErrorreupdateZprotected_objects)rlr_rprurnrW related_namee)rPrrrcs  zNestedObjects.collectcs(tj||}|j|jjS)N)rirelated_objectsZselect_relatedr/r()rlrelatedr_qs)rPrrrzszNestedObjects.related_objectscCs||krgS|j|g}x9|jj|fD]"}|j|j|||q9W|rw||g}n |g}|r|j||S)N)rTrkgetrI_nestedrJ)rlrWseenr\childrenchildretrrrr~s     zNestedObjects._nestedcCsOt}g}x9|jjdfD]"}|j|j|||q%W|S)z4 Return the graph as a nested list. N)rarkr}rIr~)rlr\rrootsrootrrrrds   zNestedObjects.nestedcOsdS)z We always want to load the objects into memory so that we can display them to the user in confirm page. Fr)rlrmrnrrrcan_fast_deleteszNestedObjects.can_fast_delete) rrr rjrrrcrzr~rdrrr)rPrrbs    rbcCsjt|tjtjjfr*|j}n*t|tjjrN|jj}n|}d|j d|j iS)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. rSr^) r:rZModelbaseZ ModelBaserRqueryQuerySetrXrSr^)rWr-rrrmodel_format_dicts  rcCsmt|tjjr6|dkr-|j}|j}t|}|d|d}}t|||pidS)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. NrSr^r)r:rrrcountrXrr)rWndZsingularpluralrrrmodel_ngettexts     rc Cs|j}yt||}Wnttfk rt|rS|}||}nft||r|dkrt||}||}n-t||}t|r|}n|}d}YnXd}t||}|||fS)N__str__)rR_get_non_gfk_fieldrrcallabler*rv)r(rW model_adminr-fattrr8rrr lookup_fields$     rcCs{|j|}|jr=|jr+|j s4|jr=t|jrw|j rwt|drw|j|krwt |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_one related_modelZ one_to_manyrZ many_to_manyr*rr)r-r(r/rrrrs  1 rFc Cs5d}yGt|j|}y |j}Wn!tk rK|jjj}YnXWntk r|dkrt|jj}t}nqt|r|}nt||rt ||}nt||rt ||}n~|r||j kr|j |}nYd||jj f}|r3|d|j j f7}|rM|d|j j 7}t|t|drt|j}nt|trt|drt|jdr|jj}nBt|r|j dkrd}qt|j }n t|}Yn$tk rt|}|}YnX|r-||fS|SdS) 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. NrzUnable to lookup '%s' on %sz or %sshort_descriptionfgetzz--)rrRrSAttributeErrorrrr;rr*rvrKZ object_namerPrrr:propertyrrr) r(rXrZ return_attrformrr/labelmessagerrrlabel_for_field2sT               rc CsSd}yt|j|}Wnttfk r6YnXt|drO|j}|S)Nr2 help_text)rrRrrr*r)r(rXrr/rrrhelp_text_for_fieldns rcCs@ddlm}t|ddr;t|jj||St|tjrW||S|dkrg|St|tj rt j t j |St|tjtjfrt j |St|tjrt j||jSt|tjtjfrt j|St|tjr/|r/td|j|St||SdS)Nr) _boolean_icon flatchoicesz{}),django.contrib.admin.templatetags.admin_listrrvdictrr}r:rZ BooleanFieldZ DateTimeFieldr localizer template_localtimeZ DateFieldZ TimeFieldZ DecimalField number_formatZdecimal_placesZ IntegerFieldZ FloatFieldZ FileFieldr urldisplay_for_value)r8r/empty_value_displayrrrrdisplay_for_fieldzs$    rcCsddlm}|r ||S|dkr0|St|trIt|St|tjrqtjtj |St|tj tj frtj|St|t t jtfrtj|St|ttfrdjdd|DSt|SdS)Nr)rz, css|]}t|VqdS)N)r;)rrrrrr%sz$display_for_value..)rrr:boolr;datetimer rr rdatetimeintdecimalZDecimalfloatrrGrHjoin)r8rbooleanrrrrrs      rc@seZdZdS)NotRelationFieldN)rrr rrrrrs rcCs-t|dr#|jdjjStdS)Nr#rr&)r*r#r+rXr)r/rrrget_model_from_relationsrc Csg}|}|jt}x|D]}|jj|}t|t|dkrzyt|Wntk ryPYnX|jr|jo|j r|j }|j j }n|j j}|j}|jd|q"W|tj|fS)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. rr)r'rrRr)r]rrrZ auto_createdZconcreteZrelated_query_nameZ remote_fieldrXr/r(rinsertr)rXr$Z reversed_pathparentpiecespiecer/rxrrrreverse_field_paths"     rcCsb|jt}g}xF|D]>}|r;t|d}n|}|j|jj|qW|S)a; Return list of Fields given path relative to model. e.g. (ModelX, "user__groups__name") -> [ , , , ] rr&)r'rrrJrRr))rXr$rrKrrrrrget_fields_from_paths  rc Cs|j}tdt||}WdQRXg}|rN|jdiin"|jrp|jdd|ii|rtdx|D]}xB|jD]7}|jddt|jjdt|iiqWx^|jD]S\}} |jddt|jjdt|dt|j d| iiqWxB|j D]7} |jddt| jjdt| iiq@WqWWdQRX|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. NaddedchangedrKr(objectrZdeleted) changed_datatranslation_override#_get_changed_field_labels_from_formrJZ new_objectsr;rRrSZchanged_objectsZformsZdeleted_objects) rZformsetsrTrchanged_field_labelsZchange_messageZformsetZ added_objectZchanged_objectZchanged_fieldsZdeleted_objectrrrconstruct_change_messages8      $%rc Cscg}xV|D]N}y|j|jp)|}Wntk rG|}YnX|jt|q W|S)N)rKrKeyErrorrJr;)rrrr.Zverbose_field_namerrrrs   r)sN   )   8?  <     1