U R+|j<@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZedZed d Zd d ZGd ddejZGdddZGdddZGdddZdddZddZddZGdddZdS)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)treedjango.db.modelsPathInfozGfrom_opts to_opts target_fields join_field m2m direct filtered_relationccs&|V|D]}t|EdHqdSN)__subclasses__ subclassesclssubclassrdc:\users\snmko\desktop\web_content\django_cbt\venv\Lib\site-packages\django/db/models/query_utils.pyr s r cseZdZdZdZdZdZeZdZdeeefZ dddfd d Z d d Z d dZ ddZ ddZddZdddZddZefddZddZZS)Qze Encapsulate filters as objects that can then be combined logically (using `&` and `|`). ANDORXORTNF) _connector_negatedcsZ||jkr8ddd|jddD}td|dtj|t|||ddS)Nz, css|] }|VqdSr r).0connrrr 3szQ.__init__..z_connector must be one of z , or None.)children connectornegated) connectorsjoin ValueErrorsuper__init__sorteditems)selfrrargskwargsZconnector_reprs __class__rrr$1s z Q.__init__cCsbt|dddkrt||s$|S|s:t|tr:|S|j|d}|||||||S)N conditionalF)r)getattr TypeErrorcopy isinstancercreateadd)r'otherrobjrrr_combine;s   z Q._combinecCs|||jSr )r5rr'r3rrr__or__HszQ.__or__cCs|||jSr )r5rr6rrr__and__Ksz Q.__and__cCs|||jSr )r5rr6rrr__xor__Nsz Q.__xor__cCs|}||Sr )r/negate)r'r4rrr __invert__Qsz Q.__invert__cCs(|j|||dd|d\}}|||S)NF) allow_joinsZ split_subqZcheck_filterable summarize)Z_add_qZ promote_joins)r'queryr<reuser=Zfor_saveZclauseZjoinsrrrresolve_expressionVs  zQ.resolve_expressionccsH|V|jD]6}t|tr"|d}t|dr<|EdHq |Vq dS)zg Recursively yield this Q object and all subexpressions, in depth-first order. rflattenN)rr0tuplehasattrrA)r'childrrrrAfs   z Q.flattenc Cs ddlm}m}ddlm}ddlm}ddlm}|d}| D]*\} } t | ds^|| } |j | | dd qD| |d d t |j jr|t||d |d n |||j|d} z| |dk WStk r} ztd|| WYd Sd} ~ XYnXdS)z| Do a database query to check if the expressions of the Q instance matches against the expressions. r) BooleanFieldValue)Coalesce)Query)SINGLENr@F)selectr_checkT) output_field)usingz.Got a database error calling check() on %r: %s)Zdjango.db.modelsrErFZdjango.db.models.functionsrGZdjango.db.models.sqlrHZdjango.db.models.sql.constantsrIr&rCZadd_annotationrfeaturesZsupports_comparing_boolean_exprZadd_qrZ get_compilerZ execute_sqlrloggerwarning) r'ZagainstrMrErFrGrHrIr>namevaluecompilererrrcheckus&       zQ.checkcCsfd|jj|jjf}|dr*|dd}t|j}i}|j|jkrN|j|d<|j r\d|d<|||fS)Nz%s.%szdjango.db.models.query_utilsr rTr) r+ __module____name__ startswithreplacerBrrdefaultr)r'pathr(r)rrr deconstructs     z Q.deconstruct)NTNFF)rWrV __qualname____doc__rrrrZr,r r$r5r7r8r9r;r@rArrUr\ __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)r'rarrrr$szDeferredAttribute.__init__NcCsT|dkr |S|j}|jj}||krL||}|dkrD|j|gdn|||<||S)zx Retrieve and caches the value from the datastore on the first lookup. Return the cached value. N)fields)__dict__raattname_check_parent_chainZrefresh_from_db)r'instancerdata field_namevalrrr__get__s zDeferredAttribute.__get__cCs6|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_linkramodel primary_keyr-rd)r'rfoptsZ link_fieldrrrres  z%DeferredAttribute._check_parent_chain)N)rWrVr]r^r$rjrerrrrr`s r`c@s eZdZdZddZddZdS)class_or_instance_methodz Hook used in RegisterLookupMixin to return partial functions depending on the caller type (instance or class of models.Field). cCs||_||_dSr ) class_methodinstance_method)r'rprqrrrr$sz!class_or_instance_method.__init__cCs$|dkrt|j|St|j|Sr ) functoolspartialrprq)r'rfownerrrrrjsz class_or_instance_method.__get__N)rWrVr]r^r$rjrrrrrosroc@seZdZddZejddddZddZeeeZ e eZd d Z d d Z e d dZe ddZdddZdddZeeeZe eZdddZdddZeeeZe eZdS)RegisterLookupMixincCs||dSr ) get_lookupsget)r' lookup_namerrr _get_lookupszRegisterLookupMixin._get_lookupN)maxsizecCsddt|D}||S)NcSsg|]}|jdiqS) class_lookups)rcrw)rparentrrr sz9RegisterLookupMixin.get_class_lookups..)inspectgetmro merge_dicts)rr{rrrget_class_lookupssz%RegisterLookupMixin.get_class_lookupscCs$|}t|dd}r ||S|SNinstance_lookups)rr-)r'r{rrrrget_instance_lookupssz(RegisterLookupMixin.get_instance_lookupscCsNddlm}||}|dkr4t|dr4|j|S|dk rJt||sJdS|S)Nr)LookuprL)django.db.models.lookupsrryrCrL get_lookup issubclass)r'rxrfoundrrrrs   zRegisterLookupMixin.get_lookupcCsNddlm}||}|dkr4t|dr4|j|S|dk rJt||sJdS|S)Nr) TransformrL)rrryrCrL get_transformr)r'rxrrrrrrs   z!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)Zdictsmergeddrrrrs  zRegisterLookupMixin.merge_dictscCst|D]}|jqdSr )r r cache_clearrrrr_clear_cached_class_lookups s z/RegisterLookupMixin._clear_cached_class_lookupscCs4|dkr|j}d|jkri|_||j|<||S)Nr{)rxrcr{rrlookuprxrrrregister_class_lookups  z)RegisterLookupMixin.register_class_lookupcCs,|dkr|j}d|jkri|_||j|<|Sr)rxrcrr'rrxrrrregister_instance_lookups   z,RegisterLookupMixin.register_instance_lookupcCs"|dkr|j}|j|=|dS)zn Remove given lookup from cls lookups. For use in tests only as it's not thread-safe. N)rxr{rrrrr_unregister_class_lookup&sz,RegisterLookupMixin._unregister_class_lookupcCs|dkr|j}|j|=dS)zs Remove given lookup from instance lookups. For use in tests only as it's not thread-safe. N)rxrrrrr_unregister_instance_lookup0sz/RegisterLookupMixin._unregister_instance_lookup)N)N)N)N)rWrVr]ryrr lru_cacherrrorv classmethodrr staticmethodrrrrZregister_lookuprrZ_unregister_lookuprrrrrus4         ruFcCs|js dS|jjr|sdS|rD|r2||kr2dS|sD|j|krDdS|sR|jrRdS|r|r|j|kr||krtd|jjjd|jddS)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 .zM cannot be both deferred and traversed using select_related at the same time.T) Z remote_fieldZ parent_linkZrelated_query_namerQnullrrlrkZ object_name)raZ restricted requestedZ select_maskreverserrrselect_related_descend?s.  rcCsLtdt|dD]4}t|d|}||r|||dfSqdS)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. rrN)Nr)rangelenrr!rw)Z lookup_parts annotationsnZlevel_n_lookuprrrrefs_expressionfs  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(|jjkp(|kSr )rkZconcrete_modelZget_parent_list)rnrlrrrU{s  z-check_rel_lookup_compatibility..checkrmF)r-rlrk)rlZ target_optsrarUrrrcheck_rel_lookup_compatibilityss 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.)r" relation_namealiasr0rrr[)r'rrrrrr$s zFilteredRelation.__init__cCs4t||jstS|j|jko2|j|jko2|j|jkSr )r0r+NotImplementedrrrr6rrr__eq__s    zFilteredRelation.__eq__cCs,t|j|jd}|j|_|jdd|_|S)Nr)rrrrr[)r'clonerrrrszFilteredRelation.clonecOs tddS)zz QuerySet.annotate() only accepts expression-like arguments (with a resolve_expression() method). z0FilteredRelation.resolve_expression() is unused.N)NotImplementedError)r'r(r)rrrr@sz#FilteredRelation.resolve_expressioncCs&|j}|j|jt|jd}||S)N)r?)r>Zbuild_filtered_relation_qrsetr[compile)r'rS connectionr>whererrras_sqlszFilteredRelation.as_sqlN) rWrVr]r^rr$rrr@rrrrrrs   r)F)r^rrr~logging collectionsrZdjango.core.exceptionsrZ django.dbrrrZdjango.db.models.constantsrZ django.utilsr getLoggerrOr r Noderr`rorurrrrrrrrs,     |)g '