U ou f6@sdZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z mZddlmZmZmZdd d gZeeeee fZGd ddZGd d d eZddddddd ZdS)z Wrapper for the layout. ) annotations) GeneratorIterableUnion)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.) NrzFocusableElement | 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.) r r_stack search_links_child_to_parentappendnextfind_all_windows StopIterationrfocusvisible_windows)selfrrer"@/tmp/pip-unpacked-wheel-8milen2s/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|||_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.z5Invalid value. Window does not appear in the layout: T skip_hiddenrz,Invalid value. Container cannot be focused: )r*r%r/r buffernamer ValueErrorrr is_focusablecurrent_controlr r rr'rr-rreversedr)r r1controlwindowscwr"r"r#rXsR            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_bufferr6rrr9r r r'r)r r1elementr"r"r# has_focuss           zLayout.has_focusrcCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr-r(r"r"r#r9szLayout.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.)rr-r'r7)r r;Zwindowr"r"r#r9s   r cCs |jdS)z=Return the :class:`.Window` object that is currently focused.rC)rr(r"r"r#r'szLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r r1r"r"r#r'scCs |j|jkS)z#True if we are searching right now.)r9rr(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)r9r*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*r r-r8)r r>r"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").0r>rr"r# sz8Layout.get_visible_focusable_windows..)rrHr(r"rJr#get_visible_focusable_windowssz$Layout.get_visible_focusable_windowsz Buffer | NonecCs|j}t|tr|jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)r9r*r r5r Z ui_controlr"r"r#r@s 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*r r-r r5r6)r rNr>r"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.) )r9r*r rMr"r"r#buffer_has_focusszLayout.buffer_has_focuscCs4z|jdjWStk r.|jdjYSXdS)zJ Get the :class:`.UIControl` to previously had the focus. rCN)rr- IndexErrorr(r"r"r#previous_control szLayout.previous_controlcCs"t|jdkr|jdd|_dS)z= Give the focus to the last focused control. rNrC)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. rrNrLrTindexr'r7rr r<rWr"r"r# focus_next1s  zLayout.focus_nextcCs`|}t|dkr\z||j}Wntk r<d}YnX|dt|}|||dS)z> Focus the previous visible/focusable Window. rrNrVrXr"r"r#focus_previousAs  zLayout.focus_previousIterable[Container]ccst|jEdHdS)zX Walk through all the layout nodes (and their children) and yield them. N)rrr(r"r"r#rQsz 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#rGWs 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#rjs z-Layout.update_parents_relations..walkN)rrr(r"r^r#update_parents_relationsds zLayout.update_parents_relationscCs|j|jdSr,)rclearrresetr(r"r"r#rbss 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)rKeyErrorr.r"r"r# get_parent|s zLayout.get_parent)N)__name__ __module__ __qualname____doc__r$r)rr/rrBpropertyr9setterr'rDrFrHrLr@rOrPrSrUrYrZrrGr`rbrdr"r"r"r#rsJ "N          c@s eZdZdS)rN)rerfrgr"r"r"r#rsFr r?r[)rr4rccsD|rt|tr|sdS|V|D]}t||dEdHq(dS)z: Walk through layout, starting at this container. Nr3)r*r filterr]r)rr4r=r"r"r#rs N)F)rh __future__rtypingrrrZprompt_toolkit.bufferrZ containersrr r r r controlsr rr__all__r%r0r Exceptionrrr"r"r"r#s  m