U AfR@s<ddlmZmZddlTddlmZGdddejjZ dS))OptionalList)*) CmfTreeNodec@seZdZdZdZdZdZejj j ddddd gZ e d d Z e d d Ze d-ddZe ddZe ddZe ddZe d.eeeedddZe edddZe eddd Ze edd!d"Ze d#d$Ze d%d&Ze d/d'd(Ze d)d*Ze d+d,Z dS)0 CmfMenuTreeZMenuTreeZevarootZMNTuКорень менюget_parents_listpublic_get_treetree_menu_in_rootget_nodeget_treecCstjjtjkrdSdS)NTF)modelsZCmfPersonGroupZsharelink_groupidgZcurrent_person__member_ofrr ./common/models/cmf_menu_tree.py_allow_add_sharelinksz CmfMenuTree._allow_add_sharelinkc Csg}t|}|jdd|gdddddddgdgd }|}|D]}t}|j|_|j|_|j|_|jj |_|j j pzd |_ |j |_ |j j |_ |j j |_ |j rt|j jn|j|_|j|_t|d|j|_d |_|j|_|r|j|_nd |_||q@|S) Ntree_parent_id==ordernotree_node_is_branch parent_idsharelink_hash ui_view_formsys_type)filterfieldsZorderby@BTNOACCESS)cmfutilget_model_by_idlistr CmfUiMenuNoder class_namecodervaluernamerr ui_moduleui_namegetattrlowervf allow_moverrappend)clsr node_listmodelZobj_list add_sharelinkvnoderrrclass_node_lists6          zCmfMenuTree.class_node_listFc Cs t|}|jdd|gddddddd d d d d dddddddddddddgd}|}g}|dksl|jdkrp|SddtD}|jdkr|jdkrg}|D]V} || j d d!dd"gddd"ggd#d|j ggd"ddd$d%d&d'd(d d)d*d+d,d-d!dgd.q||_|jD]} | j d"krq|s,| j r,qn|rF| j sF| j sFqt| j}t} | j| _| j | _ | j| _| j| _t| dr| jdkr|j | _n| j| _| jpd/| _d| jkr| j| _d| jkr| j| _d| jkr| j| _| j| _| jd0krd1| j| _|| _ | j| _| jrVs z-CmfMenuTree.obj_node_list..Z CmfFolderarchiveOR cmf_archivedTZ tree_parentZ is_publicrsystemZhas_tree_nodesr tree_hiddenrperm_policy_anonymousperm_policy_guest logic_prefix)rZinclude_archivedrrZ CmfComponentu▣ FrcSs|jSNrxrrrz+CmfMenuTree.obj_node_list..key),rrsgetrr4 iter_modelsr"rextendslistrr@r>r8Zget_model_by_namer!r r#hasattrrr__dict__r5children_countr6r7r%rr&r'r+r( cmf_owner_idperm_effective_acl_idrArBr?r)r*rC only_archivedrr,sort) r-rrUr/pr0r.Z node_modelsr4Z node_modelr1r2rrr obj_node_list;s                 zCmfMenuTree.obj_node_listcCs8d|krt|}|jStj|dgdj}|s2dS|S)N:Classrr)rrmenu_tree_parent_id get_obj_by_idr)r-r r/rrrrtree_parent_id_gets zCmfMenuTree.tree_parent_id_getcCs>|sgSg}|}||kr:||}|s.|r.gS||q|SrD)r]r,)r-node_idrresZ cur_node_idrrr get_walkups  zCmfMenuTree.get_walkupcOs |j||SrD)r )r-argskwargsrrrrszCmfMenuTree.public_get_tree2N)level expand_nodesreturnc Cs|rd|krt|sdS|dkrtt}|rt|D]B}|r0|dks0d|krJq0t|sVq0||||||q0g}|j||d}|||d8}|D]Z} | jdkr|j | j ||dd | _ d| _ q| j r|d ks| j |kr|j | j |||d | _ q|S) u Получить дерево в развернутом виде :param tree_parent_id: Корневой элемент :param level: Уровень развернутости дерева, базовый уровень 1 и далее вложенные в него узлы :param expand_nodes: Список узлов которые раскрыты пользователем :param set expand_all: Весь список список узлов которые раскрыты пользователем от родителя до текущего узла :return: Список узлов раскрытых до определенного уровня и в соответствии с настройками пользователя rYNZnull:rUr<T) expand_allrUr)rr\setupdater`addr.rNrr r r4rUr) r-rrdrerjrUr^r_r.r2rrrr s.     zCmfMenuTree.get_tree)rfcCsH|s|S|js|||S||tj|j|d}||||SdS)N)r r)rr,rr\_check_parents)r-objresultrrrrrns  zCmfMenuTree._check_parentscCsPg}|s |Sddg}|D]}||kr||qtj||d}||||}|S)u Получить совокупный список из id родителей и самого объекта :param str id: id обекта, родителей которого нужно узнать :param fields: перечень полей, которые дудут присутствовать у объектов в результирующем списке :return: Список из id. r rrZ)r,rr\rn)r-r rrpZREQUIRED_PARAMETERSirorrrrs  zCmfMenuTree.get_parents_listcCs`d|krt|}d}nB|}tj|ddddddd d d d d ddddgd}|sVdSt}|j|_|j|_|j|_|jj |_|j j pd|_ d |j kr|j j |_ d |j kr|jj |_ d |j kr|jj |_ |j|_|j|_|j|_|jrt|jjn|j|_|j|_d|_|j|_|j|_t|dd|_t|d|j|_|rV|j|_nd|_|S)u@ Получить CmfUiMenuNode для ноды rYNrrr?rrrr5r6r7rr#r%rSrTrrZrTFr)rrrr\r!r r"r#rr$rrQr5rRr6r7r%rrr&r'r+rSrTr(r?r)r*r)r-r r/r2r0r1rrrr s`              zCmfMenuTree.get_nodec Cs|D]}tjjj|dgd}|s"q|g}|}g}ddlm}|D]V}|jddtdd|Dgdgd d } |td d| Dkr|S| | qD|sqt d t |q(qdS) Nr rZr)CmfActiveEntityrINcSsg|] }|jqSrr r:crrrr;Isz1CmfMenuTree.tree_menu_in_root..T)rrZinclude_templatescSsg|] }|jqSrrtrurrrr;Ksz)Iter public_tree_menu_in_root to_process:) cmfZincluderr\Zcommon.models.cmf_active_entityrrZiter_subclassesrOr rNrdebuglen) r-r^Z root_nodesZroot_node_tuuidroZ children_newZchildrenrrZ model_clsZchildrens_iterrrrr 4s"  &  zCmfMenuTree.tree_menu_in_rootcCstjj|ddddgd}g}|D]}t|j}t}|jd|_|j|_|jd|_ |j |_ |j |_ |j pt|j|_|j|_|j|_|j|_d|_t|d|j|_|j|_d|_||q |S)N obj_id_idrrr)rrrYF)r ZCmfMenuTreeItemr rrrzr!r"r r#menu_tree_node_is_branchrrverbose_name_plural verbose_namer%rr&r'rr(r)r*rr+r,)r-r menu_itemsretitemr/r2rrrmenu_item_listTs,   zCmfMenuTree.menu_item_listcCsXd|kr2t|}t|dr&|}q@||}n|j||d}||}|||S)NrYmenu_class_node_listrh)rrrPrr3rXrrN)r-rrUr/Znodesr~rrrr.ss      zCmfMenuTree.node_listcCsg}tD]}t|dr |jdk r |j|kr t}|jd|_|jd|_|j|_|j|_ |j |_ |j pr|j |_|j|_|j|_|j|_d|_|j|_d|_||q |jddd|S)Nr[rYFcSs|jSrDrErFrrrrHrIz2CmfMenuTree.helper_subclass_list..rJ)rrMrPr[r!r"r r#r{rZmenu_tree_ordernorr|r}r%rr&r'rr)r*r+r,rV)r-rr.r/r2rrrhelper_subclass_lists.   z CmfMenuTree.helper_subclass_listcCs tjdS)NzCmfMenuTree:Class)r rr)r-rrrrsz CmfMenuTree.menu_class_node_list)F)rcNNF)F)!__name__ __module__ __qualname__r'r&Z code_prefixr}rwr CmfEntityZ api_methods staticmethodr classmethodr3rXr]r`rintr rr rrnrr!r r rr.rrrrrrrsR   ^   * ,    rN) typingrrZ cmf.includeZcommon.models.cmf_tree_noderrwr rrrrrrs