o i6@sdZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZgdZe ee ee fZGd d d ZGd d d eZddddZdS)z Wrapper for the layout. ) annotations)Dict GeneratorIterableListOptionalUnion)Buffer) AnyContainerConditionalContainer ContainerWindow to_container) BufferControlSearchBufferControl UIControl)LayoutInvalidLayoutErrorwalkc@s@eZdZdZ dLdMd d ZdNd d ZdOddZdPddZdQddZdRddZ e dSddZ e j dTddZ e dUd!d"Z e j dVd#d"Z e dWd$d%Ze dXd'd(ZdYd*d+ZdZd-d.Ze d[d0d1Zd\d3d4Ze dWd5d6Ze dSd7d8Zd]d9d:Zd]d;d<Zd]d=d>Zd^d@dAZd^dBdCZd]dDdEZd]dFdGZd_dJdKZdS)`raL The layout for a prompt_toolkit :class:`~prompt_toolkit.application.Application`. This also keeps track of which user control is focused. :param container: The "root" container for the layout. :param focused_element: element to be focused initially. (Can be anything the `focus` function accepts.) N containerr focused_elementFocusableElement | NonereturnNonec Cstt||_g|_i|_i|_|dur0z |jt|Wnty/}zt d|d}~ww| |g|_ dS)Nz>Invalid layout. The layout does not contain any Window object.) rr_stack search_links_child_to_parentappendnextfind_all_windows StopIterationrfocusvisible_windows)selfrrer&W/var/www/edux/Edux_v2/venv/lib/python3.10/site-packages/prompt_toolkit/layout/layout.py__init__'s"   zLayout.__init__strcCsd|jd|jdS)NzLayout(z, current_window=))rcurrent_windowr$r&r&r'__repr__IszLayout.__repr__Generator[Window, None, None]ccs$|D] }t|tr|VqdS)zJ Find all the :class:`.UIControl` objects in this layout. N)r isinstancer)r$itemr&r&r'r Ls   zLayout.find_all_windowsIterable[UIControl]ccs|D]}|jVqdSN)r contentr$rr&r&r'find_all_controlsTs  zLayout.find_all_controlsvalueFocusableElementcCst|tr'|D]}t|tr|jj|kr||dSq td|dt|trM|D]}t|trD|j|krD||dSq0td|dt|t ri||vr\td| sdtd||_ dSt |}t|t r||vrtd|f||_dSg}t|ddD]}t|t r|j r||qt|jD] }||vr||_dSq|r|d |_dStd |) a Focus the given UI element. `value` can be either: - a :class:`.UIControl` - a :class:`.Buffer` instance or the name of a :class:`.Buffer` - a :class:`.Window` - Any container object. In this case we will focus the :class:`.Window` from this container that was focused most recent, or the very first focusable :class:`.Window` of the container. Nz,Couldn't find Buffer in the current layout: .z7Invalid value. Container does not appear in the layout.z*Invalid value. UIControl is not focusable.z7Invalid value. Window does not appear in the layout: %rT skip_hiddenrz,Invalid value. Container cannot be focused: )r/r)r5rbuffernamer" ValueErrorr r is_focusablecurrent_controlrrr r+rr3rreversedr)r$r6controlwindowscwr&r&r'r"Xs^              z Layout.focusboolcCst|tr|jdur dS|jj|kSt|tr|j|kSt|tr&|j|kSt|}t|tr4|j |kSt |D] }||j krBdSq8dS)z Check whether the given control has the focus. :param value: :class:`.UIControl` or :class:`.Window` instance. NFT) r/r)current_bufferr<r rr?rrr+r)r$r6elementr&r&r' has_focuss           zLayout.has_focusrcCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr3r,r&r&r'r?s zLayout.current_controlrAcCs,|D] }|j|kr||_dSqtd)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)r r3r+r=)r$rAwindowr&r&r'r?s  rcCs |jdS)z=Return the :class:`.Window` object that is currently focused.rI)rr,r&r&r'r+s zLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r$r6r&r&r'r+scCs |j|jvS)z#True if we are searching right now.)r?rr,r&r&r' is_searchings zLayout.is_searchingBufferControl | NonecCs |j}t|tr|j|SdS)zY Return the :class:`.BufferControl` in which we are searching or `None`. N)r?r/rrget)r$rAr&r&r'search_target_buffer_controls  z#Layout.search_target_buffer_controlIterable[Window]ccs.|D]}t|tr|jr|VqdS)zl Return all the :class:`.Window` objects which are focusable (in the 'modal' area). N)walk_through_modal_arear/rr3r>)r$rDr&r&r'get_focusable_windowss  zLayout.get_focusable_windows list[Window]cs|jfdd|DS)zO Return a list of :class:`.Window` objects that are focusable. csg|]}|vr|qSr&r&).0rDr#r&r' sz8Layout.get_visible_focusable_windows..)r#rQr,r&rTr'get_visible_focusable_windowssz$Layout.get_visible_focusable_windows Buffer | NonecCs|j}t|tr |jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)r?r/rr;r$ ui_controlr&r&r'rFs zLayout.current_buffer buffer_namecCsB|D]}t|trt|jtr|jjj|kr|jjSqdS)zt Look in the layout for a buffer with the given name. Return `None` when nothing was found. N)rr/rr3rr;r<)r$rZrDr&r&r'get_buffer_by_name s  zLayout.get_buffer_by_namecCs|j}t|tS)z Return `True` if the currently focused control is a :class:`.BufferControl`. (For instance, used to determine whether the default key bindings should be active or not.) )r?r/rrXr&r&r'buffer_has_focuss zLayout.buffer_has_focuscCs.z|jdjWSty|jdjYSw)zJ Get the :class:`.UIControl` to previously had the focus. rI)rr3 IndexErrorr,r&r&r'previous_control!s  zLayout.previous_controlcCs&t|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. r NrI)lenrr,r&r&r' focus_last+szLayout.focus_lastcCsb|}t|dkr/z||j}Wn tyd}Yn w|dt|}|||dSdS)z: Focus the next visible/focusable Window. rr NrVr`indexr+r=r"r$rBrcr&r&r' focus_next2  zLayout.focus_nextcCsb|}t|dkr/z||j}Wn tyd}Yn w|dt|}|||dSdS)z> Focus the previous visible/focusable Window. rr Nrbrdr&r&r'focus_previousBrfzLayout.focus_previousIterable[Container]ccst|jEdHdS)zX Walk through all the layout nodes (and their children) and yield them. N)rrr,r&r&r'rRsz Layout.walkccsH|j}|s||jvr|j|}|s||jvs t|EdHdS)zn Walk through all the containers which are in the current 'modal' part of the layout. N)r+is_modalrr)r$rootr&r&r'rPXs  zLayout.walk_through_modal_areacs(idfdd |j|_dS) z= Update child->parent relationships mapping. r%r rrcs"|D] }||<|qdSr2) get_children)r%rCparentsrr&r'rks  z-Layout.update_parents_relations..walkN)r%r rr)rrr,r&rlr'update_parents_relationses  zLayout.update_parents_relationscCs|j|jdSr2)rclearrresetr,r&r&r'rpts z Layout.resetr Container | NonecCs"z|j|WStyYdSw)zo Return the parent container for the given container, or ``None``, if it wasn't found. N)rKeyErrorr4r&r&r' get_parent}s   zLayout.get_parentr2)rr rrrr)rr))rr.)rr1)r6r7rr)r6r7rrE)rr)rArrr)rr)r6rrr)rrE)rrL)rrO)rrR)rrW)rZr)rrW)rr)rrh)rr rrq)__name__ __module__ __qualname____doc__r(r-r r5r"rHpropertyr?setterr+rKrNrQrVrFr[r\r_rarergrrPrnrprsr&r&r&r'rsL "    O                 rc@s eZdZdS)rN)rtrurvr&r&r&r'rsrFrr r:rErrhccsF|rt|tr|sdS|V|D] }t||dEdHqdS)z: Walk through layout, starting at this container. Nr9)r/r filterrkr)rr:rCr&r&r'rs rN)F)rr r:rErrh)rw __future__rtypingrrrrrrprompt_toolkit.bufferr containersr r r rrcontrolsrrr__all__r)r7r Exceptionrrr&r&r&r's   n