o ӷ6io@sdZddlmZddlmZddlmZddlmZm Z ddl m Z ddl mZddlmZd ZGd d d ZGd d d eejZGdddeejZGdddejZGdddejZGdddejZdS)z) Form components for working with trees. )forms)NON_FIELD_ERRORS) smart_str)conditional_escape mark_safe) gettext_lazy) InvalidMove)DEFAULT_LEVEL_INDICATOR) MoveNodeFormTreeNodeChoiceFieldTreeNodeMultipleChoiceFieldTreeNodePositionFieldcs4eZdZfddZddZddZddZZS) TreeNodeChoiceFieldMixincs\|dt|_|dd|_t|dr|jj}||j|j }t j |g|Ri|dS)Nlevel_indicator start_levelrmodel) popr rrhasattrr _mptt_metaorder_by tree_id_attr left_attrsuper__init__)selfquerysetargskwargs mptt_opts __class__O/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/mptt/forms.pyrs  z!TreeNodeChoiceFieldMixin.__init__cCst||jj}||jS)N)getattrr level_attrrrobjlevelr!r!r"_get_relative_level%s z,TreeNodeChoiceFieldMixin._get_relative_levelcCs"||}tt|jtd|S)Nd)r(rrrminr%r!r!r"_get_level_indicator)s z-TreeNodeChoiceFieldMixin._get_level_indicatorcCs"||}t|dtt|S)zs Creates labels which represent the tree level of each node when generating option labels.  )r+rrr)rr&rr!r!r"label_from_instance-s z,TreeNodeChoiceFieldMixin.label_from_instance)__name__ __module__ __qualname__rr(r+r- __classcell__r!r!rr"rs   rc@eZdZdZdS)r z"A ModelChoiceField for tree nodes.Nr.r/r0__doc__r!r!r!r"r 6sr c@r2)r z*A ModelMultipleChoiceField for tree nodes.Nr3r!r!r!r"r :sr cs\eZdZdZdZdZdZdZeedfeedfeedfeed ffZ fd d Z Z S) r z?A ChoiceField for specifying position relative to another node.z first-childz last-childleftrightz First childz Last childz Left siblingz Right siblingcs(d|vr |j|d<tj|i|dS)Nchoices)DEFAULT_CHOICESrr)rrrrr!r"rOs zTreeNodePositionField.__init__) r.r/r0r4 FIRST_CHILD LAST_CHILDLEFTRIGHT_r8rr1r!r!rr"r @s    r cs8eZdZdZeddZeZfddZddZ Z S)r z A form which allows the user to move a given node from one location in its tree to another, with optional restriction of the nodes which are valid target nodes for the move. N)rc s||_|dd}|dd}|dd}|dd}tj|i||j}|durM|jjd i|jt||j|j dt||j |j dt||j i}||j d _ ||j d j jd <|rd||j d _|rn||j d _dSdS) a The ``node`` to be moved must be provided. The following keyword arguments are also accepted:: ``valid_targets`` Specifies a ``QuerySet`` of valid targets for the move. If not provided, valid targets will consist of everything other node of the same type, apart from the node itself and any descendants. For example, if you want to restrict the node to moving within its own tree, pass a ``QuerySet`` containing everything in the node's tree except itself and its descendants (to prevent invalid moves) and the root node (as a user could choose to make the node a sibling of the root node). ``target_select_size`` The size of the select element used for the target node. Defaults to ``10``. ``position_choices`` A tuple of allowed position choices and their descriptions. Defaults to ``TreeNodePositionField.DEFAULT_CHOICES``. ``level_indicator`` A string which will be used to represent a single tree level in the target options. valid_targetsNtarget_select_size position_choicesr__gte__ltetargetsizepositionr!)noderrrr _tree_managerexcluderr#r right_attrfieldsrwidgetattrsrr7) rrGrrr>r?rAroptsrr!r"rbs*       zMoveNodeForm.__init__c CsNz|j|jd|jd|jWSty&}z |||jt<d}~ww)al Attempts to move the node using the selected target and position. If an invalid move is attempted, the related error message will be added to the form's non-field errors and the error will be re-raised. Callers should attempt to catch ``InvalidNode`` to redisplay the form with the error, should it occur. rDrFN)rGmove_to cleaned_datar error_classerrorsr)rer!r!r"saves zMoveNodeForm.save) r.r/r0r4r rDr rFrrTr1r!r!rr"r Xs   4r cs,eZdZdZfddZfddZZS) MPTTAdminFormzd A form which validates that the chosen parent for a node isn't one of its descendants. cstj|i||jr=|jjr?|jjjsA|j}|jjj}|j |j }|rC|j }|j |jddjdddd}||_ dSdSdSdSdS)NT include_selfpk)flat)pk__in)rrinstancerX_stateadding_metarrrKget parent_attrrrIget_descendants values_list)rrrr[rN parent_field parent_qsrr!r"rs    zMPTTAdminForm.__init__cszt}|jjj}||j}|jr;|r;|j|jddr;|j|j vr+| |j |j<|j |j t d|j |j=|S)NTrVzInvalid parent)rcleanr^rrr_r`r[is_descendant_of_errorsrQappendr=rP)rrPrNparentrr!r"res      zMPTTAdminForm.clean)r.r/r0r4rrer1r!r!rr"rUs rUN)r4djangordjango.forms.formsrdjango.utils.encodingrdjango.utils.htmlrrdjango.utils.translationrr=mptt.exceptionsr mptt.settingsr __all__rModelChoiceFieldr ModelMultipleChoiceFieldr ChoiceFieldr Formr ModelFormrUr!r!r!r"s"        R