h+dZddlZddlmZddlmZddlmZddlm Z m Z ddl m Z ddl mZdd lmZmZdd lmZefd Zefd Zefd ZefdZdZdZdZdZdefdZdZy)z' PolymorphicQuerySet support functions N)deque)apps) ContentType)FieldDoesNotExist FieldError)models)Q)ForeignObjectRel RelatedField)DEFAULT_DB_ALIAScg}|jjD]b\}}t||||}t|tr||=|d||d<4t|t j sO||=|j|d|S)a8 Translate the keyword argument list for PolymorphicQuerySet.filter() Any kwargs with special polymorphic functionality are replaced in the kwargs dict with their vanilla django equivalents. For some kwargs a direct replacement is not possible, as a Q object is needed instead to implement the required functionality. In these cases the kwarg is deleted from the kwargs dict and a Q object is added to the return list. Modifies: kwargs dict Returns: a list of non-keyword-arguments (Q objects) to be added to the filter() query. usingr)copyitems(_translate_polymorphic_filter_definition isinstancetuplerr append)queryset_modelkwargsradditional_args field_pathvalnew_exprs K/home/dcms/DCMS/lib/python3.12/site-packages/polymorphic/query_translate.py2translate_polymorphic_filter_definitions_in_kwargsrs O!;;=..0 - C; J5  h &z""*1+F8A;  &(( +z"  " "8 , - c\fdt|tjr |||S)Nctt|jD]W}|j|}t|tt fr'|\}}t |||}|s?||j|<O||Yy)z#process all children of this Q noderN)rangelenchildrenrrlistr) my_modelnodeichildkeyrrtree_node_correct_field_specsrs rr+zEtranslate_polymorphic_Q_object..tree_node_correct_field_specs<sws4==)* ?AMM!$E%%/ SCc3e'/DMM!$.h> ?r)rrr )rpotential_q_objectrr+s `@rtranslate_polymorphic_Q_objectr-;s*?"$fhh/%n6HI rc j|Dcgc]#}t|tj||%c}Scc}w)aE Translate the non-keyword argument list for PolymorphicQuerySet.filter() In the args list, we return all kwargs to Q-objects that contain special polymorphic functionality with their vanilla django equivalents. We traverse the Q object tree for this (which is simple). Returns: modified Q objects r)r-rdeepcopy)rargsrqs r0translate_polymorphic_filter_definitions_in_argsr2Ss6`d Z[&~t}}Q7GuU  s(0cv|dk(r t||S|dk(rt|d|Sd|vryt||}||fS)a Translate a keyword argument (field_path=field_val), as used for PolymorphicQuerySet.filter()-like functions (and Q objects). A kwarg with special polymorphic functionality is translated into its vanilla django equivalent, which is returned, either as tuple (field_path, field_val) or as Q object. Returns: kwarg tuple or Q object or None (if no change is required) instance_ofrnot_instance_ofT)r5r___N)create_instanceof_q translate_polymorphic_field_path)rr field_valrnewpaths rrrcsU ]""9E:: ( ("9d%PP j /~zJG Y rc t|tstd||jd\}}}|s|S|s Jd|dd}|ddk(rd}|j d}d |vr~|jd \}}}t j ||}|sJd |jd |d t||sd |jzdz|jzdz}t| |jj|} t| ttfr|S t|} | j!|d}|sJd |d|jdt#||} |rd} nd} | | z } | r| d z } | |z } | S#t$rYhwxYw)am Translate a field path from a keyword argument, as used for PolymorphicQuerySet.filter()-like functions (and Q objects). Supports leading '-' (for order_by args). E.g.: if queryset_model is ModelA, then "ModelC___field3" is translated into modela__modelb__modelc__field3. Returns: translated path (unchanged, if no translation needed) zExpected field name as string: r6zPolymorphicModel: z: bad field specificationFr-T__zPolymorphicModel: model z (in app z ) not found!z*PolymorphicModel: queryset filter error: "z" is not derived from ""Nz not found (not a subclass of z)!)rstr ValueError partitionlstripr get_model__name__ issubclassAssertionError_meta get_fieldr r r_get_all_sub_modelsget_create_base_path) rr classnameseppure_field_pathnegatedappnamemodelefield submodelsbasepathr:s rr8r8s j# &::,GHH&0&:&:5&A#IsO  P*:,6OPP9G|s$$S)  y"+"5"5d";iw 2_00@ 'R^__u%0<..!+,!))*  !# # "((229=E%,0@!AB"!C(7  i. &yk1OP^PgPgOhhj k u!7H xG4 G N+!   s%2E66 FFc2i}t|g}|r|j}t|tjr|tjk7r|j |vr|||j k7rv|j jd|j }||j j jd||j j }td|d|d|||j <|j|j|r|S)z:#Collect all sub-models, this should be optimized (cached).z1PolymorphicModel: model name alone is ambiguous: z and zH match! In this case, please use the syntax: applabel__ModelName___field) rpopleftrFrModelrErH app_labelrextend__subclasses__) base_modelresultqueuerRname1name2s rrJrJs  F :, E   eV\\ *u /D~~'EVENN5K,K ;;0015>>2BCenn-33==>au~~@V@_@_?`a!GweTYSZ[WX &+F5>> " U))+, " Mrc|jD]m}||k(r t|cSt||}|s$|jjs|jj r t|cS|dt|cSy)Nr=r?) __bases___get_query_related_namerLrHabstractproxy) baseclassmyclassbpaths rrLrLs{   E >*73 3 A. ww177==.w77r"9'"B!CDD E rc|jjD]E}t|tjs|j j s5|jcS|jjS)N) rH local_fieldsrr OneToOneField remote_field parent_linkrelated_query_namerElower)rifs rreres] ]] ' '* a-- .1>>3M3M'') )*    ! ! ##rFc|syt|ttfs!ddlm}t ||r|g}n t dt||}tt|}|r|}|S)a Helper function for instance_of / not_instance_of Creates and returns a Q object that filters for the models in modellist, including all subclasses of these models (as we want to do the same as pythons isinstance() ). . We recursively collect all __subclasses__(), create a Q filter for each, and or-combine these Q objects. This could be done much more efficiently however (regarding the resulting sql), should an optimization be needed. Nr)PolymorphicModelzdPolymorphicModel: instance_of expects a list of (polymorphic) models or a single (polymorphic) model)polymorphic_ctype__in) rr%rrrurF TypeError_get_mro_content_type_idsr sorted) modellistr5rrucontenttype_idsr1s rr7r7sk  i$ /, i!1 2" I9  0 5AO  78A B Hrct}|D]{}tjj|j |d}|j |j |j}|sa|jt||}|S)NF)for_concrete_model) setrobjects db_manager get_for_modeladdpkr]updaterx)rrr{rRct subclassess rrxrx seOQ  + +E 2 @ @[` @ aBEE"))+   " "#rs~ :@J,#3DN^0Rb "2B :IX2  $49@P Br