ie AFdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZddlmZdd lmZej(d Zed d Zd ZGddej0ZGddZGddZGddZddZdZdZGddZ y)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) FieldError)DEFAULT_DB_ALIAS DatabaseError connections) LOOKUP_SEP)tree)cached_property) make_hashabledjango.db.modelsPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationc#hK||jD]}t|Ed{y7wN)__subclasses__ subclassesclssubclasss EE:\TRUCKING_PROJECT\Lib\site-packages\django/db/models/query_utils.pyrrs4 I&&((h'''('s &202ceZdZdZdZdZdZeZdZdddfd Z d Z d Z d Z d Z dZ ddZdZefdZdZedZdZdZxZS)Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). ANDORXORTNF) _connector_negatedc`t|g|t|j||y)N)children connectornegated)super__init__sorteditems)selfrrargskwargs __class__s rr"z Q.__init__2s3 5t5fV\\^45   ct|dddur t||s|jS|s t|tr|jS|j |}|j |||j |||S)N conditionalF)r)getattr TypeErrorcopy isinstancercreateadd)r%otherconnobjs r_combinez Q._combine9sv 5- /5 8E" "::< E1-99; kkDk) d t r)c:|j||jSr)r5rr%r2s r__or__zQ.__or__Fs}}UDGG,,r)c:|j||jSr)r5rr7s r__and__z Q.__and__I}}UDHH--r)c:|j||jSr)r5rr7s r__xor__z Q.__xor__Lr;r)cF|j}|j|Sr)r.negate)r%r4s r __invert__z Q.__invert__Osiik  r)c\|j|||dd|\}}|j||S)NF) allow_joins split_subqcheck_filterable summarize)_add_q promote_joins)r%queryrBreuserEfor_saveclausejoinss rresolve_expressionzQ.resolve_expressionTsB   #" %  E" r)c#K||jD]@}t|tr|d}t|dr|j Ed{=|By7 w)zg Recursively yield this Q object and all subexpressions, in depth-first order. flattenN)rr/tuplehasattrrP)r%childs rrPz Q.flattendsV  ]] E%'aui( ==?**   +sAA A Ac Jddlm}m}ddlm}ddlm}ddlm}|d}|jD]-\} } t| ds|| } |j| | d /|j|d d t|jjr)|jt!||d | n|j||j#|} | j%|duS#t&$r!} t(j+d|| Yd} ~ y d} ~ wwxYw)z| Do a database query to check if the expressions of the Q instance matches against the expressions. r) BooleanFieldValue)Coalesce)Query)SINGLENrMF)selectrO_checkT) output_field)usingz.Got a database error calling check() on %r: %s)django.db.modelsrUrVdjango.db.models.functionsrWdjango.db.models.sqlrXdjango.db.models.sql.constantsrYr$rRadd_annotationrfeaturessupports_comparing_boolean_expradd_qr get_compiler execute_sqlrloggerwarning) r%againstr]rUrVrWrXrYrHnamevaluecompileres rcheckzQ.checkss 97.9d "==? @RSDT]]# >>T\\ )#'>>F< <* 'E%'" U%e,e -&  'Xr)c`t|tstS|j|jk(Sr)r/rNotImplementedr{r7s r__eq__zQ.__eq__s%%#! !~~..r)c,t|jSr)hashr{)r%s r__hash__z Q.__hash__sDMM""r))NTNFF)rsrr __qualname____doc__rrrrvr+r"r5r8r:r=r@rMrPrrorxr r{rr __classcell__)r(s@rrr%s C B CGK)-  -.. SX $4: "  / #r)rc$eZdZdZdZddZdZy)DeferredAttributez A wrapper for a deferred-loading field. When the value is read from this object the first time, the query is executed. c||_yr)field)r%rs rr"zDeferredAttribute.__init__s  r)Nc||S|j}|jj}||vr]|j|}|E|j!|jj r t d|j|g||S|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. z4Cannot read a generated field from an unsaved model.)fields)__dict__rattname_check_parent_chainpk generatedAttributeErrorrefresh_from_db)r%instancerdata field_namevals r__get__zDeferredAttribute.__get__s  K  ZZ'' T !**84C{;;&4::+?+?(N(( (=J$'Z Jr)c|j}|j|jj}|jjr%|j|k7rt ||j Sy)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)_metaget_ancestor_linkrmodel primary_keyr,r)r%ropts link_fields rrz%DeferredAttribute._check_parent_chainsU ~~++DJJ,<,<= :: ! !djjJ&>8Z%7%78 8r)r)rsrrrrr"rrr)rrrs  . r)rceZdZdZdZdZy)class_or_instance_methodz Hook used in RegisterLookupMixin to return partial functions depending on the caller type (instance or class of models.Field). c ||_||_yr) class_methodinstance_method)r%rrs rr"z!class_or_instance_method.__init__s(.r)c| tj|j|Stj|j|Sr) functoolspartialrr)r%rowners rrz class_or_instance_method.__get__s:  $$T%6%6> >  !5!5x@@r)N)rsrrrrr"rrr)rrrs /Ar)rceZdZdZej dZdZeeeZ e eZdZ dZ e dZe dZd d Zd d ZeeeZe eZd d Zd d ZeeeZe eZy)RegisterLookupMixincB|jj|dSr) get_lookupsget)r% lookup_names r _get_lookupzRegisterLookupMixin._get_lookups!%%k488r)ctj|Dcgc]}|jjdi }}|j |Scc}wN class_lookups)inspectgetmrorr merge_dicts)rparentrs rget_class_lookupsz%RegisterLookupMixin.get_class_lookupssQDK>>RUCV 9?FOO   4  }-- s#AcP|j}t|ddx}ri||S|SNinstance_lookups)rr,)r%rrs rget_instance_lookupsz(RegisterLookupMixin.get_instance_lookupss=..0 &t-?F F  F8m8'78 8r)cddlm}|j|}|'t|dr|jj |S| t ||sy|S)Nr)Lookupr\)django.db.models.lookupsrrrRr\ get_lookup issubclass)r%rrfounds rrzRegisterLookupMixin.get_lookupsQ3  - =WT>:$$// < <  Zv%> r)cddlm}|j|}|'t|dr|jj |S| t ||sy|S)Nr) Transformr\)rrrrRr\ get_transformr)r%rrrs rrz!RegisterLookupMixin.get_transform sQ6  - =WT>:$$22;? ?  Zy%A r)cLi}t|D]}|j||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)dictsmergedds rrzRegisterLookupMixin.merge_dictss- % A MM!   r)cXt|D]}|jjyr)rr cache_clearrs r_clear_cached_class_lookupsz/RegisterLookupMixin._clear_cached_class_lookups"s'"3 5H  & & 2 2 4 5r)Nc| |j}d|jvri|_||j|<|j|Sr)rrrrrlookuprs rregister_class_lookupz)RegisterLookupMixin.register_class_lookup'sH   ,,K #,, . "C )/+& '') r)cj| |j}d|jvri|_||j|<|Sr)rrrr%rrs rregister_instance_lookupz,RegisterLookupMixin.register_instance_lookup0s<   ,,K T]] 2$&D !-3k* r)cZ| |j}|j|=|jy)zn Remove given lookup from cls lookups. For use in tests only as it's not thread-safe. N)rrrrs r_unregister_class_lookupz,RegisterLookupMixin._unregister_class_lookup=s/   ,,K   k * '')r)c:| |j}|j|=y)zs Remove given lookup from instance lookups. For use in tests only as it's not thread-safe. N)rrrs r_unregister_instance_lookupz/RegisterLookupMixin._unregister_instance_lookupGs#   ,,K  ! !+ .r)r)rsrrrrrcacherrrr classmethodrr staticmethodrrrrregister_lookuprr_unregister_lookuprr)rrrs9__..  ++<>RSK#$5655/7O((=>*/2 "= ++CDr)rc^|jsy|jjr|sy|r&|r|j|vry|s|j|vry|s |jry|rN|rL|j|vr>||vr:t d|j jjd|jdy)af Return True if this field should be used to descend deeper for select_related() purposes. Used by both the query construction code (compiler.get_related_selections()) and the model instance creation code (compiler.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. * select_mask - the dictionary of selected fields. * reverse - boolean, True if we are checking a reverse select related FzField rqzM cannot be both deferred and traversed using select_related at the same time.T) remote_field parent_linkrelated_query_namerknullrrr object_name)r restricted requested select_maskreverses rselect_related_descendrVs    %%g u//1B5::Y6 %**  JJ) #  $U[[&&2231UZZLAL L   r)ctdt|dzD]4}tj|d|}|j |s-|||dfcSy)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. rOrN)Nr)rangelenrjoinr) lookup_parts annotationsnlevel_n_lookups rrefs_expressionr}s\ 1c,'!+ ,4#a):; ??> *!<#33 34 r)cxfd}||xs+t|ddxr||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 cjj|jk(xs:|jjjvxs|jvSr)rconcrete_modelget_parent_list)rrs rroz-check_rel_lookup_compatibility..checksV KK & &$*=*= = /""ekk&A&A&CC /,,.. r)rF)r,rr)r target_optsrros` rcheck_rel_lookup_compatibilityrs<   }e,Iu{{7H7H1Ir)cDeZdZdZeddZdZdZdZdZ dZ y ) FilteredRelationz7Specify custom filtering in the ON clause of SQL joins. conditionc|s td||_d|_t|ts td||_d|_y)Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.) ValueError relation_namealiasr/rrresolved_condition)r%rrs rr"zFilteredRelation.__init__sG=> >* )Q'IJ J#"&r)ct||jstS|j|jk(xr4|j|jk(xr|j |j k(Sr)r/r(r~rrrr7s rrzFilteredRelation.__eq__sY%0! !   %"5"5 5 2 ekk) 2%//1 r)ct|j|j}|j|_|jx}|j |_|S)Nr)rrrrrclone)r%rrs rrzFilteredRelation.clonesK !3!3t~~Njj "&"9"9 9  F'9'?'?'AE $ r)cn|j}|jx}r|j||_|Sr)rrrelabeled_clone)r% change_maprrs rrz FilteredRelation.relabeled_clones7 !&!9!9 9  9'9'I'I*'UE $ r)cv|j}|j|j|dddd|_|S)NTF) can_reuserBrCupdate_join_typesr)r build_filterrr)r%rHrIr&r'rs rrMz#FilteredRelation.resolve_expressionsI #(#5#5 NN# $6$ $   r)c8|j|jSr)compiler)r%rm connections ras_sqlzFilteredRelation.as_sqls 7 788r)N) rsrrrrrr"rrrrMrrr)rrrs(A343 '  9r)r)F)!rrrlogging collectionsrdjango.core.exceptionsr django.dbrrrdjango.db.models.constantsr django.utilsr django.utils.functionalr django.utils.hashabler getLoggerrhr rNoderrrrrrrrrr)rrs"-BB13/   - . M  ( J# J#Z**Z A A dEdEN$N :2929r)