o j6@sdZddlmZddlmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZmZmZgdZeeBeBe BZGd d d ZGd d d eZddddZdS)z Wrapper for the layout. ) annotations) GeneratorIterable)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 containerrfocused_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.) r r_stack search_links_child_to_parentappendnextfind_all_windows StopIterationrfocusvisible_windows)selfrrer"/root/aizidognhua/tmp/workspace/projects/ec89d86c-575f-41c9-af57-ac45cbdbf775/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__JszLayout.__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#rMs   zLayout.find_all_windowsIterable[UIControl]ccs|D]}|jVqdSN)rcontentr rr"r"r#find_all_controlsUs  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|||_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.z5Invalid value. Window does not appear in the layout: T skip_hiddenrz,Invalid value. Container cannot be focused: )r+r%r1r buffernamer ValueErrorrr is_focusablecurrent_controlr r rr'rr/rreversedr)r r2controlwindowscwr"r"r#rYsZ              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_bufferr8rrr;r r r'r)r r2elementr"r"r# has_focuss           zLayout.has_focusrcCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr/r(r"r"r#r;s zLayout.current_controlr=cCs,|D] }|j|kr||_dSqtd)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)rr/r'r9)r r=windowr"r"r#r;s  r cCs |jdS)z=Return the :class:`.Window` object that is currently focused.rE)rr(r"r"r#r's zLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r r2r"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+r rget)r r=r"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+r r/r:)r r@r"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").0r@rr"r# sz8Layout.get_visible_focusable_windows..)rrMr(r"rPr#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+r r7r ui_controlr"r"r#rBs 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+r r/r r7r8)r rVr@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.) )r;r+r rTr"r"r#buffer_has_focuss zLayout.buffer_has_focuscCs.z|jdjWSty|jdjYSw)zJ Get the :class:`.UIControl` to previously had the focus. rE)rr/ IndexErrorr(r"r"r#previous_control!s  zLayout.previous_controlcCs&t|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. rNrE)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. rrNrRr\indexr'r9rr r>r_r"r"r# focus_next2  zLayout.focus_nextcCsb|}t|dkr/z||j}Wn tyd}Yn w|dt|}|||dSdS)z> Focus the previous visible/focusable Window. rrNr^r`r"r"r#focus_previousBrbzLayout.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#rLXs  zLayout.walk_through_modal_areacs(idfdd |j|_dS) z= Update child->parent relationships mapping. r!r rrcs"|D] }||<|qdSr.) get_children)r!r?parentsrr"r#rks  z-Layout.update_parents_relations..walkN)r!r rr)rrr(r"rhr#update_parents_relationses  zLayout.update_parents_relationscCs|j|jdSr.)rclearrresetr(r"r"r#rlts 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)rKeyErrorr0r"r"r# get_parent}s   zLayout.get_parentr.)rrrrrr)rr%)rr*)rr-)r2r3rr)r2r3rrA)rr)r=rrr)rr )r2r rr)rrA)rrH)rrK)rrN)rrS)rVr%rrS)rr)rrd)rr rrm)__name__ __module__ __qualname____doc__r$r)rr1rrDpropertyr;setterr'rGrJrMrRrBrWrXr[r]rarcrrLrjrlror"r"r"r#rsL "    N                 rc@s eZdZdS)rN)rprqrrr"r"r"r#rsrFrr r6rArrdccsF|rt|tr|sdS|V|D] }t||dEdHqdS)z: Walk through layout, starting at this container. Nr5)r+rfilterrgr)rr6r?r"r"r#rs rN)F)rr r6rArrd)rs __future__rcollections.abcrrprompt_toolkit.bufferr containersrrr r r controlsr r r__all__r%r3r Exceptionrrr"r"r"r#s  m