o Df6@sdZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z mZddlmZmZmZgdZeeeee fZGd d d ZGd d d eZddddZdS)z Wrapper for the layout. ) annotations) GeneratorIterableUnion)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#U/home/ubuntu/webapp/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&r2r buffernamer ValueErrorrr is_focusablecurrent_controlr r rr(rr0rreversedr)r!r3controlwindowscwr#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_bufferr9rrr<r r r(r)r!r3elementr#r#r$ has_focuss           zLayout.has_focusrcCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr0r)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.)rr0r(r:)r!r>windowr#r#r$r<s  r cCs |jdS)z=Return the :class:`.Window` object that is currently focused.rF)rr)r#r#r$r(s zLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r!r3r#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!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 r0r;)r!rAr#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#).0rAr r#r$ sz8Layout.get_visible_focusable_windows..)r rNr)r#rQr$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 r8r! ui_controlr#r#r$rCs 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 r0r r8r9)r!rWrAr#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 rUr#r#r$buffer_has_focuss zLayout.buffer_has_focuscCs.z|jdjWSty|jdjYSw)zJ Get the :class:`.UIControl` to previously had the focus. rF)rr0 IndexErrorr)r#r#r$previous_control!s  zLayout.previous_controlcCs&t|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. rNrF)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. rrNrSr]indexr(r:rr!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_rar#r#r$focus_previousBrczLayout.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$rMXs  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#rir$update_parents_relationses  zLayout.update_parents_relationscCs|j|jdSr/)rclearrresetr)r#r#r$rmts 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)rKeyErrorr1r#r#r$ get_parent}s   zLayout.get_parentr/)rrrrrr)rr&)rr+)rr.)r3r4rr)r3r4rrB)rr)r>rrr)rr )r3r rr)rrB)rrI)rrL)rrO)rrT)rWr&rrT)rr)rre)rr rrn)__name__ __module__ __qualname____doc__r%r*rr2rrEpropertyr<setterr(rHrKrNrSrCrXrYr\r^rbrdrrMrkrmrpr#r#r#r$rsL "    N                 rc@s eZdZdS)rN)rqrrrsr#r#r#r$rsrFrr r7rBrreccsF|rt|tr|sdS|V|D] }t||dEdHqdS)z: Walk through layout, starting at this container. Nr6)r,r filterrhr)rr7r@r#r#r$rs rN)F)rr r7rBrre)rt __future__rtypingrrrprompt_toolkit.bufferr containersrr r r r controlsr rr__all__r&r4r Exceptionrrr#r#r#r$s  m