hWdZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z m Z dd lmZmZmZmZd ZGd d e Zd ZGdde Zy)z QuerySet for PolymorphicModel N) defaultdict) ContentType)FieldDoesNotExist)FilteredRelation) ModelIterableQQuerySet) translate_polymorphic_field_path0translate_polymorphic_filter_definitions_in_args2translate_polymorphic_filter_definitions_in_kwargstranslate_polymorphic_Q_objectdc(eZdZdZfdZdZxZS)PolymorphicModelIterablez ModelIterable for PolymorphicModel Yields real instances if qs.polymorphic_disabled is False, otherwise acts like a regular ModelIterable. crt|}|jjr|S|j |SN)super__iter__querysetpolymorphic_disabled_polymorphic_iterator)self base_iter __class__s A/home/dcms/DCMS/lib/python3.12/site-packages/polymorphic/query.pyrz!PolymorphicModelIterable.__iter__!s4G$& == - - )))44c#K g}d}ttD]} t|}|j|!|j j |}|D]}||ry_#t$rd}Y8wxYww)a! Here we do the same as:: real_results = queryset._get_real_instances(list(base_iter)) for o in real_results: yield o but it requests the objects in chunks from the database, with Polymorphic_QuerySet_objects_per_request per chunk TFN)range(Polymorphic_QuerySet_objects_per_requestnextappend StopIterationr_get_real_instances)rrbase_result_objects reached_endio real_resultss rrz.PolymorphicModelIterable._polymorphic_iterator's"$ KCD YA'..q1  ==<<=PQL!  )%"&Ks'A5A#,A5# A2.A51A22A5)__name__ __module__ __qualname____doc__rr __classcell__rs@rrrs5 rrcd|jvr |}||_|S|}|jjD]\}}||j|<|S)zF Upcast a class to a different type without asking questions. __init__)__dict__ritems)clsobjnewkvs r transmogrifyr9Hs^% JeLL&&( DAqCLLO JrceZdZdZfdZfdZdZde_eeZdfd Z dZ dZ d Z fd Z fd Zfd Zfd ZdZdZdZfdZfdZfdZdZfdZGddeZddZxZS)PolymorphicQuerySetz QuerySet for PolymorphicModel Contains the core functionality for PolymorphicModel Usually not explicitly needed, except if a custom queryset class is to be used. cjt||i|t|_d|_t df|_y)NFT)rr1r_iterable_classrsetpolymorphic_deferred_loadingrargskwargsrs rr1zPolymorphicQuerySet.__init__fs5 $)&)7$)! .1UDM)rct||i|}|j|_tj|jd|jdf|_|S)Nrr )r_clonercopyr?)rrArBr6rs rrDzPolymorphicQuerySet._clonersZgnd-f-#'#<#< IId77: ;  - -a 0, ( rcLddlm}|j|}d|_|S)Nr )PolymorphicManagerT)managersrG from_queryset_built_with_as_manager)r4rGmanagers r as_managerzPolymorphicQuerySet.as_manager|s(07$22379)-&rTclt|}|D]}|jt| |||S)N)ignore_conflicts)listpre_save_polymorphicr bulk_create)robjs batch_sizerNr5rs rrQzPolymorphicQuerySet.bulk_creates?Dz 'C  $ $ & 'w"4FV"WWrc~|j}d|_t|jtr t |_|S)zswitch off polymorphic behaviour for this query. When the queryset is evaluated, only objects of the type of the base class used for this query are returned.T)rDr issubclassr=rr)rqss rnon_polymorphicz#PolymorphicQuerySet.non_polymorphics5[[]"& b((*B C!.B  rc&|j|S)zOFilter the queryset to only include the classes in args (and their subclasses).) instance_offilterrrAs rrYzPolymorphicQuerySet.instance_ofs{{t{,,rc&|j|S)zJFilter the queryset to exclude the classes in args (and their subclasses).)not_instance_ofrZr\s rr^z#PolymorphicQuerySet.not_instance_ofs{{4{00rct|j||j}t|j||j}t ||z}t ||||S)N)queryset_modelrAusing)r`rBra)negaterArB)r modeldbr rOr_filter_or_exclude)rrbrArB q_objectsadditional_argsrs rrez&PolymorphicQuerySet._filter_or_excludes_D::D M::fDGG I0w)d6)RRrc|Dcgc]*}t|trt|j|n|,}}t ||Scc}w)zBtranslate the field paths in the args, then call vanilla order_by.) isinstancestrr rcrorder_by)r field_namesars rrkzPolymorphicQuerySet.order_bysZ !  !S! -TZZ ;   w--  s/Ac|Dcgc]}t|j|}}t| |}|j ||Scc}w)a! Translate the field paths in the args, then call vanilla defer. Also retain a copy of the original fields passed, which we'll need when we're retrieving the real instance (since we'll need to translate them again, as the model will have changed). )r rcrdefer!_polymorphic_add_deferred_loadingrfieldsrm new_fieldscloners rrozPolymorphicQuerySet.defersMPVV!6tzz1EV V z* //7 WAc|Dcgc]}t|j|}}t| |}|j ||Scc}w)a  Translate the field paths in the args, then call vanilla only. Also retain a copy of the original fields passed, which we'll need when we're retrieving the real instance (since we'll need to translate them again, as the model will have changed). )r rcronly"_polymorphic_add_immediate_loadingrqs rrwzPolymorphicQuerySet.onlysMPVV!6tzz1EV V j) 008 Wruc|j\}}|r|j|df|_y|j|df|_y)z Follows the logic of django.db.models.query.Query.add_deferred_loading(), but for the non-translated field names that were passed to self.defer(). TFN)r?union differencerrlexistingros rrpz5PolymorphicQuerySet._polymorphic_add_deferred_loadingsG ;;% 08{0KT0QD -190C0CK0PRW0WD -rc|j\}}t|}d|vrJ|jd|j|jj j j|r|j|df|_y|df|_y)z Follows the logic of django.db.models.query.Query.add_immediate_loading(), but for the non-translated field names that were passed to self.only() pkFN) r?r>removeaddrc_metarnamer{r|s rrxz6PolymorphicQuerySet._polymorphic_add_immediate_loadings ;;%+& ;    t $ OODJJ,,//44 5 1<0F0Fx0PRW0WD -1t|trtj|yt|tr|j yt |dr!|jD] }||ytj|j|_ y)Nget_source_expressions) rirrrcr conditionhasattrrr r)rmsource_expression patch_lookuprs rrzAPolymorphicQuerySet._process_aggregate_args..patch_lookups!Q.tzz1=A/0Q[[)45)*)A)A)C8%(4$%678:$**affMrct|trfdj|yt|dr!|j D] }||yd|j vsJy)zy*args might be complex expressions too in django 1.8 so the testing for a '___' is rather complex on this onectt|jD];}|j|}t|tk(rd|dvr,J||=y)z#process all children of this Q node___rN)rlenchildrentypetuple)my_modelnoder'child(_PolymorphicQuerySet___lookup_assert_msgtree_node_test___lookups rrzcPolymorphicQuerySet._process_aggregate_args..test___lookup..tree_node_test___lookup s`"3t}}#56E $ a 0;%/#(a#8N:NN#84HeDErrNr)rirrcrrr)rmrrrr test___lookups @rrzBPolymorphicQuerySet._process_aggregate_args..test___lookupsr!Q E( A645)*)A)A)C9%(4%&789AFF*@,@@*rN)values)rrArBrmrrrs` @@@r_process_aggregate_argsz+PolymorphicQuerySet._process_aggregate_argssM A N A2 A !   A O rcF|j||t||i|S)ztranslate the polymorphic field paths in the kwargs, then call vanilla annotate. _get_real_instances will do the rest of the job after executing the query.)rrannotater@s rrzPolymorphicQuerySet.annotate"s* $$T62w000rcn|j|||j}tt||i|S)ztranslate the polymorphic field paths in the kwargs, then call vanilla aggregate. We need no polymorphic object retrieval for aggregate => switch it off.)rrWrr; aggregate)rrArBrVrs rrzPolymorphicQuerySet.aggregate(s; $$T62  ! ! #("7HHHrc4t||i|}d|_|S)NT)r_valuesr)rrArBrtrs rrzPolymorphicQuerySet._values2s#00%)" rc g}tt}tt}|jj}tj j |j}|j|jdj}|j|jdj}t|D]\} } | j|k(r|j| '| j} | j} | J| |k(r|jt| | k|j!| j#} || jt%| ||| j| t'|f|jd|j)D] \} } || }| j*j |jj,d i|d| i}|j.j0|j._g}|j2d}|D] } t5| |}|j|"tA||j.jBd f|j._!|Dcic]}t%|||}}|D]\} }|| } t%| |}|jE|}|,tGjF|}|j}|| k7r t||}|j.jHrB|j.jHjKD]}t%| |}tM||||j.jNrB|j.jNjKD]}t%| |}tM||||||< |D cgc]} | s|  }} |j.jHr;t|j.jHjK}|D] }||_( |j.jNr;t|j.jNjK}|D] }||_) |S#t6$rKd|vrB|j9dd} | j:j=|n#t>$rYYwxYwYwxYwcc}wcc} w) ai Polymorphic object loader Does the same as: return [ o.get_real_instance() for o in base_result_objects ] but more efficiently. The list base_result_objects contains the objects from the executed base class query. The class of all of them is self.model (our base model). Some, many or all of these objects were not created and stored as class self.model, but as a class derived from self.model. We want to re-fetch these objects from the db as their original class so we can return them just as they were created/saved. We identify these objects by looking at o.polymorphic_ctype, which specifies the real class of these objects (the class at the time they were saved). First, we sort the result objects in base_result_objects for their subclass (from o.polymorphic_ctype), and then we execute one db query per subclass of objects. Here, we handle any annotations from annotate(). Finally we re-sort the resulting objects into the correct order and return them as a list. F)for_concrete_modelTN__inrrr )*rrOrcpolymorphic_primary_key_namerobjects db_managerrd get_for_modelr enumeratepolymorphic_ctype_idr"get_real_instance_class#get_real_concrete_instance_class_idr9 get_for_id model_classgetattrrr3 _base_objectsr[queryselect_relatedr?r AssertionError rpartitionr get_fieldrr>deferred_loadinggetrE annotationskeyssetattr extra_selectpolymorphic_annotate_namespolymorphic_extra_select_names)rr% resultlistidlist_per_modelindexlist_per_modelpk_namecontent_type_managerself_model_class_idself_concrete_model_class_idr' base_objectreal_concrete_classreal_concrete_class_ididlistindices real_objectsdeferred_loading_fieldsexisting_fieldsfieldtranslated_field_name real_objectreal_objects_dictjo_pk real_classanno_field_nameattrselect_field_nameannotate_namesextra_select_namess rr$z'PolymorphicQuerySet._get_real_instances=s8 't,)$/**99 +22==dggF2@@ JJ5A " (<'I'I JJ4(J( " %((;< ,NA{//3FF!!+.&1&I&I&K#)4)X)X)Z&)1+/KK%%l3F &TU+?*I*I.+!km(%%89@@V]A^_'(;<CCQJDXY%%d+- ,:,<+A+A+CD , ' )*=>GW.<<GGPWWit$v.L15 0I0IL   -') #"??BO( F,L+U-)((../DE- F0+, ++A.3L   / O[!?J W-{:! !  ,11!4 {G4/33D9 &#ii 4 (@@B !44".z;"GK::))+/::+A+A+F+F+HD&{OD _dCD::**-1ZZ-D-D-I-I-KF)&{4EF ->EF!, 1 3 ,WD ,L",1Aqa1 1 :: ! !!$**"8"8"="="?@N) H 9G 6 H :: " "!%djj&=&=&B&B&D!E ) P =O : PM&~ 160@0@0G0K-%/55??@UV0%$% W.!>2sH% Q&;R=4S<S&!R:R$#R:$ R2 -R:1R2 2R:9R:c|jjr,djd|jD}d|dSt ||i|S)N, c32K|]}t|ywrrepr.0r(s r z/PolymorphicQuerySet.__repr__..s!>a$q'!>[  ])rc"polymorphic_query_multiline_outputjoinallr__repr__)rrArBresultrs rrzPolymorphicQuerySet.__repr__sM :: 8 8\\!>488:!>>Fxr? "7#T4V4 4rceZdZdZy)!PolymorphicQuerySet._p_list_classc>djd|D}d|dS)Nrc32K|]}t|ywrrrs rrz=PolymorphicQuerySet._p_list_class.__repr__..s!8a$q'!8rrr)r)rrArBrs rrz*PolymorphicQuerySet._p_list_class.__repr__s$\\!84!88Fxr? "rN)r*r+r,rrrr _p_list_classrs #rrc ||}|j|}|jjs|Stj |}|S)z Cast a list of objects to their actual classes. This does roughly the same as:: return [ o.get_real_instance() for o in base_result_objects ] but more efficiently. :rtype: PolymorphicQuerySet )r$rcrr;r)rr%olistclists rget_real_instancesz&PolymorphicQuerySet.get_real_instancessL k  &"& (()<=zz<<L#11%8 r)NFr)r*r+r,r-r1rDrL queryset_only classmethodrQrWrYr^rerkrorwrprxrrrrr$rrOrrr.r/s@rr;r;\s : $JZ(JX - 1 S.   XC&0d1 IiV5## rr;)r-rE collectionsr"django.contrib.contenttypes.modelsrdjango.core.exceptionsrdjango.db.modelsrdjango.db.models.queryrrr query_translater r r rr rr9r;rrrrsS #:4-==,/(,},^ (k(kr