B CVvg0@sdZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl m Z ddlmZedd ZGd d d eZGd d d eedZddZGddde jZGdddZGdddZd ddZddZddZGdddZdS)!z Various data structures used in query construction. Factored out from django.db.models.query to avoid making the main module very large and/or so that they can be used by other modules without getting into circular import difficulties. N) namedtuple)FieldDoesNotExist FieldError) LOOKUP_SEP)tree)RemovedInDjango40WarningPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationcs<eZdZeddZddZfddZfddZZS) InvalidQueryTypecCsttfS)N)rr)selfr @/tmp/pip-install-o3oxmrkh/Django/django/db/models/query_utils.py _subclassesszInvalidQueryType._subclassescCstjdtdddS)Nz\The InvalidQuery exception class is deprecated. Use FieldDoesNotExist or FieldError instead.)category stacklevel)warningswarnr)r r r r Z__warnszInvalidQueryType.__warncs |t||jpt|S)N)_InvalidQueryType__warn isinstancer super__instancecheck__)r instance) __class__r r r&sz"InvalidQueryType.__instancecheck__cs |t||jpt|S)N)r issubclassr r__subclasscheck__)r subclass)rr r r*sz"InvalidQueryType.__subclasscheck__) __name__ __module__ __qualname__propertyr rrr __classcell__r r )rr r s  r c@s eZdZdS) InvalidQueryN)rrrr r r r r!/sr!) metaclassccs*|Vx|D]}t|EdHqWdS)N)__subclasses__ subclasses)clsrr r r r$3sr$csjeZdZdZdZdZeZdZdddfdd Zd d Z d d Z ddZ ddZ dddZ ddZZS)Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). ANDORTNF) _connector_negatedcs"tj|t|||ddS)N)children connectornegated)r__init__sorteditems)r r)r*argskwargs)rr r r.Dsz Q.__init__cCst|ts"t|dddks"t||sBt|dr8|St|St|trl|sl|\}}}t|||St|}||_| ||| |||S)N conditionalFTcopy) rr&getattr TypeErrorhasattrr4 deconstructtyper,add)r otherconn_r1r2objr r r _combineGs   z Q._combinecCs|||jS)N)r?r()r r;r r r __or__WszQ.__or__cCs|||jS)N)r?r')r r;r r r __and__Zsz Q.__and__cCs$t|}|||j||S)N)r9r:r'negate)r r>r r r __invert__]s z Q.__invert__cCs&|j|||ddd\}}|||S)NF) allow_joinsZ split_subqZcheck_filterable)Z_add_qZ promote_joins)r queryrDreuseZ summarizeZfor_saveZclauseZjoinsr r r resolve_expressioncs   zQ.resolve_expressioncCsfd|jj|jjf}|dr*|dd}t|j}i}|j|jkrN|j|d<|j r\d|d<|||fS)Nz%s.%szdjango.db.models.query_utilszdjango.db.modelsr)Tr*) rrr startswithreplacetupler+r,defaultr-)r pathr1r2r r r r8ms     z Q.deconstruct)NTNFF)rrr__doc__r'r(rKr3r.r?r@rArCrGr8r r r )rr r&9s r&c@s*eZdZdZddZd ddZddZdS) DeferredAttributez A wrapper for a deferred-loading field. When the value is read from this object the first time, the query is executed. cCs ||_dS)N)field)r rOr r r r.szDeferredAttribute.__init__NcCsT|dkr |S|j}|jj}||krL||}|dkrD|j|gdn|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. N)fields)__dict__rOattname_check_parent_chainZrefresh_from_db)r rr%data field_namevalr r r __get__s zDeferredAttribute.__get__cCs6|j}||jj}|jjr2|j|kr2t||jSdS)z Check if the field value can be fetched from a parent field already loaded in the instance. This can be done if the to-be fetched field is a primary key field. N)_metaZget_ancestor_linkrOmodel primary_keyr5rR)r roptsZ link_fieldr r r rSs  z%DeferredAttribute._check_parent_chain)N)rrrrMr.rWrSr r r r rNzs rNc@steZdZeddZeejddddZddZd d Z e d d Z ed dZ edddZ edddZdS)RegisterLookupMixincCs||dS)N) get_lookupsget)r% lookup_namer r r _get_lookupszRegisterLookupMixin._get_lookupN)maxsizecCsddt|D}||S)NcSsg|]}|jdiqS) class_lookups)rQr^).0parentr r r sz3RegisterLookupMixin.get_lookups..)inspectgetmro merge_dicts)r%rbr r r r]szRegisterLookupMixin.get_lookupscCsNddlm}||}|dkr4t|dr4|j|S|dk rJt||sJdS|S)Nr)Lookup output_field)django.db.models.lookupsrir`r7rj get_lookupr)r r_rifoundr r r rls   zRegisterLookupMixin.get_lookupcCsNddlm}||}|dkr4t|dr4|j|S|dk rJt||sJdS|S)Nr) Transformrj)rkrnr`r7rj get_transformr)r r_rnrmr r r ros   z!RegisterLookupMixin.get_transformcCs$i}xt|D]}||qW|S)z Merge dicts in reverse to preference the order of the original list. e.g., merge_dicts([a, b]) will preference the keys in 'a' over those in 'b'. )reversedupdate)ZdictsZmergeddr r r rhszRegisterLookupMixin.merge_dictscCs xt|D]}|jq WdS)N)r$r] cache_clear)r%rr r r _clear_cached_lookupssz)RegisterLookupMixin._clear_cached_lookupscCs4|dkr|j}d|jkri|_||j|<||S)Nrb)r_rQrbrt)r%lookupr_r r r register_lookups  z#RegisterLookupMixin.register_lookupcCs|dkr|j}|j|=dS)zn Remove given lookup from cls lookups. For use in tests only as it's not thread-safe. N)r_rb)r%rur_r r r _unregister_lookupsz&RegisterLookupMixin._unregister_lookup)N)N)rrr classmethodr` functools lru_cacher]rlro staticmethodrhrtrvrwr r r r r\s    r\FcCs|js dS|jjr|sdS|rD|r2||kr2dS|sD|j|krDdS|sR|jrRdS|r|j|kr|r|j|krd|jjj|jf}t |dS)au Return True if this field should be used to descend deeper for select_related() purposes. Used by both the query construction code (sql.query.fill_related_selections()) and the model instance creation code (query.get_klass_info()). Arguments: * field - the field to be checked * restricted - a boolean field, indicating if the field list has been manually restricted using a requested clause) * requested - The select_related() dictionary. * load_fields - the set of fields to be loaded on this model * reverse - boolean, True if we are checking a reverse select related FzXField %s.%s cannot be both deferred and traversed using select_related at the same time.T) Z remote_fieldZ parent_linkZrelated_query_namenamenullrRrYrXZ object_namer)rOZ restricted requestedZ load_fieldsreversemsgr r r select_related_descends$   rcCsVxPtdt|dD]:}t|d|}||kr||r||||dfSqWdS)z Check if the lookup_parts contains references to the given annotations set. Because the LOOKUP_SEP is contained in the default annotation names, check each prefix of the lookup_parts for a match. rN)Fr )rangelenrjoin)Z lookup_parts annotationsnZlevel_n_lookupr r r refs_expression s rcs,fdd}||p*t|ddo*||jjS)z Check that self.model is compatible with target_opts. Compatibility is OK if: 1) model and opts match (where proxy inheritance is removed) 2) model is parent of opts' model or the other way around cs*jj|jkp(|jjkp(|kS)N)rXZconcrete_modelZget_parent_list)r[)rYr r checksz-check_rel_lookup_compatibility..checkrZF)r5rYrX)rYZ target_optsrOrr )rYr check_rel_lookup_compatibilitys rc@s@eZdZdZedddZddZddZd d Zd d Z d S)FilteredRelationz7Specify custom filtering in the ON clause of SQL joins.) conditioncCs:|s td||_d|_t|ts*td||_g|_dS)Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.) ValueError relation_namealiasrr&rrL)r rrr r r r.6s zFilteredRelation.__init__cCs4t||jstS|j|jko2|j|jko2|j|jkS)N)rrNotImplementedrrr)r r;r r r __eq__@s    zFilteredRelation.__eq__cCs,t|j|jd}|j|_|jdd|_|S)N)r)rrrrrL)r cloner r r rIszFilteredRelation.clonecOs tddS)zz QuerySet.annotate() only accepts expression-like arguments (with a resolve_expression() method). z0FilteredRelation.resolve_expression() is unused.N)NotImplementedError)r r1r2r r r rGOsz#FilteredRelation.resolve_expressioncCs&|j}|j|jt|jd}||S)N)rF)rEZbuild_filtered_relation_qrsetrLcompile)r compiler connectionrEwherer r r as_sqlVszFilteredRelation.as_sqlN) rrrrMr&r.rrrGrr r r r r3s   r)F)rMr4ryrfr collectionsrZdjango.core.exceptionsrrZdjango.db.models.constantsrZ django.utilsrZdjango.utils.deprecationrrr9r Exceptionr!r$Noder&rNr\rrrrr r r r s(     A(C %