U d6@sdZddlmZddlmZmZmZmZmZm Z ddl m Z ddl m Z mZmZmZmZddlmZmZmZdd d gZe ee ee fZGd ddZGd d d eZddddddd ZdS)z Wrapper for the layout. ) annotations)Dict GeneratorIterableListOptionalUnion)Buffer) AnyContainerConditionalContainer ContainerWindow to_container) BufferControlSearchBufferControl UIControlLayoutInvalidLayoutErrorwalkc@seZdZdZdLddddddZd d d d Zd d ddZdd ddZdddddZdddddZ e dd ddZ e j dddddZ e dd d d!Z e j dddd"d!Z e dd d#d$Ze d%d d&d'Zd(d d)d*Zd+d d,d-Ze d.d d/d0Zd d.d1d2d3Ze dd d4d5Ze dd d6d7Zdd d8d9Zdd d:d;Zdd dd d?d@Zd>d dAdBZdd dCdDZdd dEdFZdGdHdIdJdKZdS)MraL 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.) Nr zFocusableElement | NoneNone) containerfocused_elementreturnc Cst||_g|_i|_i|_|dkrlz|jt|Wqvtk rh}zt d|W5d}~XYqvXn | |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%@/tmp/pip-unpacked-wheel-4x_7prb2/prompt_toolkit/layout/layout.py__init__'s  zLayout.__init__str)rcCsd|jd|jdS)NzLayout(z, current_window=))rcurrent_windowr#r%r%r&__repr__IszLayout.__repr__zGenerator[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&rLs  zLayout.find_all_windowszIterable[UIControl]ccs|D] }|jVqdSN)rcontentr#rr%r%r&find_all_controlsTs zLayout.find_all_controlsFocusableElement)valuercCst|trR|D]*}t|tr|jj|kr||dSqtd|dn@t|tr|D](}t|trd|j|krd||dSqdtd|dnt|t r||krtd| std||_ nt |}t|t r||krtd|f||_ng}t|ddD](}t|t r|j r||qt|jD]}||krR||_dSqR|r|d |_dStd |dS) 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(r2rbuffernamer! ValueErrorr r is_focusablecurrent_controlrrrr*rr0rreversedr)r#r4controlwindowscwr%r%r&r!XsV            z Layout.focusboolcCst|tr$|jdkrdS|jj|kSt|tr8|j|kSt|trL|j|kSt|}t|trh|j |kSt |D]}||j krpdSqpdSdS)z Check whether the given control has the focus. :param value: :class:`.UIControl` or :class:`.Window` instance. NFT) r-r(current_bufferr9r rr<rrr*r)r#r4elementr%r%r& has_focuss           zLayout.has_focusrcCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr0r+r%r%r&r<szLayout.current_control)r>rcCs0|D]}|j|kr||_dSqtddS)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)rr0r*r:)r#r>Zwindowr%r%r&r<s   rcCs |jdS)z=Return the :class:`.Window` object that is currently focused.rF)rr+r%r%r&r*szLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r#r4r%r%r&r*scCs |j|jkS)z#True if we are searching right now.)r<rr+r%r%r& is_searchingszLayout.is_searchingzBufferControl | NonecCs$|j}t|tr|j|SdSdS)zY Return the :class:`.BufferControl` in which we are searching or `None`. N)r<r-rrget)r#r>r%r%r&search_target_buffer_controls  z#Layout.search_target_buffer_controlzIterable[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-rr0r;)r#rAr%r%r&get_focusable_windowss zLayout.get_focusable_windowsz list[Window]cs|jfdd|DS)zO Return a list of :class:`.Window` objects that are focusable. csg|]}|kr|qSr%r%).0rAr"r%r& sz8Layout.get_visible_focusable_windows..)r"rKr+r%rMr&get_visible_focusable_windowssz$Layout.get_visible_focusable_windowsz Buffer | NonecCs|j}t|tr|jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)r<r-rr8r#Z ui_controlr%r%r&rCs zLayout.current_buffer) buffer_namercCsB|D]4}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-rr0rr8r9)r#rQrAr%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-rrPr%r%r&buffer_has_focusszLayout.buffer_has_focuscCs4z|jdjWStk r.|jdjYSXdS)zJ Get the :class:`.UIControl` to previously had the focus. rFN)rr0 IndexErrorr+r%r%r&previous_control!szLayout.previous_controlcCs"t|jdkr|jdd|_dS)z= Give the focus to the last focused control. r NrF)lenrr+r%r%r& focus_last+szLayout.focus_lastcCs`|}t|dkr\z||j}Wntk r<d}YnX|dt|}|||dS)z: Focus the next visible/focusable Window. rr NrOrWindexr*r:r!r#r?rZr%r%r& focus_next2s  zLayout.focus_nextcCs`|}t|dkr\z||j}Wntk r<d}YnX|dt|}|||dS)z> Focus the previous visible/focusable Window. rr NrYr[r%r%r&focus_previousBs  zLayout.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.walkccs6|j}|s$||jkr$|j|}qt|EdHdS)zn Walk through all the containers which are in the current 'modal' part of the layout. N)r*Zis_modalrr)r#rootr%r%r&rJXs zLayout.walk_through_modal_areacs.idddfdd |j|_dS)z= Update child->parent relationships mapping. r r)r$rcs"|D]}||<|qdSr/) get_children)r$r@parentsrr%r&rks z-Layout.update_parents_relations..walkN)rrr+r%rar&update_parents_relationses zLayout.update_parents_relationscCs|j|jdSr/)rclearrresetr+r%r%r&rets z Layout.resetr zContainer | None)rrcCs(z |j|WStk r"YdSXdS)zo Return the parent container for the given container, or ``None``, if it wasn't found. N)rKeyErrorr1r%r%r& get_parent}s zLayout.get_parent)N)__name__ __module__ __qualname____doc__r'r,rr2r!rEpropertyr<setterr*rGrIrKrOrCrRrSrVrXr\r]rrJrcrergr%r%r%r&rsJ "O          c@s eZdZdS)rN)rhrirjr%r%r%r&rsFr rBr^)rr7rccsD|rt|tr|sdS|V|D]}t||dEdHq(dS)z: Walk through layout, starting at this container. Nr6)r-r filterr`r)rr7r@r%r%r&rs N)F)rk __future__rtypingrrrrrrZprompt_toolkit.bufferr Z containersr r r rrcontrolsrrr__all__r(r3r Exceptionrrr%r%r%r&s   n