o ӷ6ie@s`ddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z ddl mZddlmZddlmZdd lmZdd lmZdd lmZmZmZdd lmZdd lmZddl m!Z!dZ"GdddZ#Gddde$Z%e%dZ&GdddZ'GdddeZ(ddZ)Gddde j*e(dZ+ddZ,z ddl-Z-e,e+_,WdSe.yYdSw) N)reducewraps)settings)models) truncate_name) ModelBaseQ)DeferredAttribute)gettext)cached_field_value)TreeForeignKeyTreeManyToManyFieldTreeOneToOneField) TreeManager) node_moved)_get_tree_model) MPTTModel MPTTModelBase MPTTOptionsr rrrc@s&eZdZdddZddZddZdS) _classpropertyNcCs||_||_dSN)fgetfset)selfgettersetterrP/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/mptt/models.py__init__"s z_classproperty.__init__cCs ||Sr)r)rclsownerrrr__get__&s z_classproperty.__get__cCs|jstd|||dS)NzThis classproperty is read only)rAttributeError)rr r!valuerrr__set__)sz_classproperty.__set__r)__name__ __module__ __qualname__rr"r%rrrrr!s  rcseZdZdfdd ZZS)classpropertytyperNcs6|duri}t|d|d|d|dS)Nr"r% __delete____doc__)superrget)rnamebasesmembers __class__rrr0szclasspropertytype.__init__)rN)r&r'r(r __classcell__rrr1rr)/sr) classpropertyc@sbeZdZdZgZdZdZdZdZdZ ddd Z d d Z d d Z ddZ ddZddZddZdS)rz Options class for MPTT models. Use this as an inner class called ``MPTTMeta``:: class MyModel(MPTTModel): class MPTTMeta: order_insertion_by = ['name'] parent_attr = 'myparent' lftrghttree_idlevelparentNcKs|r t|jng}|t|ddd|Dvr!td|D]\}}|dddkr0q#t|||q#t|jtrD|jg|_dSt|jt rRt|j|_dS|jdur\g|_dSdS)Ntree_manager_attrcSsg|]}|dqS)rr).0optrrr Tsz(MPTTOptions.__init__..zu`tree_manager_attr` has been removed; you should instantiate a TreeManager as a normal manager on your model instead.__) list__dict__itemsextend ValueErrorsetattr isinstanceorder_insertion_bystrtuple)roptskwargskeyr$rrrrOs"     zMPTTOptions.__init__cCsdd|jDS)Ncss(|]\}}|ddkr||fVqdS)r_Nr)r;kvrrr hs&z'MPTTOptions.__iter__..)rArBrrrr__iter__gszMPTTOptions.__iter__cCs|j|}||S)z Gets the value of the given fieldname for the instance. This is not the same as getattr(). This function will return IDs for foreignkeys etc, rather than doing a database query. )_meta get_fieldvalue_from_object)rinstance field_namefieldrrrget_raw_field_valueks  zMPTTOptions.get_raw_field_valuecCs|j|}t||j|dS)z Sets the value of the given fieldname for the instance. This is not the same as setattr(). This function requires an ID for a foreignkey (etc) rather than an instance. N)rSrTrEattname)rrVrWr$rXrrrset_raw_field_valueus zMPTTOptions.set_raw_field_valuecCsi|_|jh}|jr!|jD]}|ddkr|dd}||q |}|D]$}|rB|j|}|j|vrB|j|jvrBt |j|<q'| |||j|<q'dS)a) Caches (in an instance._mptt_cached_fields dict) the original values of: - parent pk - fields specified in order_insertion_by These are used in save() to determine if the relevant fields have changed, so that the MPTT fields need to be updated. r-N) _mptt_cached_fields parent_attrrGaddget_deferred_fieldsrSrTrZrAr rY)rrV field_namesfdeferred_fieldsrWrXrrrupdate_mptt_cached_fields~s&          z%MPTTOptions.update_mptt_cached_fieldscCsg}g}|j}|j}tj}tj}|D]R} | ddkr#| dd} d} nd} t|| } | dur:|j| } | |d} | durD| dk} d} td i| | | i} |t |d d |D| g|| | fqt ||S) a Creates a filter which matches suitable right siblings for ``node``, where insertion should maintain ordering according to the list of fields in ``order_insertion_by``. For example, given an ``order_insertion_by`` of ``['field1', 'field2', 'field3']``, the resulting filter should correspond to the following SQL:: field1 > %s OR (field1 = %s AND field2 > %s) OR (field1 = %s AND field2 = %s AND field3 > %s) rr\r]N__lt__gtT__isnullcSs"g|] \}}tdi||iqS)rr)r;rcrOrrrr=s"z8MPTTOptions.insertion_target_filters..r) appendoperatorand_or_getattrrSrTpre_saver r)rrVrGfieldsfiltersfields__appendfilters__appendrkrlrW filter_suffixr$rXqrrrinsertion_target_filterss,      z$MPTTOptions.insertion_target_filterscCsd}|jru|dus|dkru|j}|jdd}|||}|r4|tdi|j|i@}||jn|tdi|jdi@}||j|j j |j j |j|}|jr_|j|jd}z |ddd}W|StytY|Sw|S)aV Attempts to retrieve a suitable right sibling for ``node`` underneath ``parent`` (which may be ``None`` in the case of root nodes) so that ordering by the fields specified by the node's class' ``order_insertion_by`` option is maintained. Returns ``None`` if no suitable sibling can be found. Nrpkr]r)rGget_descendant_count _mptt_metarur r_ri left_attr tree_id_attrr2 _tree_manager db_manager_statedbfilterorder_byrwexclude IndexError)rnoder9 right_siblingrJrrpquerysetrrrget_ordered_insertion_targets4    z(MPTTOptions.get_ordered_insertion_targetr)r&r'r(r+rGrz right_attrr{ level_attrr_rrRrYr[rerurrrrrr>s    # -rcs,eZdZdZfddZeddZZS)rz# Metaclass for MPTT models c sJ|dkr|ikrt||||Sd}ztWn ty"d}Ynw|dd}|s2Gddd}tt|}|D]}t|drX|jD]\}} |dkrMqD||vrWt ||| qDq:t ||d<tj} | ||||} | | } d| _ |rx| g}n d d | D}|D]}|jjs|jjs|jj|ur|| _ nq| | j urt| _| S) z Create subclasses of MPTTModel. This: - adds the MPTT fields to the class - adds a TreeManager to the model NewBaseFTMPTTMetaNc@s eZdZdS)z'MPTTModelBase.__new__..MPTTMetaN)r&r'r(rrrrrsryr:cSsg|] }t|tr|qSr) issubclassr)r;baserrrr='sz)MPTTModelBase.__new__..)r,__new__r NameErrorpop frozensetdirhasattrryrErregister_mptt_tracking_basemrorSabstractproxyr| tree_model threadinglocal _threadlocal) meta class_namer/ class_dict is_MPTTModelrinitial_optionsrr.r$ super_newr r1rrrsP          zMPTTModelBase.__new__c Ks^t|tjs ttdt|dstdi||_t|j dd}zt Wn t y.Y|Swt|t s[t |j }tt|dddD] }tt ||rO||=qC|dt t||_ t||u}|rt}|D]}t|dr}|d d |j jDqj|j} | jf} | j| j| j| jf} | D]} | |vrtj| | vdd } | || q|jj|jjf}|j jD] }t|j|krn&q|j jtj |t!|j j"d |j j#d |jjd |jjd tj j$d|s-d}|j%rt&|j%t'r|j%}n|j j(D]}t&|t'r |j)|ur |}nq|r|j)|ur|*|}n|dur$t'}||d||_+|S)z For the weird cases when you need to add tree-ness to an *existing* class. For other cases you should subclass MPTTModel instead of calling this. z0register() expects a Django model class argumentryrFr]rrScSg|]}|jqSr)r.)r;rcrrrr=ez*MPTTModelBase.register..)db_indexeditablerM_idx)ror.Nr|r),rrModelrDrMrrryrmrSrrr@ __bases__rangeleninsertrIrsetrupdate local_fieldsr{rzrrPositiveIntegerFieldcontribute_to_classindexesroriIndexr app_label model_namemax_name_length_default_managerrFrmanagersmodel_copy_to_modelr|)rr rKrr/iis_cls_tree_modelexisting_field_namesr mptt_meta indexed_attrsrbrWrX index_fieldsindex tree_manager cls_managerrrrr6s    X                zMPTTModelBase.register)r&r'r(r+r classmethodrr3rrr1rrs  8rcstfdd}|S)Ncs6|jjrtdj|jjd|g|Ri|S)Nz7Cannot call %(function)s on unsaved %(class)s instances)functionclass)r~addingrDr&r2rargsrKfuncrr_fnszraise_if_unsaved.._fn)r)rrrrrraise_if_unsavedsrcseZdZdZGdddZeZfddZddZe dd Z e d d Z e d d Z e ddZe ddZe ddZe ddZedHddZeddZeddZedIddZdd ZedId!d"Zed#d$Zed%d&Zed'd(ZedId)d*Zd+d,Z -   .dJd/d0Zd1d2Zd3d4Z d5d6Z!edId7d8Z"edId9d:Z#dKd;d<Z$dLd>d?Z%d@dAZ&fdBdCZ'd.e'_(fdDdEZ)d.e)_(dFdGZ*Z+S)Mrz% Base class for tree models. c@seZdZdZdS)zMPTTModel.MetaTN)r&r'r(rrrrrMetasrcs4t|dr |tj|i||j|dS)N_check_no_testing_generators)rrr,rryrerr1rrrs zMPTTModel.__init__cCst|j|d}t||S)N_attr)rmry)r fieldnametranslated_fieldnamerrr _mpttfields zMPTTModel._mpttfieldcCs|jsdSt|jjddS)NTmptt_updates_enabled)rrmrr rrr_mptt_updates_enableds  zMPTTModel._mptt_updates_enabledcCs||jus Jd||j_dS)Nz=Can't enable or disable mptt updates on a non-tracking class.)rrr)r r$rrr_set_mptt_updates_enableds  z#MPTTModel._set_mptt_updates_enabledcCs*|jsdSt|jdsd|j_|jjduS)NFmptt_delayed_tree_changes)rrrrrrrr_mptt_is_trackings   zMPTTModel._mptt_is_trackingcCs.||jus Jd|jrJdt|j_dS)N:Can't start or stop mptt tracking on a non-tracking class.z!mptt tracking is already started.)rrrrrrrrr_mptt_start_trackings zMPTTModel._mptt_start_trackingcCs4||jus Jd|jsJd|jj}d|j_|S)Nrzmptt tracking isn't started.)rrrr)r resultsrrr_mptt_stop_trackings zMPTTModel._mptt_stop_trackingcCs|jsdS|jj|dSr)rrrr`)r r7rrr_mptt_track_tree_modifiedsz#MPTTModel._mptt_track_tree_modifiedcs`|jsdS|jj}r |sdSdkr t }||fdd|D}||j_dS)Nrcs h|] }|kr |n|qSrr)r;t num_insertedr7rr s z8MPTTModel._mptt_track_tree_insertions..)rrrrdifference_update)r r7rchangesdeleted new_changesrrr_mptt_track_tree_insertionss  z%MPTTModel._mptt_track_tree_insertionsFc Cs|j}|r|s|jS|jj|jd}n0|j}|r d|}t||j}t||j}|s6|d8}|d7}|jj ||| dd}| |}t |drrg}|} |sXt| |j } | durk|| t| |j } | dus\|||_|S)a Creates a ``QuerySet`` containing the ancestors of this model instance. This defaults to being in descending order (root ancestor first, immediate parent last); passing ``True`` for the ``ascending`` argument will reverse the ordering (immediate parent first, root ancestor last). If ``include_self`` is ``True``, the ``QuerySet`` will also include this model instance. rvr\r]r7) left__lte right__gter7_mptt_use_cached_ancestorsN)ry is_root_noder|nonerrwrzrmr _mptt_filterrrrr_rireverse _result_cache) r ascending include_selfrJqsrleftright ancestorsprrr get_ancestors s>        zMPTTModel.get_ancestorsc Cs|j}t||j}t||j}tdid|j|d|j||j|di}tdid|j|d|j||j|di}|j||BS)z Returns a ``QuerySet`` containing the ancestors, the model itself and the descendants, in tree order. z%s__ltez%s__gter7Nr) ryrmrzrr r{rr|r)rrJrrr descendantsrrr get_familyDs         zMPTTModel.get_familycCsPt|dr|jjdd|jDd}|j|_|S|r!|jS|jj|dS)a Returns a ``QuerySet`` containing the immediate children of this model instance, in tree order. The benefit of using this method over the reverse relation provided by the ORM to the instance's children is that a database query can be avoided in the case where the instance is a leaf node (it has no children). If called from a template where the tree has been walked by the ``cache_tree_children`` filter, no database query is required. _cached_childrencSrrrv)r;nrrrr=prz*MPTTModel.get_children..)pk__inr9)rr|rrr is_leaf_noderr)rrrrr get_childrenas  zMPTTModel.get_childrencCsp|r|s |jS|jj|jdS|j}t||j}t||j}|s,|d7}|d8}|jj | d||dS)z Creates a ``QuerySet`` containing descendants of this model instance, in tree order. If ``include_self`` is ``True``, the ``QuerySet`` will also include this model instance. rvr]r7)r7 left__gter) rr|rrrwryrmrzrrr)rrrJrrrrrget_descendantsys     zMPTTModel.get_descendantscCs.|ddur dS|d|dddS)zL Returns the number of descendants this model instance has. rNrrr]r>)rrQrrrrxszMPTTModel.get_descendant_countcCs*|j|d}|jj|t|jjddS)z Creates a ``QuerySet`` containing leafnodes of this model instance, in tree order. If ``include_self`` is ``True``, the ``QuerySet`` will also include this model instance (if it is a leaf node) )rr])r)rr|rrFryr)rrrrrr get_leafnodess zMPTTModel.get_leafnodescOsv|jj|i|}|r|jj|d|dd}n|jj|t||jjd|dd}|dd}|r9|dp:dS) z Returns this model instance's next sibling in the tree, or ``None`` if it doesn't have a next sibling. Nr7)r9 tree_id__gt_idr) parent__pkleft__gtr]r)r|rrrrrmryr_)r filter_args filter_kwargsrsiblingsrrrget_next_siblings zMPTTModel.get_next_siblingcOs|j}|jj|i|}|r%|jj|d|dd}|d|j}n|jj|t||j d|dd}|d|j }|dd}|rK|d pLdS) z Returns this model instance's previous sibling in the tree, or ``None`` if it doesn't have a previous sibling. Nr7)r9 tree_id__ltr\rr)r right__ltr]r) ryr|rrrrrr{rmr_r)rr r rJrr rrrget_previous_siblings" zMPTTModel.get_previous_siblingcCs6|rt||jjkr|S|jj|dddS)zF Returns the root node of this model instance's tree. r7N)r7r9)rtyper|rrrr-rQrrrget_rootszMPTTModel.get_rootcCsN|r |jjdd}nt||jjd}|jj|d}|s%|j|jd}|S)a  Creates a ``QuerySet`` containing siblings of this model instance. Root nodes are considered to be siblings of other root nodes. If ``include_self`` is ``True``, the ``QuerySet`` will also include this model instance. Nrr)rrv)rr|rrmryr_rrw)rrr parent_idrrr get_siblingss zMPTTModel.get_siblingscCst||jjS)zE Returns the level of this node (distance from root) )rmryrrQrrr get_levelszMPTTModel.get_level first-childTcCs|jj||||||ddS)zn Convenience method for calling ``TreeManager.insert_node`` with this model instance. allow_existing_pkrefresh_targetN)r| insert_node)rtargetpositionsaverrrrr insert_ats  zMPTTModel.insert_atcC | S)zg Returns ``True`` if this model instance is a child node, ``False`` otherwise. )rrQrrr is_child_node zMPTTModel.is_child_nodecCr)z{ Returns ``True`` if this model instance is a leaf node (it has no children), ``False`` otherwise. )rxrQrrrr r zMPTTModel.is_leaf_nodecCst||jjdduS)zf Returns ``True`` if this model instance is a root node, ``False`` otherwise. rN)rmryr_rQrrrr'szMPTTModel.is_root_nodecCsn|j}|r |j|jkr dSt||jt||jkrdSt||j}t||j}|t||jko6|t||jkS)z Returns ``True`` if this model is a descendant of the given node, ``False`` otherwise. If include_self is True, also returns True if the two nodes are the same node. TF)ryrwrmr{rzr)rotherrrJrrrrris_descendant_of.s  zMPTTModel.is_descendant_ofcCs|r |j|jkr dS||S)z Returns ``True`` if this model is an ancestor of the given node, ``False`` otherwise. If include_self is True, also returns True if the two nodes are the same node. T)rwr")rr!rrrris_ancestor_ofDs zMPTTModel.is_ancestor_ofcCs|j|||dS)a! Convenience method for calling ``TreeManager.move_node`` with this model instance. NOTE: This is a low-level method; it does NOT respect ``MPTTMeta.order_insertion_by``. In most cases you should just move the node yourself by setting node.parent. N)r| move_node)rrrrrrmove_toOszMPTTModel.move_toNcCsh|jdus |ddurdS|j}|jjdurdSt|ds1|jj}||}|j|jd |_ |j S)Nr7FT _mptt_savedrv) rwrrS remote_fieldrr2 _base_managerusingrexistsr&)rr)rJmanagerrrr _is_savedYs   zMPTTModel._is_savedcCsdddlm}g}|jj|jj|jj|jjf}|jjD]}|j |vr/t ||s/|j s/| |j q|S)zFReturns the list of user defined (i.e. non-mptt internal) field names.r) AutoField) django.db.models.fieldsr-ryrzrr{rrSconcrete_fieldsr.rF primary_keyri)rr-rbinternal_fieldsrXrrr_get_user_field_namesfs    zMPTTModel._get_user_field_namesc sh|jj}|jj}|j}|sx|sx|ddurot||j}|rS|dd}t||j|t||j |dt||j |ddt||j |d|j |dnt||jdt||j dt||j dt||j dtj|i|S|||j}|dd } |d d } d} |} | s| s4|j|d d r4|j|j} | turd }n| |k}|rt|jdkr|jD]\}}|tur|| vrd }n ||||krd }nq|s|sd }|j|dnM|s9|s9| dur9|d} t||j}|d}|dd}|j|t||j |t||j|t||j |dt||j |ddd }|st||j}|||j| z|||}|dur|o~t||j t||j kp~t||jt||jkp~t||j t||j k}|r|j j||dd d dn;|dur|j }z|j|jd d|j d}|j j||dd d dWnt!yYn w|j j||dd d|dur|rd|"d}|j ||W|||j|n |||j|wt#j$|j|t||jdn|||j||s3t|dkr'|ds&t%|}|&|d<t'|}n]|ds3|&|d<nPt||jrCt||j rCnAt||j}d}|sS|dur]|j(r]|||}|r||j)|dd d d|r{d|"d}|j ||n|j)|dd dztj|i|W| dur|j *| dn| dur|j *| dwwd |_+|,|dS)aO If this is a new node, sets tree fields up before it is inserted into the database, making room in the tree structure as necessary, defaulting to making the new node the last child of its parent. It the node's left and right edge indicators already been set, we take this as indication that the node has already been set up for insertion, so its tree fields are left untouched. If this is an existing node and its parent has been changed, performs reparenting in the tree structure, defaulting to making the node the last child of its new parent. In either case, if the node's class has its ``order_insertion_by`` tree option set, the node will be inserted or moved to the appropriate position to maintain ordering by the specified field. rNr]r8r7r>r force_updateF force_insertr))r)T)rrrvr\rz last-child)r)senderrVr update_fieldsr)rrr)-r2rrryrr r_rErzrrr{r|'_post_insert_update_cached_parent_rightr,rrYr-rar,r^r rrBrrmr[r _move_node root_nodesrrwrrrxrsendr@r2rIrGr_create_tree_spacer&re)rrrK do_updates track_updatesrJr9rrr3r4collapse_old_treerd old_parent_id same_orderrW old_valuer7rupdate_cached_parentr:rightmost_sibling right_shiftr1rrrzsB                    $         zMPTTModel.savec sz|jj|jj|jjg}|j|dWn |jjyYnw|d|dd}|d}|d}|j |||t ||jj }|rT| d}|j ||tj|i|S)aJCalling ``delete`` on a node will delete it as well as its full subtree, as opposed to reattaching all the subnodes to its parent node. There are no argument specific to a MPTT model, all the arguments will be passed directly to the django's ``Model.delete``. ``delete`` will not return anything.)rorrr]r7r>)ryrrzr{refresh_from_dbr2 DoesNotExistrr| _close_gapr r_rxr8r,delete) rrrKfields_to_refresh tree_width target_rightr7r9rEr1rrrI[s(  zMPTTModel.deletecCsl|jsdSt|j}|j}||jjj|jd|j |j |j |j d}| D] \}}t|||q)dS)Nrvr)rwrr|ryr)r~rrvaluesrzrrr{rBrE)rr+rJrMrNrOrrr _mptt_refreshs&   zMPTTModel._mptt_refresh)FF)F)rFFT)rr),r&r'r(r+rrobjectsrrrrrrrrrrrrrrrrrxrr rrrrrrrrr"r#r%r,r2r alters_datarIrNr3rrr1rrsz         :              ` #r) metaclasscCsrtjdddgkr3t}t|d}|dd}|dd}d|vr5ttd d s7td |d d Sd Sd S)z9Check that we are not generationg model from model_bakeryr]r>testr/ model_bakeryMPTT_ALLOW_TESTING_GENERATORSFzThe z populates django-mptt fields with random values which leads to unpredictable behavior. If you really want to generate this model that way, please set MPTT_ALLOW_TESTING_GENERATORS=True in your settings.N) sysargvinspect currentframegetouterframessplitrmr Exception)rcurframe call_frame call_filecall_directoryrrrrs    r)/rZrjrXr functoolsrr django.confr django.dbrdjango.db.backends.utilsrdjango.db.models.baserdjango.db.models.queryr django.db.models.query_utilsr django.utils.translationr rM mptt.compatr mptt.fieldsr rr mptt.managersr mptt.signalsr mptt.utilsr__all__rpropertyr)r4rrrrrrrV ImportErrorrrrrsN             </ d