Mh6dZddlmZddlmZmZmZddlmZddl m Z m Z m Z m Z mZddlmZmZmZgdZeeeee fZGd d ZGd d eZdddZdS)z Wrapper for the layout. ) annotations) GeneratorIterableUnion)Buffer) AnyContainerConditionalContainer ContainerWindow to_container) BufferControlSearchBufferControl UIControl)LayoutInvalidLayoutErrorwalkceZdZdZ d4d5d Zd6d Zd7d Zd8dZd9dZd:dZ e d;dZ e j ddZ e d?dZe d@dZdAd ZdBd"Ze dCd$ZdDd&Ze d?d'Ze d;d(ZdEd)ZdEd*ZdEd+ZdFd-ZdFd.ZdEd/ZdEd0ZdGd3ZdS)HraL 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 containerr focused_elementFocusableElement | NonereturnNonecNt||_g|_i|_i|_|] |jt |n7#t$r}td|d}~wwxYw| |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)selfrres \/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/prompt_toolkit/layout/layout.py__init__zLayout.__init__(s &i00$& GI =?  "  ""4(=(=(?(?#@#@AAAA    (T  JJ ' ' '.0s9A'' B1BBstrc(d|jd|jdS)NzLayout(z, current_window=))rcurrent_windowr$s r&__repr__zLayout.__repr__Js TTTD 1133  g}55'.:MQV:V:VJJw'''FFVEVVVWW Wv & &6 1133  g}55'.E:Q:QJJw'''FFVEVVVWW Wy ) ). D224444 M%%'' O !MNNN#(D !''E%((  5 5 7 777$YPUYY',###e666**A!!V,,*1G1G1I1I*q)))"$+..AG||./+$ *1!*D'F L5LLr.boolct|tr|jdS|jj|kSt|tr |j|kSt|t r |j|kSt|}t|tr |j |kSt|D]}||j krdSdS)z Check whether the given control has the focus. :param value: :class:`.UIControl` or :class:`.Window` instance. NFT) r1r(current_bufferr@rrrCr r r+r)r$r9elements r& has_focuszLayout.has_focuss eS ! ! 5"*u&+u4 4 eV $ $ 0&%/ / eY ' ' '50 0 ''E%(( *e33 $E{{$$G$"555#tt6ur.rc&|jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr6r,s r&rCzLayout.current_controls {2&&r.rEcx|D]}|j|kr ||_dStd)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)r r6r+rA)r$rEwindows r&rCzLayout.current_controlsQ ++--  F~((&,#)CDDDr.r c|jdS)z=Return the :class:`.Window` object that is currently focused.rO)rr,s r&r+zLayout.current_windows{2r.c:|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r$r9s r&r+zLayout.current_windows  5!!!!!r.c|j|jvS)z#True if we are searching right now.)rCrr,s r& is_searchingzLayout.is_searchings#t'888r.BufferControl | Nonecr|j}t|tr|j|SdS)zY Return the :class:`.BufferControl` in which we are searching or `None`. N)rCr1rrget)r$rEs r&search_target_buffer_controlz#Layout.search_target_buffer_controls;& g2 3 3 $((11 14r.Iterable[Window]c#K|D]4}t|tr|jr|V5dS)zl Return all the :class:`.Window` objects which are focusable (in the 'modal' area). N)walk_through_modal_arear1r r6rB)r$rHs r&get_focusable_windowszLayout.get_focusable_windowss^ --//  A!V$$ )?)?)A)A   r. list[Window]cR|jfd|DS)zO Return a list of :class:`.Window` objects that are focusable. cg|]}|v| Sra).0rHr#s r& z8Layout.get_visible_focusable_windows..s#PPPa1;O;O;O;O;Or.)r#r])r$r#s @r&get_visible_focusable_windowsz$Layout.get_visible_focusable_windowss3 .PPPP45577PPPPr. Buffer | NonecL|j}t|tr|jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)rCr1rr?r$ ui_controls r&rKzLayout.current_buffers, ) j- 0 0 %$ $tr. buffer_namec|D]T}t|tr=t|jtr#|jjj|kr|jjcSUdS)zt Look in the layout for a buffer with the given name. Return `None` when nothing was found. N)rr1r r6rr?r@)r$rirHs r&get_buffer_by_namezLayout.get_buffer_by_name sj  , ,A!V$$ ,AI})M)M ,9#(K779++++tr.c:|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.) )rCr1rrgs r&buffer_has_focuszLayout.buffer_has_focuss) *m444r.cl |jdjS#t$r|jdjcYSwxYw)zJ Get the :class:`.UIControl` to previously had the focus. rO)rr6 IndexErrorr,s r&previous_controlzLayout.previous_control!sF  +;r?* * + + +;r?* * * * +s 33cbt|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. rNrO)lenrr,s r& focus_lastzLayout.focus_last+s7 t{  a  +crc*DKKK r.c|}t|dkr` ||j}|dzt|z}n#t$rd}YnwxYw|||dSdS)z: Focus the next visible/focusable Window. rrNrdrsindexr+rAr"r$rFrws r& focus_nextzLayout.focus_next24466 w< Focus the previous visible/focusable Window. rrNrvrxs r&focus_previouszLayout.focus_previousBrzr{Iterable[Container]c#>Kt|jEd{VdS)zX Walk through all the layout nodes (and their children) and yield them. N)rrr,s r&rz Layout.walkRs0'''''''''''r.c#K|j}|s3||jvr*|j|}|s ||jv*t|Ed{VdS)zn Walk through all the containers which are in the current 'modal' part of the layout. N)r+is_modalrr)r$roots r&r\zLayout.walk_through_modal_areaXs---// /dd.C&C&C(.D--// /dd.C&C&C::r.cJidfd |j|_dS)z= Update child->parent relationships mapping. r%r rrcV|D]}||<|dSr5) get_children)r%rGparentsrs r&rz-Layout.update_parents_relations..walkks>^^%%   Q  r.N)r%r rr)rr)r$rrs @@r&update_parents_relationszLayout.update_parents_relationsesP        T^ 'r.cj|j|jdSr5)rclearrresetr,s r&rz Layout.resetts3 !!! r.r Container | Nonec@ |j|S#t$rYdSwxYw)zo Return the parent container for the given container, or ``None``, if it wasn't found. N)rKeyErrorr7s r& get_parentzLayout.get_parent}s7  (3 3   44 s  r5)rr rrrr)rr()rr/)rr3)r9r:rr)r9r:rrI)rr)rErrr)rr )r9r rr)rrI)rrV)rrZ)rr^)rre)rir(rre)rr)rr~)rr rr)__name__ __module__ __qualname____doc__r'r-r r8r"rMpropertyrCsetterr+rUrYr]rdrKrkrmrqrtryr}rr\rrrrar.r&rrs48 0 0 0 0 0DUUUU$$$$LLLL\2'''X'  E E E EX""""999X9   X QQQQX    555X5+++X+++++'''' '''' ((((     ( ( ( (r.rceZdZdS)rN)rrrrar.r&rrsDr.rFrr r>rIrr~c#K|r+t|tr|sdS|V|D]}t ||Ed{VdS)z: Walk through layout, starting at this container. Nr=)r1r filterrr)rr>rGs r&rrs  y"6 7 7  ""  OOO  # # % %44{3333333333344r.N)F)rr r>rIrr~)r __future__rtypingrrrprompt_toolkit.bufferr containersr r r r r controlsrrr__all__r(r:r Exceptionrrrar.r&rs_#"""""----------((((((DCCCCCCCCC    fi=>hhhhhhhhV         4444444r.