a xd6@sJddlZddlmZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZGdd d e ZGd d d ZGd d d ee ZGdddZGdddeejZGdddZGdddeejZGdddZGdddee ZGdddZGdddeejZGdddej ZGd d!d!ZGd"d#d#eejZdS)$N)ObjectDoesNotExist) connectionmodels) LOOKUP_SEP) OneToOneField OneToOneRel) ModelIterableQuerySet)Joinc@seZdZddZdS)InheritanceIterablec cs|j}t|}t|ddrt|jj}t|jt dd}|D]z}d}|D]}| ||}|rJqdqJ|sl|}t|ddr|j D]}t ||t||q~|D]}t ||t||q|Vq>n |EdHdS)N subclassesFT)keyreverse _annotated) querysetrgetattrtuplequeryextrakeyssortedr len_get_sub_obj_recursersetattr) selfriterextrasr objZsub_objskr L/var/www/html/Ranjet/env/lib/python3.9/site-packages/model_utils/managers.py__iter__ s(     zInheritanceIterable.__iter__N)__name__ __module__ __qualname__r"r r r r!r sr csneZdZfddZddZfddZdfd d Zfd d Zdd dZdddZ ddZ ddZ Z S)InheritanceQuerySetMixincstj|i|t|_dSN)super__init__r Z_iterable_classrargskwargs __class__r r!r)*sz!InheritanceQuerySetMixin.__init__c Gsd}|j|j|d}|s|}nbg}|D]T}||jur6q&t|tfsP|j||d}||vrd||q&td|d|q&|}|j j }|r|j |}n|}t|t }t|j j t } |r| r|j j |||_ |S)Nlevelsz3{!r} is not in the discovered subclasses, tried: {}z, )_get_subclasses_recursemodel isinstancestr_get_ancestors_pathappend ValueErrorformatjoinrZselect_relateddictupdater ) rr r0Zcalculated_subclassesZverified_subclassessubclassZprevious_select_relatednew_qsZprevious_is_dictZ new_is_dictr r r!select_subclasses.s@      z*InheritanceQuerySetMixin.select_subclassesc sHi}dD]}t||rt||||<qtjfi|}|j||SN)r r)hasattrrr(_chain__dict__r;)rr,r;nameZchainedr-r r!rAXs  zInheritanceQuerySetMixin._chainNFc s4t}dD] }t||rt||t||q|Sr?)r(_cloner@rr)rklasssetupr,qsrCr-r r!rDbs   zInheritanceQuerySetMixin._clonecs2tj|i|}dd|Dt||_|S)NcSsg|] }|jqSr )Z default_alias).0ar r r! kz5InheritanceQuerySetMixin.annotate..)r(annotatelistrr)rr+r,Zqsetr-r r!rLisz!InheritanceQuerySetMixin.annotatecsddjD}fdd|D}g}|r6|d8}|D]L}|sJ|durx|j|jj|dD]}||t|q\||q:|S)z Given a Model class, find all related objects, exploring children recursively, returning a `list` of strings representing the relations for select_related cSsg|]}t|tr|qSr )r3r)rHfr r r!rJts zDInheritanceQuerySetMixin._get_subclasses_recurse..cs<g|]4}t|jtrt|jjr|jjur|jr|qSr )r3fieldr issubclassr2 parent_link)rHrelr2r r!rJxs   Nr/)_metaZ get_fieldsr1rOr2r6get_accessor_namer)rr2r0Zrelated_objectsZrelsr rRr<r rSr!r1ns&   z0InheritanceQuerySetMixin._get_subclasses_recursecCst||jstd||jg}|j|j}|r<|d8}|dur|j}|d||sf|dur||j}|j|j}qgetr*r r r! get_subclasssz%InheritanceQuerySetMixin.get_subclass)NF)N)N) r#r$r%r)r>rArDrLr1r5rr^ __classcell__r r r-r!r&)s *     r&c@seZdZddZdS)InheritanceQuerySetc sVg}|D]2|ddfddjjDdq|j|jd|gdS)zQ Fetch only objects that are instances of the provided model(s). (z AND csg|]}djj|jqS)z"{}"."{}" IS NOT NULL)r8rUdb_tableZattname)rHrOrSr r!rJs z3InheritanceQuerySet.instance_of..)z OR )where)r6r9rUparentsvaluesr>r)rrZ where_queriesr rSr! instance_ofs zInheritanceQuerySet.instance_ofN)r#r$r%rgr r r r!r`sr`c@s0eZdZeZddZddZddZddZd S) InheritanceManagerMixincCs ||jSr')_queryset_classr2rr r r! get_querysetsz$InheritanceManagerMixin.get_querysetcGs|j|Sr')rkr>)rr r r r!r>sz)InheritanceManagerMixin.select_subclassescOs|j|i|Sr')rkr^r*r r r!r^sz$InheritanceManagerMixin.get_subclasscGs|j|Sr')rkrg)rrr r r!rgsz#InheritanceManagerMixin.instance_ofN) r#r$r%r`rirkr>r^rgr r r r!rhs rhc@s eZdZdS)InheritanceManagerNr#r$r%r r r r!rlsrlcs0eZdZfddZddZfddZZS)QueryManagerMixincs6|r|d|_ntjfi||_d|_tdS)Nr)_qrQ _order_byr(r)r*r-r r!r)s  zQueryManagerMixin.__init__cGs ||_|Sr')rq)rr+r r r!order_byszQueryManagerMixin.order_bycs,t|j}|jdur(|j|jS|Sr')r(rkfilterrorqrr)rrGr-r r!rks  zQueryManagerMixin.get_queryset)r#r$r%r)rrrkr_r r r-r!rns rnc@s eZdZdS) QueryManagerNrmr r r r!rtsrtc@seZdZdZddZdS)SoftDeletableQuerySetMixinzr QuerySet for SoftDeletableModel. Instead of removing instance sets its ``is_removed`` field to True. cCs|jdddS)zd Soft delete objects from queryset (set their ``is_removed`` field to True) TZ is_removedN)r;rjr r r!deletesz!SoftDeletableQuerySetMixin.deleteN)r#r$r%__doc__rwr r r r!rusruc@s eZdZdS)SoftDeletableQuerySetNrmr r r r!ry srycs2eZdZdZeZddfdd ZddZZS)SoftDeletableManagerMixinzf Manager that limits the queryset by default to show only not removed instances of model. F)_emit_deprecation_warningscs||_tj|i|dSr')emit_deprecation_warningsr(r))rr{r+r,r-r r!r)sz"SoftDeletableManagerMixin.__init__cCs\|jr"d|jjj}t|t|j|jd}t |drD|j |d<|j fi|j ddS)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.r2Zusing_hintshintsFrv) r|r8r2r.r#warningswarnDeprecationWarning_dbr@r~rirs)rZwarning_messager,r r r!rks   z&SoftDeletableManagerMixin.get_queryset) r#r$r%rxryrir)rkr_r r r-r!rzsrzc@s eZdZdS)SoftDeletableManagerNrmr r r r!r2src@seZdZddZdddZdS) JoinQuerysetcCs*|\}}dd|D}t|}||S)NcSs$g|]}t|trd|n|qS)z'{}')r3r4r8)rHpr r r!rJ<s z1JoinQueryset.get_quoted_query..)Zsql_with_paramsr)rrparamsr r r!get_quoted_query8s  zJoinQueryset.get_quoted_queryNc s`d|rfdd|jjjD}|r,|dnd}djj}|pRt|j|d}|s`tdz |jWnt yd|j jYn0| }}nd }jj }d|j}d jt|d }t}||Wdn1s0YGfd d d tj} t| jj|jdd jjjdd} |jj| dd|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 querset 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. idcs"g|]}t|ddjkr|qS)Z related_modelN)rr2)rHfkrjr r!rJVsz%JoinQueryset.join..rNz{}_setz(QuerySet is not related to current modelZ 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_columnrcs4eZdZejjejdZGfdddZdS)z$JoinQueryset.join..TempModel)Z on_deleteZ db_columnto_fieldcseZdZdZZdS)z)JoinQueryset.join..TempModel.MetaFN)r#r$r%Zmanagedrbr ) TABLE_NAMEr r!MetasrN) r#r$r%rZ ForeignKeyr2Z DO_NOTHINGZtemp_keyrr rrrrr r! TempModel{srz INNER JOINF)rZ parent_aliasZ table_aliasZ join_typeZ join_fieldZnullable)Zreuse)r2rUfieldsr8r#lowerrr7columnAttributeErrorrOonlyZobjectsallrrr4rcursorexecuterZModelr rbZget_initial_aliasZ tempmodel_setrRr9) rrGrZ model_setr r=rZsqlrrconnr rr!r9GsL          * zJoinQueryset.join)N)r#r$r%rr9r r r r!r6src@seZdZdZeZddZdS)JoinManagerMixinze Manager that adds a method join. This method allows you to join two querysets together. cCs|j|j|jdS)Nr})rir2rrjr r r!rkszJoinManagerMixin.get_querysetN)r#r$r%rxrrirkr r r r!rsrc@s eZdZdS) JoinManagerNrmr r r r!rsr)rZdjango.core.exceptionsrZ django.dbrrZdjango.db.models.constantsrZdjango.db.models.fields.relatedrrZdjango.db.models.queryrr Z#django.db.models.sql.datastructuresr r r&r`rhManagerrlrnrtruryrzrrrrr r r r!s*   !]