f5ddlZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZGdd e ZGd d ZGd d ee ZGddZGddeejZGddZGddeejZGddZGddee ZGddZGddeejZGddej ZGd d!ZGd"d#eejZdS)$N)ObjectDoesNotExist) connectionmodels) LOOKUP_SEP) OneToOneField OneToOneRel) ModelIterableQuerySet)JoinceZdZdZdS)InheritanceIterablec #K|j}t|}t|ddrt|jj}t|jtd}|D]}d}|D]}| ||}|rn|s|}t|ddr)|j D]!}t||t||"|D]!}t||t||"|VdS|Ed{VdS)N subclassesFT)keyreverse _annotated) querysetr getattrtuplequeryextrakeyssortedrlen_get_sub_obj_recursersetattr) selfriterextrasrobjsub_objsks J/var/www/equiseq/venv/lib/python3.11/site-packages/model_utils/managers.py__iter__zInheritanceIterable.__iter__ sW=X&& 8\5 1 1 8>/446677F  3dKKKJ  #A&;;CCCG"!G8\599=%0==GCOO<<<<99AGQQ8888 !  $OOOOOOOOON)__name__ __module__ __qualname__r%r&r$r r s#r&r c`eZdZfdZdZfdZd fd ZfdZd dZd d Z d Z d Z xZ S)InheritanceQuerySetMixincRtj|i|t|_dSN)super__init__r _iterable_classrargskwargs __class__s r$r0z!InheritanceQuerySetMixin.__init__*s-$)&)))2r&c d}||j|}|s|}ng}|D]}||jur t|tfs|||}||vr||St d|d||}|r |j |}n|}||_ |S)Nlevelsz3{!r} is not in the discovered subclasses, tried: {}z, ) _get_subclasses_recursemodel isinstancestr_get_ancestors_pathappend ValueErrorformatjoinselect_relatedr)rrr8calculated_subclassesverified_subclassessubclassnew_qss r$select_subclassesz*InheritanceQuerySetMixin.select_subclasses.s$ $ < < Jv!=!'!' -.JJ"$ &  tz))!(SF331#77  8 1 1H444'..x8888$MTT$dii0E&F&FHH-J  (T(*5FFF& r&c i}dD]%}t||rt||||<&tjdi|}|j||S)Nrrr*)hasattrrr/_chain__dict__update)rr4rMnamechainedr5s r$rKzInheritanceQuerySetMixin._chainRst0 3 3DtT"" 3&tT22t  %''.**6**'''r&NFc t}dD]1}t||rt||t ||2|S)NrI)r/_clonerJrr)rklasssetupr4qsrNr5s r$rQzInheritanceQuerySetMixin._clone\sY WW^^  0 7 7DtT"" 7D'$"5"5666 r&ctj|i|}d|Dt|z|_|S)Ncg|] }|j Sr*) default_alias).0as r$ z5InheritanceQuerySetMixin.annotate..es999q1?999r&)r/annotatelistrr)rr3r4qsetr5s r$r[z!InheritanceQuerySetMixin.annotatecsMuww00099D999D.ns8+++![))+ +++r&cg|]M}t|jtr1t|jjr|jjuD|jK|NSr*)r;fieldr issubclassr: parent_link)rXrelr:s r$rZzDInheritanceQuerySetMixin._get_subclasses_recurse..rsl   #)]33 39?E22 SY_,,- -,,r&Nr7)_meta get_fieldsr9rbr:r>get_accessor_namer)rr:r8related_objectsrelsrrerEs ` r$r9z0InheritanceQuerySetMixin._get_subclasses_recursehs ++{--//+++    *      aKF 7 7C I $ < < !=!8!8IIH%%--//*.sH555(..K(M555r&)z OR )where)r>rArgparentsvaluesrGr)rr where_queriesr:s @r$ instance_ofzInheritanceQuerySet.instance_ofs   E  w||5555 %{299;; 555(("  "    &t%v.44FKK %2;/ / r&)r'r(r)r0rrr|r}s@r$rrs`r&rceZdZdS) QueryManagerNrr*r&r$rrrr&rceZdZdZdZdS)SoftDeletableQuerySetMixinzr QuerySet for SoftDeletableModel. Instead of removing instance sets its ``is_removed`` field to True. c2|ddS)zd Soft delete objects from queryset (set their ``is_removed`` field to True) T is_removedN)rMrs r$deletez!SoftDeletableQuerySetMixin.deletes t $$$$$r&N)r'r(r)__doc__rr*r&r$rrs- %%%%%r&rceZdZdS)SoftDeletableQuerySetNrr*r&r$rrrr&rc2eZdZdZeZddfd ZdZxZS)SoftDeletableManagerMixinzf Manager that limits the queryset by default to show only not removed instances of model. F)_emit_deprecation_warningscH||_tj|i|dSr.)emit_deprecation_warningsr/r0)rrr3r4r5s r$r0z"SoftDeletableManagerMixin.__init__s-)C&$)&)))))r&c |jr>d|jjj}t j|t|j|jd}t|dr |j |d<|j di| dS)zA Return queryset limited to not removed entries. a {0}.objects model manager will include soft-deleted objects in an upcoming release; please use {0}.available_objects to continue excluding soft-deleted objects. See https://django-model-utils.readthedocs.io/en/stable/models.html#softdeletablemodel for more information.r:using_hintshintsFrr*) rr@r:r5r'warningswarnDeprecationWarning_dbrJrrr)rwarning_messager4s r$rz&SoftDeletableManagerMixin.get_querysets  ) ?< fTZ)233  M/+= > > >:99 4 " " *"kF7O#t#--f--444FFFr&) r'r(r)rrrr0rr|r}s@r$rr sl,O9>*******GGGGGGGr&rceZdZdS)SoftDeletableManagerNrr*r&r$rr,rr&rceZdZdZddZdS) JoinQuerysetcp|\}}d|D}t|}||zS)NcHg|]}t|trd|dn| S)')r;r<)rXps r$rZz1JoinQueryset.get_quoted_query..6sH   !S!! )JJJJJ'(   r&)sql_with_paramsr)rrparamss r$get_quoted_queryzJoinQueryset.get_quoted_query2sN--// v     vv~r&Nc d |rfd|jjjD}|r|dnd}jjd}|pt |j|d}|st d |j n#t$rd |j j YnwxYw| }}n5d }jj }d |j}d t!| }t#j5}||dddn #1swxYwYG fd d t(j} t-| jj|jdd jjjd } |j| d|S)a Join one queryset together with another using a temporary table. If no queryset is used, it will use the current queryset and join that to itself. `Join` either uses the current queryset and effectively does a self-join to create a new limited queryset OR it uses a queryset given by the user. The model of a given queryset needs to contain a valid foreign key to the current queryset to perform a join. A new queryset is then created. idcHg|]}t|ddjk|S) related_modelN)rr:)rXfkrs r$rZz%JoinQueryset.join..Ps<255CCCCCr&rN_setz(QuerySet is not related to current model temp_stuffz DROP TABLE IF EXISTS {table_name}; DROP INDEX IF EXISTS {table_name}_id; CREATE TEMPORARY TABLE {table_name} AS {query}; CREATE INDEX {table_name}_{fk_column} ON {table_name} ({fk_column}); ) table_name fk_columnrcjeZdZejjejZGfddZdS)$JoinQueryset.join..TempModel) on_delete db_columnto_fieldceZdZdZZdS))JoinQueryset.join..TempModel.MetaFN)r'r(r)managedr) TABLE_NAMEsr$Metar}s%r&rN) r'r(r)r ForeignKeyr: DO_NOTHINGtemp_keyr)rrrrsr$ TempModelrusq(v(  +#! H & & & & & & & & & & & &r&rz INNER JOINF)r parent_alias table_alias join_type join_fieldnullable)reuse)r:rgfieldsr'lowerrr?columnAttributeErrorrbonlyobjectsallrrr@r<rcursorexecuterModelr rget_initial_alias tempmodel_setrerA)rrTr model_setrrFrsqlrrconnrrrs` @@@r$rAzJoinQueryset.joinAs  .X^2B&A$B:.4466<<rs555555((((((((111111FFFFFFFF::::::::444444-