AHjWLdZddlmZddlZddlmZmZddlZddl m Z ddl m Z m Z ddlmZddlmZmZmZmZmZmZdd lmZmZdd lmZ d dd ZGd d eZGddeZej@jCdGddee Z"y)z This module contains the `Pilot` class used by [App.run_test][textual.app.App.run_test] to programmatically operate an app. See the guide on how to [test Textual apps](/guide/testing). ) annotationsN)AnyGeneric) wait_for_idle)App ReturnType)HeadlessDriver)Click MouseDown MouseEvent MouseMoveMouseUpResize)OffsetSize)Widgetc Z|jj|z\}}|||dd||||||d }|S)zNGet the arguments to pass into mouse messages for the click and hover methods.r) widgetxydelta_xdelta_ybuttonshiftmetactrlscreen_xscreen_y)regionoffset) targetr rrrcontrolclick_xclick_ymessage_argumentss E/root/tools/cai/cai_env/lib/python3.12/site-packages/textual/pilot.py_get_mouse_message_argumentsr'sN}}++f4GW    ceZdZdZy) OutOfBoundszFRaised when the pilot mouse target is outside of the (visible) screen.N__name__ __module__ __qualname____doc__r(r&r*r*2sPr(r*ceZdZdZy)WaitForScreenTimeoutzException raised if messages aren't being processed quickly enough. If this occurs, the most likely explanation is some kind of deadlock in the app code. Nr+r0r(r&r2r26sr(r2T)angularceZdZdZddZddZeddZddZddZ d ddZ d dd Z d dd Z d dd Z d dd Z d dd Z d d dZd!d"dZd#d$dZd%dZd%dZd&dZy)'PilotzPilot object to drive an app.c||_yN_app)selfapps r&__init__zPilot.__init__As  r(c#,Kd|jfyw)Nr;r8r:s r& __rich_repr__zPilot.__rich_repr__DsTYYsc|jS)z$App: A reference to the application.r8r>s r&r;z Pilot.appGsyyr(cK|r<|jj|d{|jd{yy77w)zOSimulate key-presses. Args: *keys: Keys to press. N)r9 _press_keys_wait_for_screen)r:keyss r&pressz Pilot.pressLsA ))''- - -'') ) )  - )s!AAAAAAc Kt||}t|jjtr||jj_|jj t|||jd{y7w)zResize the terminal to the given dimensions. Args: width: The new width of the terminal. height: The new height of the terminal. N) r isinstancer;_driverr _size post_messagerpause)r:widthheightsizes r&resize_terminalzPilot.resize_terminalVsbE6" dhh&& 7%)DHH   " fT401jjlsBBB BNc K |jttg||d|||d{S7#t$r}|dd}~wwxYww)aSimulate a [`MouseDown`][textual.events.MouseDown] event at a specified position. The final position for the event is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Args: widget: A widget or selector used as an origin for the event offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to target a specific widget. However, if the widget is currently hidden or obscured by another widget, the event may not land on the widget you specified. offset: The offset for the event. The offset is relative to the selector / widget provided or to the screen, if no selector is provided. shift: Simulate the event with the shift key held down. meta: Simulate the event with the meta key held down. control: Simulate the event with the control key held down. Raises: OutOfBounds: If the position for the event is outside of the (visible) screen. Returns: True if no selector was specified or if the event landed on the selected widget, False otherwise. rr rrrr"N)_post_mouse_eventsr r r*r:rr rrr"errors r& mouse_downzPilot.mouse_downdsa@ "00I&1  "T ! ".A%0.0A0 A<AAc K |jttg||d|||d{S7#t$r}|dd}~wwxYww)aSimulate a [`MouseUp`][textual.events.MouseUp] event at a specified position. The final position for the event is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Args: widget: A widget or selector used as an origin for the event offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to target a specific widget. However, if the widget is currently hidden or obscured by another widget, the event may not land on the widget you specified. offset: The offset for the event. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. shift: Simulate the event with the shift key held down. meta: Simulate the event with the meta key held down. control: Simulate the event with the control key held down. Raises: OutOfBounds: If the position for the event is outside of the (visible) screen. Returns: True if no selector was specified or if the event landed on the selected widget, False otherwise. rQrRN)rSr rr*rTs r&mouse_upzPilot.mouse_upsa@ "00G$1  "T ! "rWc K |jtttg||d||||d{S7#t$r}|dd}~wwxYww)aSimulate clicking with the mouse at a specified position. The final position to be clicked is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Implementation note: This method bypasses the normal event processing in `App.on_event`. Example: The code below runs an app and clicks its only button right in the middle: ```py async with SingleButtonApp().run_test() as pilot: await pilot.click(Button, offset=(8, 1)) ``` Args: widget: A widget or selector used as an origin for the click offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to click on a specific widget. However, if the widget is currently hidden or obscured by another widget, the click may not land on the widget you specified. offset: The offset to click. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. shift: Click with the shift key held down. meta: Click with the meta key held down. control: Click with the control key held down. times: The number of times to click. 2 will double-click, 3 will triple-click, etc. Raises: OutOfBounds: If the position to be clicked is outside of the (visible) screen. Returns: `True` if no selector was specified or if the selected widget was under the mouse when the click was initiated. `False` is the selected widget was not under the pointer. rQ)rr rrrr"timesN)rSr rr r*)r:rr rrr"r[rUs r&clickz Pilot.clicksfV "00GU+1     "T ! "s0A +646A 6 AAAA cLK|j|||||dd{S7w)aSimulate double clicking with the mouse at a specified position. Alias for `pilot.click(..., times=2)`. The final position to be clicked is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Implementation note: This method bypasses the normal event processing in `App.on_event`. Example: The code below runs an app and double-clicks its only button right in the middle: ```py async with SingleButtonApp().run_test() as pilot: await pilot.double_click(Button, offset=(8, 1)) ``` Args: widget: A widget or selector used as an origin for the click offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to click on a specific widget. However, if the widget is currently hidden or obscured by another widget, the click may not land on the widget you specified. offset: The offset to click. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. shift: Click with the shift key held down. meta: Click with the meta key held down. control: Click with the control key held down. Raises: OutOfBounds: If the position to be clicked is outside of the (visible) screen. Returns: `True` if no selector was specified or if the selected widget was under the mouse when the click was initiated. `False` is the selected widget was not under the pointer. r[Nr\r:rr rrr"s r& double_clickzPilot.double_click*VZZtWAZNNNN $"$cLK|j|||||dd{S7w)aSimulate triple clicking with the mouse at a specified position. Alias for `pilot.click(..., times=3)`. The final position to be clicked is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Implementation note: This method bypasses the normal event processing in `App.on_event`. Example: The code below runs an app and triple-clicks its only button right in the middle: ```py async with SingleButtonApp().run_test() as pilot: await pilot.triple_click(Button, offset=(8, 1)) ``` Args: widget: A widget or selector used as an origin for the click offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to click on a specific widget. However, if the widget is currently hidden or obscured by another widget, the click may not land on the widget you specified. offset: The offset to click. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. shift: Click with the shift key held down. meta: Click with the meta key held down. control: Click with the control key held down. Raises: OutOfBounds: If the position to be clicked is outside of the (visible) screen. Returns: `True` if no selector was specified or if the selected widget was under the mouse when the click was initiated. `False` is the selected widget was not under the pointer. r_Nr`ras r& triple_clickzPilot.triple_click$rcrdcK|jd{ |jtg||dd{S7'7#t$r}|dd}~wwxYww)a9Simulate hovering with the mouse cursor at a specified position. The final position to be hovered is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. Args: widget: A widget or selector used as an origin for the hover offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to hover a specific widget. However, if the widget is currently hidden or obscured by another widget, the hover may not land on the widget you specified. offset: The offset to hover. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. Raises: OutOfBounds: If the position to be hovered is outside of the (visible) screen. Returns: True if no selector was specified or if the hover landed on the selected widget, False otherwise. Nr)r)rKrSr r*)r:rr rUs r&hoverz Pilot.hoverQs\8jjl "00)ffUV0WW W W "T ! "s@A>AAAAAA A AAAc FK|j} | j} || } n$t|tr|} n| j |} t | |||||} t | d| d}|| jvr tdd} td|dzD]g}|D]`}|jd{| }|turi|d|i}| | j|\} }|di|}|| _ | j|bi|jd{|duxs| | uS7p7w) aSimulate a series of mouse events to be fired at a given position. The final position for the events is computed based on the selector provided and the offset specified and it must be within the visible area of the screen. This function abstracts away the commonalities of the other mouse event-related functions that the pilot exposes. Args: widget: A widget or selector used as the origin for the event's offset. If this is not specified, the offset is interpreted relative to the screen. You can use this parameter to try to target a specific widget. However, if the widget is currently hidden or obscured by another widget, the events may not land on the widget you specified. offset: The offset for the events. The offset is relative to the widget / selector provided or to the screen, if no selector is provided. shift: Simulate the events with the shift key held down. meta: Simulate the events with the meta key held down. control: Simulate the events with the control key held down. times: The number of times to click. 2 will double-click, 3 will triple-click, etc. Raises: OutOfBounds: If the position for the events is outside of the (visible) screen. Returns: True if no selector was specified or if the *final* event landed on the selected widget, False otherwise. N)rrrr"rrz"M  '"M",,V4M8    )#.0A#0FG  &N  1eai( -E#) -jjl""+"e+777F$#43#4#4f#=LIq'1&1 &,"%%e,) - -.jjl~;m!;;-#* s%B*D!,D-A!D!DD!D!c K |jj}|jg|jd}d t j d fd }|D]}|j |s dz  rt j jt j|jjjg}t j||tjd{\}}|D]} | jt|t|k(} | r td  dkDryy#t$rYywxYw7Xw) aWait for the current screen and its children to have processed all pending events. Args: timeout: A timeout in seconds to wait. Returns: `True` if all events were processed. `False` if an exception occurred, meaning that not all events could be processed. Raises: WaitForScreenTimeout: If the screen and its children didn't finish processing within the timeout. FT) with_selfrc<dzdk(rjyy)z@Decrement internal counter, and set an event if it reaches zero.rQrN)set)countcount_zero_eventsr&decrement_counterz1Pilot._wait_for_screen..decrement_counters& QJEz $$&r(rQ)timeout return_whenNz@Timed out while waiting for widgets to process pending messages.returnNone)r;rl Exception walk_childrenasyncioEvent call_later create_taskwait_exception_eventFIRST_COMPLETEDcancellenr2) r:rrlchildrenrchildwait_forrwpendingtask timed_outr}r~s @@r&rCzPilot._wait_for_screensR XX__FHHDv33d3CD"==? ' E 12   ##$4$9$9$;<##DHH$=$=$B$B$DEH '||#33 JAw     H W5I*V qy[  2s<EEAE/B E;Es r&wait_for_animationzPilot.wait_for_animations;ii  ..000 ((* 1s(AA*Ac@K|jd{|jjjd{|jd{t d{|j j jy77U7?7/w)z:Wait for any current and scheduled animations to complete.N)rCr9rwait_until_completerr;rlrr>s r&wait_for_scheduled_animationsz#Pilot.wait_for_scheduled_animationssv##%%%ii  44666##%%%o ((* &6%sEBB+BBBBB,B-*BBBBcK|jd{td{|jj|y727"w)zExit the app with the given result. Args: result: The app result returned by `run` or `run_async`. N)rCrr;exit)r:results r&rz Pilot.exit&s> ##%%%o  f &sA A A A !A  A )r;App[ReturnType]rr)rzrich.repr.Result)rr)rDstrrr)rLintrMrrr)NrrFFF) r"Widget | type[Widget] | str | Noner tuple[int, int]rboolrrr"rrr)NrFFFrQ)rrr rrrrrr"rr[rrr)Nr)r)Widget | type[Widget] | str | None | Noner rrr)NrrFFFrQ)rrzlist[type[MouseEvent]]rrr rrrrrrrr"rr[rrr)g>@)rfloatrrr7)rz float | Nonerrr)rrrr)r,r-r.r/r<r?propertyr;rErOrVrYr\rbrgrirSrCrKrrrr0r(r&r5r5=s'*  6:"( +"2+" +" +"  +"  +" +"^6:"( +"2+" +" +"  +"  +" +"^6:"(7"27" 7" 7"  7"  7"7" 7"v6:"( +O2+O +O +O  +O  +O +O^6:"( +O2+O +O +O  +O  +O +O^=A"( "9 "  "  "J=A"(X<&X<:X< X<  X<  X<X<X<X< Xrs#'':SS)! %          4Q)Q9pGJ ppr(