)xj6dZddlmZddlmZmZddlmZddlm Z m Z m Z m Z m Z ddlmZmZmZgdZeezeze zZGd d ZGd d eZdddZdS)z Wrapper for the layout. ) annotations) GeneratorIterable)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 containerrfocused_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 MC:\Projects\Historical\venv\Lib\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) r0r'current_bufferr?rrrBr r r*r)r#r8elements 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. )rr5r+s r%rBzLayout.current_controls {2&&r-rDcx|D]}|j|kr ||_dStd)zC Set the :class:`.UIControl` to receive the focus. Nz(Control not found in the user interface.)rr5r*r@)r#rDwindows r%rBzLayout.current_controlsQ ++--  F~((&,#)CDDDr-r c|jdS)z=Return the :class:`.Window` object that is currently focused.rN)rr+s r%r*zLayout.current_windows{2r-c:|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r#r8s r%r*zLayout.current_windows  5!!!!!r-c|j|jvS)z#True if we are searching right now.)rBrr+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)rBr0rrget)r#rDs 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_arear0r r5rA)r#rGs 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| Sr`).0rGr"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)rBr0r r>r# ui_controls r%rJzLayout.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)rr0r r5r r>r?)r#rhrGs 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.) )rBr0r rfs r%buffer_has_focuszLayout.buffer_has_focuss) *m444r-cl |jdjS#t$r|jdjcYSwxYw)zJ Get the :class:`.UIControl` to previously had the focus. rN)rr5 IndexErrorr+s r%previous_controlzLayout.previous_control!sF  +;r?* * + + +;r?* * * * +s 33cbt|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. rNrN)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. rrNrcrrindexr*r@r!r#rErvs r% focus_nextzLayout.focus_next24466 w< Focus the previous visible/focusable Window. rrNrurws r%focus_previouszLayout.focus_previousBryrzIterable[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]}||<|dSr4) get_children)r$rFparentsrs 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|jdSr4)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)rKeyErrorr6s r% get_parentzLayout.get_parent}s7  (3 3   44 s  r4)rrrrrr)rr')rr.)rr2)r8r9rr)r8r9rrH)rr)rDrrr)rr )r8r rr)rrH)rrU)rrY)rr])rrd)rhr'rrd)rr)rr})rr rr)__name__ __module__ __qualname____doc__r&r,rr7r!rLpropertyrBsetterr*rTrXr\rcrJrjrlrprsrxr|rr[rrrr`r-r%rrs48 0 0 0 0 0DUUUU$$$$LLLL\2'''X'  E E E EX""""999X9   X QQQQX    555X5+++X+++++'''' '''' ((((     ( ( ( (r-rceZdZdS)rN)rrrr`r-r%rrsDr-rFrr r=rHrr}c#K|r+t|tr|sdS|V|D]}t ||Ed{VdS)z: Walk through layout, starting at this container. Nr<)r0r filterrr)rr=rFs r%rrs  y"6 7 7  ""  OOO  # # % %44{3333333333344r-N)F)rr r=rHrr})r __future__rcollections.abcrrprompt_toolkit.bufferr containersrr r r r controlsr rr__all__r'r9r Exceptionrrr`r-r%rsY#"""""////////((((((DCCCCCCCCC    <)+l:hhhhhhhhV         4444444r-