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 >/tmp/pip-build-8lau8j11/django/django/db/models/query_utils.pyrs rccs/|Vx#|jD]}t|EdHqWdS)N)__subclasses__ subclasses)clssubclassr r r rsrc@s:eZdZdZdZddZddddZdS) 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)selfZsqlparamsr r r __init__(szQueryWrapper.__init__NcCs|jS)N)r)rcompiler connectionr r r as_sql+szQueryWrapper.as_sql)rrr r Zcontains_aggregaterrr r r r r!s  rcseZdZdZdZdZeZdZddddfd d Zd d Z d dZ ddZ ddZ dddddddZ ddZS)Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). ANDORT _connectorN_negatedFcs5tjd|t|jd|d|dS)Nchildren connectornegated)superrsorteditems)rrrargskwargs) __class__r r r:sz Q.__init__cCs}t|tst||s.tj|S|sAtj|St|}||_|j|||j|||S)N) isinstancer TypeErrorcopydeepcopytyper add)rotherconnobjr r r _combine=s    z Q._combinecCs|j||jS)N)r1r)rr.r r r __or__NszQ.__or__cCs|j||jS)N)r1r)rr.r r r __and__Qsz Q.__and__cCs0t|}|j||j|j|S)N)r,r-rnegate)rr0r r r __invert__Ts z Q.__invert__c Cs;|j||d|dddd\}}|j||S)N allow_joinsZ split_subqFZcheck_filterable)Z_add_qZ promote_joins)rqueryr6reuseZ summarizeZfor_saveZclauseZjoinsr r r resolve_expressionZs  zQ.resolve_expressioncCsd|jj|jjf}|jdr=|jdd}fi}}t|jdkrt|jdt r|jd}|d|di}n0t |j}|j |j krd|j i}|j rd|d<|||fS) Nz%s.%szdjango.db.models.query_utilszdjango.db.modelsrrTr) r'rr startswithreplacelenrr(rtupler defaultr!)rpathr%r&childr r r deconstructds ,   z Q.deconstruct)rrr r rrr?Z conditionalrr1r2r3r5r9rBr r )r'r r/s      rc@s=eZdZdZddZdddZddZdS) 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)rrDr r r rzszDeferredAttribute.__init__NcCs|dkr|S|j}|jj}|j|||kr|j|}|dkrz|jd|gt||}|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. Nfields)__dict__rDattnameget_check_parent_chainZrefresh_from_dbgetattr)rinstancerr field_namevalr r r __get__}s     zDeferredAttribute.__get__cCsM|j}|j|jj}|jjrI|j|krIt||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_linkrDmodel primary_keyrJrG)rrKoptsZ link_fieldr r r rIs  z%DeferredAttribute._check_parent_chain)rrr r rrNrIr r r r rCus  rCc@seZdZeddZeejddddZddZd d Z e d d Z ed dZ edddZ edddZdS)RegisterLookupMixincCs|jj|dS)N) get_lookupsrH)r lookup_namer r r _get_lookupszRegisterLookupMixin._get_lookupmaxsizeNcCs)ddtj|D}|j|S)NcSs%g|]}|jjdiqS) class_lookups)rFrH).0parentr r r s z3RegisterLookupMixin.get_lookups..)inspectZgetmro merge_dicts)rrXr r r rTszRegisterLookupMixin.get_lookupscCsnddlm}|j|}|dkrJt|drJ|jj|S|dk rjt|| rjdS|S)Nr)Lookup output_field)django.db.models.lookupsr^rVhasattrr_ get_lookup issubclass)rrUr^foundr r r rbszRegisterLookupMixin.get_lookupcCsnddlm}|j|}|dkrJt|drJ|jj|S|dk rjt|| rjdS|S)Nr) Transformr_)r`rerVrar_ get_transformrc)rrUrerdr r r rfsz!RegisterLookupMixin.get_transformcCs.i}x!t|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)ZdictsZmergeddr r r r]szRegisterLookupMixin.merge_dictscCs(x!t|D]}|jjq WdS)N)rrT cache_clear)rrr r r _clear_cached_lookupssz)RegisterLookupMixin._clear_cached_lookupscCsH|dkr|j}d|jkr-i|_||j|<|j|S)NrX)rUrFrXrk)rlookuprUr 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)rUrX)rrlrUr r r _unregister_lookups  z&RegisterLookupMixin._unregister_lookup)rrr classmethodrV functools lru_cacherTrbrf staticmethodr]rkrmrnr r r r rSs !   rSFcCs|js dS|jjr$| r$dS|r`|rF|j|krFdS| r`|j|kr`dS| rt|jrtdS|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) Z remote_fieldZ parent_linkZrelated_query_namenamenullrGrrPrOZ object_name)rDZ restricted requestedZ load_fieldsreverser r r select_related_descends" rwcCsuxhtdt|dD]M}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_partsZ annotationsnZlevel_n_lookupr r r refs_expressions  r{cs@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)rOZconcrete_modelZget_parent_list)rR)rPr r checksz-check_rel_lookup_compatibility..checkrQF)rJrPrO)rPZ target_optsrDr|r )rPr check_rel_lookup_compatibilitys r}c@s[eZdZdZdeddZddZddZd d Zd d Z d S)FilteredRelationz7Specify custom filtering in the ON clause of SQL joins. conditioncCsU|std||_d|_t|ts?td||_g|_dS)Nzrelation_name cannot be empty.z*condition argument must be a Q() instance.) ValueError relation_namealiasr(rrr@)rrrr r r r0s     zFilteredRelation.__init__cCsFt||joE|j|jkoE|j|jkoE|j|jkS)N)r(r'rrr)rr.r r r __eq__:szFilteredRelation.__eq__cCs>t|jd|j}|j|_|jdd|_|S)Nr)r~rrrr@)rcloner r r rBs zFilteredRelation.clonecOstddS)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 r9Hsz#FilteredRelation.resolve_expressioncCs7|j}|j|jdt|j}|j|S)Nr8)r7Zbuild_filtered_relation_qrsetr@compile)rrrr7wherer r r rOs !zFilteredRelation.as_sqlN) rrr r rrrrr9rr r r r r~-s    r~)r r*rpr\ collectionsrZdjango.db.models.constantsrZ django.utilsrr ExceptionrrrNoderrCrSrwr{r}r~r r r r s"    F(C$