3 (h-&@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z edd Z Gd d d e ZGd d d eZGddde jZGdddeZGdddeZdddZddZddZdS)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. )unicode_literalsN) namedtuple)FieldDoesNotExist) LOOKUP_SEP)treePathInfoz5from_opts to_opts target_fields join_field m2m directc@seZdZdZdS) InvalidQueryzE The query passed to raw isn't a safe query to use with raw. N)__name__ __module__ __qualname____doc__r r @/tmp/pip-install-q3hcpn_q/Django/django/db/models/query_utils.pyrsrc@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)selfZsqlparamsr r r__init__%szQueryWrapper.__init__NcCs|jS)N)r)rcompiler connectionr r ras_sql(szQueryWrapper.as_sql)NN)r r r r Zcontains_aggregaterrr r r rrsrcsVeZdZdZdZdZeZfddZddZdd Z d d Z d d Z dddZ Z S)Qzf Encapsulates filters as objects that can then be combined logically (using `&` and `|`). ANDORcs&tt|jt|t|jddS)N)children)superrrritems)rargskwargs) __class__r rr6sz Q.__init__cCs>t|tst|t|}||_|j|||j|||S)N) isinstancer TypeErrortypeZ connectoradd)rotherconnobjr r r_combine9s    z Q._combinecCs|j||jS)N)r(r)rr%r r r__or__BszQ.__or__cCs|j||jS)N)r(r)rr%r r r__and__Esz Q.__and__cCs$t|}|j||j|j|S)N)r#r$rnegate)rr'r r r __invert__Hs z Q.__invert__NTFcCs$|j|||dd\}}|j||S)NF) allow_joinsZ split_subq)Z_add_qZ promote_joins)rqueryr-ZreuseZ summarizeZfor_saveZclauseZjoinsr r rresolve_expressionNs zQ.resolve_expression)NTNFF)r r r r rrdefaultrr(r)r*r,r/ __classcell__r r )r 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 ||_dS)N) field_name)rr3modelr r rr[szDeferredAttribute.__init__Nc s|dkr S|j}|j}|jjkry|jj}Wn,tk rffdd|jDd}YnX|j}j||}|dkr|j jgdt |j}||j<|jS)zz Retrieves and caches the value from the datastore on the first lookup. Returns the cached value. Ncsg|]}|jjkr|qSr )attnamer3).0f)rr r msz-DeferredAttribute.__get__..r)fields) _meta__dict__getr3 get_fieldrr9name_check_parent_chainZrefresh_from_dbgetattr)rinstanceclsoptsrr7r>valr )rr__get__^s    zDeferredAttribute.__get__cCs:|j}|j|}|j|j}|jr6||kr6t||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)r:r=Zget_ancestor_linkr4 primary_keyr@r5)rrAr>rCr7Z link_fieldr r rr?xs    z%DeferredAttribute._check_parent_chain)N)r r r r rrEr?r r r rr2Vs r2c@s@eZdZddZddZddZed dd Zed d d ZdS)RegisterLookupMixinc Csly |j|Stk rTx4tj|jD]$}d|jkr8q(||jkr(|j|Sq(WYntk rfYnXdS)N class_lookups)rHKeyErrorinspectgetmror r;AttributeError)r lookup_nameparentr r r _get_lookups   zRegisterLookupMixin._get_lookupcCsPddlm}|j|}|dkr4t|dr4|jj|S|dk rLt|| rLdS|S)Nr)Lookup output_field)django.db.models.lookupsrPrOhasattrrQ get_lookup issubclass)rrMrPfoundr r rrTs   zRegisterLookupMixin.get_lookupcCsPddlm}|j|}|dkr4t|dr4|jj|S|dk rLt|| rLdS|S)Nr) TransformrQ)rRrWrOrSrQ get_transformrU)rrMrWrVr r rrXs   z!RegisterLookupMixin.get_transformNcCs,|dkr|j}d|jkri|_||j|<|S)NrH)rMr;rH)rBlookuprMr r rregister_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)rMrH)rBrYrMr r r_unregister_lookupsz&RegisterLookupMixin._unregister_lookup)N)N) r r r rOrTrX classmethodrZr[r r r rrGs   rGFcCs|js dS|jjr| rdS|rH|r4|j|kr4dS| rH|j|krHdS| rX|jrXdS|r|j|kr|r|j|krtd|jjj |jfdS)av Returns 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_namer>nullr5rr4r:Z object_name)fieldZ restricted requestedZ load_fieldsreverser r rselect_related_descends"  racCsXxNtt|dD]:}tj|d|}||kr||r||||dfSqWdffS)z A helper method to check if the lookup_parts contains references to the given annotations set. Because the LOOKUP_SEP is contained in the default annotation names we must check each prefix of the lookup_parts for a match. rNF)rangelenrjoin)Z lookup_parts annotationsnZlevel_n_lookupr r rrefs_expressions rhcs,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)r:Zconcrete_modelZget_parent_list)rC)r4r rchecksz-check_rel_lookup_compatibility..checkrFF)r@r4r:)r4Z target_optsr^rir )r4rcheck_rel_lookup_compatibilitys rj)F)r __future__rrJ collectionsrZdjango.core.exceptionsrZdjango.db.models.constantsrZ django.utilsrr ExceptionrobjectrNoderr2rGrarhrjr r r rs      *06 $