o ˷eM4@sdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z ddlmZedZed d Zd d ZGd ddejZGdddZGdddZdddZddZddZGdddZdS)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) LOOKUP_SEP)treedjango.db.modelsPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationccs(|V|D] }t|EdHqdSN)__subclasses__ subclassesclssubclassrS/var/www/ideatree/venv/lib/python3.10/site-packages/django/db/models/query_utils.pyr s  r cseZdZdZdZdZdZeZdZdddfd d Z d d Z d dZ ddZ ddZ ddZ dddZddZefddZddZZS)Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). ANDORXORTNF) _connector_negatedcs&tjg|t|||ddS)N)children connectornegated)super__init__sorteditems)selfrrargskwargs __class__rrr1s  z Q.__init__cCst|tst|dddust||s!t|dr|St|St|tr8|s8|\}}}t||i|St|}||_| ||| |||S)N conditionalFTcopy) isinstancergetattr TypeErrorhasattrr% deconstructtyperadd)rotherconn_r r!objrrr_combine8s   z Q._combinecC|||jSr )r1rrr-rrr__or__HzQ.__or__cCr2r )r1rr3rrr__and__Kr5z Q.__and__cCr2r )r1rr3rrr__xor__Nr5z Q.__xor__cCs$t|}|||j||Sr )r+r,rnegate)rr0rrr __invert__Qs z Q.__invert__cCs&|j|||ddd\}}|||S)NF) allow_joins split_subqcheck_filterable)_add_q promote_joins)rqueryr:reuse summarizefor_saveclausejoinsrrrresolve_expressionWs zQ.resolve_expressionccsJ|V|jD]}t|tr|d}t|dr|EdHq|VqdS)zg Recursively yield this Q object and all subexpressions, in depth-first order. flattenN)rr&tupler)rG)rchildrrrrGfs   z Q.flattenc Csddlm}ddlm}ddlm}|d}|D]\}}t|ds'||}|j||ddq||d d | ||j |d } z| |duWSt ye} zt d || WYd} ~ d Sd} ~ ww)z| Do a database query to check if the expressions of the Q instance matches against the expressions. r)Value)Query)SINGLENrEF)selectrF_check)usingz.Got a database error calling check() on %r: %sT)django.db.modelsrJdjango.db.models.sqlrKdjango.db.models.sql.constantsrLrr)add_annotationadd_q get_compiler execute_sqlrloggerwarning) ragainstrOrJrKrLr?namevaluecompilererrrcheckus$      zQ.checkcCsfd|jj|jjf}|dr|dd}t|j}i}|j|jkr'|j|d<|j r.d|d<|||fS)Nz%s.%szdjango.db.models.query_utilsrrTr) r# __module____name__ startswithreplacerHrrdefaultr)rpathr r!rrrr*s      z Q.deconstruct)NTNFF)r`r_ __qualname____doc__rrrrcr$rr1r4r6r7r9rErGrr^r* __classcell__rrr"rr$s$  rc@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 ||_dSr )field)rrirrrrs zDeferredAttribute.__init__NcCsZ|dur|S|j}|jj}||vr)||}|dur%|j|gd||S|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. N)fields)__dict__riattname_check_parent_chainrefresh_from_db)rinstancerdata field_namevalrrr__get__s zDeferredAttribute.__get__cCs6|j}||jj}|jjr|j|krt||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)_metaget_ancestor_linkrimodel primary_keyr'rl)rroopts link_fieldrrrrms  z%DeferredAttribute._check_parent_chainr )r`r_rerfrrsrmrrrrrhs   rhc@steZdZeddZeejddddZddZd d Z e d d Z ed dZ edddZ edddZdS)RegisterLookupMixincCs||dSr ) get_lookupsget)r lookup_namerrr _get_lookupszRegisterLookupMixin._get_lookupN)maxsizecCsddt|D}||S)NcSsg|] }|jdiqS) class_lookups)rkr|).0parentrrr sz3RegisterLookupMixin.get_lookups..)inspectgetmro merge_dicts)rrrrrr{s zRegisterLookupMixin.get_lookupscCNddlm}||}|durt|dr|j|S|dur%t||s%dS|S)Nr)Lookup output_field)django.db.models.lookupsrr~r)r get_lookup issubclass)rr}rfoundrrrr   zRegisterLookupMixin.get_lookupcCr)Nr) Transformr)rrr~r)r get_transformr)rr}rrrrrrrz!RegisterLookupMixin.get_transformcCs i}t|D]}||q|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)dictsmergeddrrrrs  zRegisterLookupMixin.merge_dictscCst|D]}|jqdSr )r r{ cache_clearr rrr_clear_cached_lookupss  z)RegisterLookupMixin._clear_cached_lookupscCs4|dur|j}d|jvri|_||j|<||S)Nr)r}rkrrrlookupr}rrrregister_lookups  z#RegisterLookupMixin.register_lookupcCs"|dur|j}|j|=|dS)zn Remove given lookup from cls lookups. For use in tests only as it's not thread-safe. N)r}rrrrrr_unregister_lookups z&RegisterLookupMixin._unregister_lookupr )r`r_re classmethodr~ functools lru_cacher{rr staticmethodrrrrrrrrrzs         rzFcCs|jsdS|jjr |s dS|r"|r||vrdS|s"|j|vr"dS|s)|jr)dS|rE|j|vrE|rE|j|vrEd|jjj|jf}t |dS)ap 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. * 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) remote_field parent_linkrelated_query_namerZnullrlrvrt object_namer)ri restricted requested load_fieldsreversemsgrrrselect_related_descend s&   rcCsVtdt|dD]}t|d|}||vr(||r(||||dfSq dS)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. rFrN)Fr)rangelenrjoin) lookup_parts annotationsnlevel_n_lookuprrrrefs_expression0s rcs,fdd}||pt|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|jjvp|vSr )rtconcrete_modelget_parent_list)rxrvrrr^Es  z-check_rel_lookup_compatibility..checkrwF)r'rvrt)rv target_optsrir^rrrcheck_rel_lookup_compatibility=s rc@s@eZdZdZedddZddZddZd d Zd d Z d S)FilteredRelationz7Specify custom filtering in the ON clause of SQL joins. conditioncCs:|std||_d|_t|tstd||_g|_dS)Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.) ValueError relation_namealiasr&rrrd)rrrrrrr]s  zFilteredRelation.__init__cCs4t||jstS|j|jko|j|jko|j|jkSr )r&r#NotImplementedrrrr3rrr__eq__gs    zFilteredRelation.__eq__cCs,t|j|jd}|j|_|jdd|_|S)Nr)rrrrrd)rclonerrrrpszFilteredRelation.clonecOstd)zz QuerySet.annotate() only accepts expression-like arguments (with a resolve_expression() method). z0FilteredRelation.resolve_expression() is unused.)NotImplementedError)rr r!rrrrEvsz#FilteredRelation.resolve_expressioncCs&|j}|j|jt|jd}||S)N)r@)r?build_filtered_relation_qrsetrdcompile)rr\ connectionr?whererrras_sql}s zFilteredRelation.as_sqlN) r`r_rerfrrrrrErrrrrrZs   r)F)rfr%rrlogging collectionsrdjango.core.exceptionsr django.dbrrdjango.db.models.constantsr django.utilsr getLoggerrWr r Noderrhrzrrrrrrrrs.     w) G%