B 't\2@sdZddlmZddlmZmZddlmZmZm Z m Z ddl m Z ddl Z dd d gZGd ddeZGd d d eZddd ZdS)z Wrapper for the layout. )unicode_literals) UIControl BufferControl) ContainerWindow to_containerConditionalContainer)BufferNLayoutInvalidLayoutErrorwalkc@s eZdZdZd5ddZddZddZd d Zd d Zd dZ e ddZ e j ddZ e ddZ e j ddZ e ddZe ddZddZddZe ddZdd Ze d!d"Ze d#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4ZdS)6r aL 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.) NcCspt||_g|_i|_i|_|dkr\y|jt|Wqftk rXt dYqfXn | |g|_ dS)Nz>Invalid layout. The layout does not contain any Window object.) r container_stack search_links_child_to_parentappendnextfind_all_windows StopIterationr focusvisible_windows)selfrZfocused_elementr;lib/python3.7/site-packages/prompt_toolkit/layout/layout.py__init__s  zLayout.__init__cCsd|j|jfS)NzLayout(%r, current_window=%r))rcurrent_window)rrrr__repr__6szLayout.__repr__ccs&x |D]}t|tr |Vq WdS)zJ Find all the :class:`.UIControl` objects in this layout. N)r isinstancer)ritemrrrr:s zLayout.find_all_windowsccsx|D] }|jVq WdS)N)rcontent)rrrrrfind_all_controlsBszLayout.find_all_controlscCst|tjrTx4|D](}t|tr|jj|kr||dSqWtd|fnFt|t rx2|D]&}t|trh|j|krh||dSqhWtd|fnt|t r||krtd| std||_ nt |}t|tr||krtd|f||_ng}x8t|ddD](}t|tr"|j r"||q"Wx(t|jD]}||krZ||_dSqZW|r|d|_dStd |fdS) 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: %r.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)rsix text_typer!rbuffernamer ValueErrorr r is_focusablecurrent_controlrrrrr r rreversedr)rvaluecontrolwindowscwrrrrFsH           z Layout.focuscCst|tjr&|jdkrdS|jj|kSt|tr:|j|kSt|trN|j|kSt|}t|t rj|j |kSxt |D]}||j krtdSqtWdSdS)z Check whether the given control has the focus. :param value: :class:`.UIControl` or :class:`.Window` instance. NFT) rr#r$current_bufferr&r rr)rrrr )rr+elementrrr has_focuss          zLayout.has_focuscCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr )rrrrr)szLayout.current_controlcCs@t|tstx$|D]}|j|kr||_dSqWtddS)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)rrAssertionErrorrr rr')rr,Zwindowrrrr)s  cCs |jdS)z? Return the :class:`.Window` object that is currently focused. r3)r)rrrrrszLayout.current_windowcCst|tst|j|dS)z: Set the :class:`.Window` object to be currently focused. N)rrr4rr)rr+rrrrscCs |j|jkS)z% True if we are searching right now. )r)r)rrrr is_searchingszLayout.is_searchingcCs|j|jS)zI Return the :class:`.BufferControl` in which we are searching or `None`. )rgetr))rrrrsearch_target_buffer_controlsz#Layout.search_target_buffer_controlccs0x*|D]}t|tr |jr |Vq WdS)zl Return all the :class:`.Window` objects which are focusable (in the 'modal' area). N)walk_through_modal_arearrr r()rr/rrrget_focusable_windowsszLayout.get_focusable_windowscs|jfdd|DS)zO Return a list of :class:`.Window` objects that are focusable. csg|]}|kr|qSrr).0r/)rrr sz8Layout.get_visible_focusable_windows..)rr9)rr)rrget_visible_focusable_windowssz$Layout.get_visible_focusable_windowscCs|j}t|tr|jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)r)rrr%)r ui_controlrrrr0s zLayout.current_buffercCsBx<|D]0}t|tr t|jtr |jjj|kr |jjSq WdS)zt Look in the layout for a buffer with the given name. Return `None` when nothing was found. N)r rrr rr%r&)rZ buffer_namer/rrrget_buffer_by_nameszLayout.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)rr)rr=rrrbuffer_has_focusszLayout.buffer_has_focuscCs.y |jdjStk r(|jdjSXdS)zJ Get the :class:`.UIControl` to previously had the focus. r3N)rr IndexError)rrrrprevious_controls zLayout.previous_controlcCs"t|jdkr|jdd|_dS)z= Give the focus to the last focused control. rNr3)lenr)rrrr focus_last szLayout.focus_lastcCs`|}t|dkr\y||j}Wntk r<d}YnX|dt|}|||dS)z: Focus the next visible/focusable Window. rrN)r<rCindexrr'r)rr-rErrr focus_nexts  zLayout.focus_nextcCs`|}t|dkr\y||j}Wntk r<d}YnX|dt|}|||dS)z> Focus the previous visible/focusable Window. rrN)r<rCrErr'r)rr-rErrrfocus_previous"s  zLayout.focus_previousccsxt|jD] }|Vq WdS)zX Walk through all the layout nodes (and their children) and yield them. N)r r)rirrrr 2sz Layout.walkccsD|j}x |s&||jkr&|j|}qWxt|D] }|Vq2WdS)zn Walk through all the containers which are in the current 'modal' part of the layout. N)rZis_modalrr )rrootrrrrr89s zLayout.walk_through_modal_areacs&ifdd|j|_dS)z= Update child->parent relationships mapping. cs&x |D]}||<|q WdS)N) get_children)er.)parentsr rrr Msz-Layout.update_parents_relations..walkN)rr)rr)rLr rupdate_parents_relationsGs zLayout.update_parents_relationscCs|j|jS)N)rclearrreset)rrrrrOVs z Layout.resetcCs$y |j|Stk rdSXdS)zo Return the parent container for the given container, or ``None``, if it wasn't found. N)rKeyError)rrrrr get_parent_s zLayout.get_parent)N)__name__ __module__ __qualname____doc__rrrr!rr2propertyr)setterrr5r7r9r<r0r>r?rBrDrFrGr r8rMrOrQrrrrr s6  I        c@s eZdZdS)r N)rRrSrTrrrrr jsFccs`t|tst|r(t|tr(|s(dS|Vx,|D] }xt||dD] }|VqJWq8WdS)z: Walk through layout, starting at this container. N)r")rrr4r filterrJr )rr"r.rHrrrr ns)F)rUZ __future__rZcontrolsrrZ containersrrrr Zprompt_toolkit.bufferr r#__all__objectr Exceptionr r rrrrs  [