3 v^/@sdZddlZddlZddlZddlmZddlmZddlm Z eddZ Gdd d e Z d d Z Gd d d ZGddde jZGdddZGdddZdddZddZddZGdddZdS)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) LOOKUP_SEP)treePathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationc@seZdZdZdS) InvalidQueryz?The query passed to raw() isn't a safe query to use with raw().N)__name__ __module__ __qualname____doc__r r @/usr/lib/python3.6/site-packages/django/db/models/query_utils.pyrsrccs*|Vx|jD]}t|EdHqWdS)N)__subclasses__ subclasses)clssubclassr r r rsrc@s&eZdZdZdZddZdddZdS) QueryWrapperz A type that indicates the contents are an SQL fragment and the associate parameters. Can be used to pass opaque data to a where-clause, for example. FcCs|t|f|_dS)N)listdata)selfsqlparamsr r r __init__(szQueryWrapper.__init__NcCs|jS)N)r)rcompiler connectionr r r as_sql+szQueryWrapper.as_sql)NN)rrr r contains_aggregaterrr r r r r!srcsjeZdZdZdZdZeZdZdddfdd 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|j||ddS)N)children connectornegated)superrsorteditems)rrr argskwargs) __class__r r r:sz Q.__init__cCsZt|tst||s tj|S|s.tj|St|}||_|j|||j|||S)N) isinstancer TypeErrorcopydeepcopytyper"add)rotherconnobjr r r _combine=s      z Q._combinecCs|j||jS)N)r3r)rr0r r r __or__NszQ.__or__cCs|j||jS)N)r3r)rr0r r r __and__Qsz Q.__and__cCs$t|}|j||j|j|S)N)r.r/rZnegate)rr2r r r __invert__Ts z Q.__invert__cCs&|j|||ddd\}}|j||S)NF) allow_joinsZ split_subqZcheck_filterable)Z_add_qZ promote_joins)rqueryr7reuse summarizefor_saveZclauseZjoinsr r r resolve_expressionZs   zQ.resolve_expressioncCsd|jj|jjf}|jdr*|jdd}fi}}t|jdkrpt|jdt rp|jd}|d|di}n t |j}|j |j krd|j i}|j rd|d<|||fS) Nz%s.%szdjango.db.models.query_utilszdjango.db.modelsrrTr ) r)rr startswithreplacelenr!r*rtupler"defaultr#)rpathr'r(childr r r deconstructds        z Q.deconstruct)NTNFF)rrr r rrrB conditionalrr3r4r5r6r<rE __classcell__r r )r)r r/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 ||_dS)N)field)rrIr r r rzszDeferredAttribute.__init__NcCsd|dkr |S|j}|jj}|j|||kr\|j|}|dkrT|j|gdt||}|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. N)fields)__dict__rIattnameget_check_parent_chainZrefresh_from_dbgetattr)rinstancerr field_namevalr r r __get__}s  zDeferredAttribute.__get__cCs6|j}|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_linkrImodel primary_keyrOrL)rrPoptsZ link_fieldr r r rNs  z%DeferredAttribute._check_parent_chain)N)rrr r rrSrNr r r r rHus rHc@steZdZeddZeejddddZddZd d Z e d d Z ed dZ edddZ edddZdS)RegisterLookupMixincCs|jj|dS)N) get_lookupsrM)r lookup_namer r r _get_lookupszRegisterLookupMixin._get_lookupN)maxsizecCsddtj|D}|j|S)NcSsg|]}|jjdiqS) class_lookups)rKrM).0parentr r r sz3RegisterLookupMixin.get_lookups..)inspectgetmro merge_dicts)rr]r r r rYszRegisterLookupMixin.get_lookupscCsPddlm}|j|}|dkr4t|dr4|jj|S|dk rLt|| rLdS|S)Nr)Lookup output_field)django.db.models.lookupsrdr[hasattrre get_lookup issubclass)rrZrdfoundr r r rhs   zRegisterLookupMixin.get_lookupcCsPddlm}|j|}|dkr4t|dr4|jj|S|dk rLt|| rLdS|S)Nr) Transformre)rfrkr[rgre get_transformri)rrZrkrjr r r rls   z!RegisterLookupMixin.get_transformcCs$i}xt|D]}|j|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)dictsZmergeddr r r rcszRegisterLookupMixin.merge_dictscCs xt|D]}|jjq WdS)N)rrY cache_clear)rrr r r _clear_cached_lookupssz)RegisterLookupMixin._clear_cached_lookupscCs4|dkr|j}d|jkri|_||j|<|j|S)Nr])rZrKr]rr)rlookuprZr 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)rZr])rrsrZr r r _unregister_lookupsz&RegisterLookupMixin._unregister_lookup)N)N)rrr classmethodr[ functools lru_cacherYrhrl staticmethodrcrrrtrur r r r rXs    rXFcCs|js dS|jjr| rdS|rH|r4|j|kr4dS| rH|j|krHdS| rX|jrXdS|r|j|kr|r|j|krtd|jjj |jfdS)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) remote_field parent_linkZrelated_query_namenamenullrLrrUrT object_name)rIZ restrictedZ requestedZ load_fieldsreverser r r select_related_descends"  rcCsZxPtdt|dD]:}tj|d|}||kr||r||||dfSqWdffS)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. r=rNF)ranger@rjoin)Z lookup_parts annotationsnZlevel_n_lookupr r r refs_expressions 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(|jjjkp(|jkS)N)rTconcrete_modelget_parent_list)rW)rUr r checksz-check_rel_lookup_compatibility..checkrVF)rOrUrT)rUZ target_optsrIrr )rUr 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_namealiasr*rrrC)rrrr r r r0s zFilteredRelation.__init__cCs0t||jo.|j|jko.|j|jko.|j|jkS)N)r*r)rrr)rr0r r r __eq__:s   zFilteredRelation.__eq__cCs,t|j|jd}|j|_|jdd|_|S)N)r)rrrrrC)rcloner r r rBszFilteredRelation.clonecOs tddS)zz QuerySet.annotate() only accepts expression-like arguments (with a resolve_expression() method). z0FilteredRelation.resolve_expression() is unused.N)NotImplementedError)rr'r(r r r r<Hsz#FilteredRelation.resolve_expressioncCs&|j}|j|jt|jd}|j|S)N)r9)r8Zbuild_filtered_relation_qrsetrCcompile)rrrr8wherer r r rOszFilteredRelation.as_sqlN) rrr r rrrrr<rr r r r r-s  r)F)r r,rwra collectionsrdjango.db.models.constantsr django.utilsrr ExceptionrrrNoderrHrXrrrrr r r r s"    F(C $