o ˷e6@sdZddlmZmZmZmZmZmZddlm Z ddl m Z m Z m Z mZmZddlmZmZmZgdZeee ee fZGdd d ZGd d d eZdd e dedee fddZdS)z Wrapper for the layout. )Dict GeneratorIterableListOptionalUnion)Buffer) AnyContainerConditionalContainer ContainerWindow to_container) BufferControlSearchBufferControl UIControl)LayoutInvalidLayoutErrorwalkc@seZdZdZ d;dedeeddfddZdefdd Z de e ddffd d Z de efd d ZdeddfddZdedefddZedefddZejdeddfddZede fddZejde ddfddZedefddZedeefddZde e fddZdee fd d!Zedeefd"d#Zd$edeefd%d&Zedefd'd(Z edefd)d*Z!dInvalid layout. The layout does not contain any Window object.) rr_stack search_links_child_to_parentappendnextfind_all_windows StopIterationrfocusvisible_windows)selfrrer#S/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/layout/layout.py__init__%s"   zLayout.__init__cCsd|jd|jdS)NzLayout(z, current_window=))rcurrent_windowr!r#r#r$__repr__HszLayout.__repr__ccs$|D] }t|tr|VqdS)zJ Find all the :class:`.UIControl` objects in this layout. N)r isinstancer )r!itemr#r#r$rKs   zLayout.find_all_windowsccs|D]}|jVqdSN)rcontentr!rr#r#r$find_all_controlsSs  zLayout.find_all_controlsvaluecCst|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|f||_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.z7Invalid value. Window does not appear in the layout: %rT skip_hiddenrz,Invalid value. Container cannot be focused: )r*strr/rbuffernamer ValueErrorrr is_focusablecurrent_controlrr rr'rr-rreversedr)r!r0controlwindowscwr#r#r$rWs^              z Layout.focuscCst|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*r4current_bufferr6rrr9rr r'r)r!r0elementr#r#r$ has_focuss           zLayout.has_focuscCs |jdjS)zI Get the :class:`.UIControl` to currently has the focus. )rr-r(r#r#r$r9s 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'r7)r!r;windowr#r#r$r9s  cCs |jdS)z=Return the :class:`.Window` object that is currently focused.rB)rr(r#r#r$r's zLayout.current_windowcCs|j|dS)z8Set the :class:`.Window` object to be currently focused.N)rr)r!r0r#r#r$r'scCs |j|jvS)z#True if we are searching right now.)r9rr(r#r#r$ is_searchings zLayout.is_searchingcCs |j}t|tr|j|SdS)zY Return the :class:`.BufferControl` in which we are searching or `None`. N)r9r*rrget)r!r;r#r#r$search_target_buffer_controls  z#Layout.search_target_buffer_controlccs.|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-r8)r!r>r#r#r$get_focusable_windowss  zLayout.get_focusable_windowscs|jfdd|DS)zO Return a list of :class:`.Window` objects that are focusable. csg|]}|vr|qSr#r#).0r>r r#r$ sz8Layout.get_visible_focusable_windows..)r rHr(r#rJr$get_visible_focusable_windowssz$Layout.get_visible_focusable_windowscCs|j}t|tr |jSdS)zD The currently focused :class:`~.Buffer` or `None`. N)r9r*rr5r! ui_controlr#r#r$r?s 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-rr5r6)r!rOr>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.) )r9r*rrMr#r#r$buffer_has_focuss zLayout.buffer_has_focuscCs.z|jdjWSty|jdjYSw)zJ Get the :class:`.UIControl` to previously had the focus. rB)rr- IndexErrorr(r#r#r$previous_control s  zLayout.previous_controlcCs&t|jdkr|jdd|_dSdS)z= Give the focus to the last focused control. r NrB)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. rr NrLrUindexr'r7rr!r<rXr#r#r$ focus_next1  zLayout.focus_nextcCsb|}t|dkr/z||j}Wn tyd}Yn w|dt|}|||dSdS)z> Focus the previous visible/focusable Window. rr NrWrYr#r#r$focus_previousAr[zLayout.focus_previousccst|jEdHdS)zX Walk through all the layout nodes (and their children) and yield them. N)rrr(r#r#r$rQsz 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$rGWs  zLayout.walk_through_modal_areacs0idtddffdd |j|_dS)z= Update child->parent relationships mapping. r"rNcs"|D] }||<|qdSr,) get_children)r"r=parentsrr#r$rjs  z-Layout.update_parents_relations..walk)r rrr(r#r`r$update_parents_relationsds  zLayout.update_parents_relationscCs|j|jdSr,)rclearrresetr(r#r#r$rdss z Layout.resetcCs"z|j|WStyYdSw)zo Return the parent container for the given container, or ``None``, if it wasn't found. N)rKeyErrorr.r#r#r$ get_parent|s   zLayout.get_parentr,)rN)+__name__ __module__ __qualname____doc__r rFocusableElementr%r4r)rr rrrr/rboolrApropertyr9setterr'rDrrFrHrrLrr?rPrQrTrVrZr\r rrGrbrdrfr#r#r#r$rsX  #O           rc@s eZdZdS)rN)rgrhrir#r#r#r$rsrFrr3rccsF|rt|tr|sdS|V|D] }t||dEdHqdS)z: Walk through layout, starting at this container. Nr2)r*r filterr_r)rr3r=r#r#r$rs rN)F)rjtypingrrrrrrprompt_toolkit.bufferr containersr r r r rcontrolsrrr__all__r4rkr Exceptionrrlrr#r#r#r$s  o