1j01/textual-paint
1
1
Fork 0
mirror of https://github.com/1j01/textual-paint.git synced 2025-01-03 04:16:48 +03:00
Code Issues Packages Projects Releases Wiki Activity

WIP: handle edge cases

Browse Source
This commit is contained in:
Isaiah Odhner 2023-09-09 21:46:03 -04:00
parent 2fb0dc2467
commit 3cebba06ec
1 changed files with 15 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/pilot_recorder.py
View File

@ -25,6 +25,7 @@ def indent(text: str, spaces: int) -> str:
OUTPUT_FILE = unique_file("tests/test_paint_something.py") OUTPUT_FILE = unique_file("tests/test_paint_something.py")
steps: list[tuple[Event, Offset, str, int|None]] = [] steps: list[tuple[Event, Offset, str, int|None]] = []
replaying: bool = False
def get_selector(target: DOMNode) -> tuple[str, int|None]: def get_selector(target: DOMNode) -> tuple[str, int|None]:
"""Return a selector that can be used to find the widget.""" """Return a selector that can be used to find the widget."""
@ -43,6 +44,9 @@ def get_selector(target: DOMNode) -> tuple[str, int|None]:
try: try:
query_result = app.query_one(selector) query_result = app.query_one(selector)
except TooManyMatches: except TooManyMatches:
# FIXME: I think this can fail due to a race condition,
# when clicking a button that closes a dialog, removing the button from the DOM.
# Maybe intercept events differently, like by overriding post_message?
return selector, app.query(selector).nodes.index(target) return selector, app.query(selector).nodes.index(target)
# smarter differentiators would be nice, like tooltip or text content, # smarter differentiators would be nice, like tooltip or text content,
# but at least with indices, you'll know when you changed the tab order # but at least with indices, you'll know when you changed the tab order
@ -56,6 +60,8 @@ def get_selector(target: DOMNode) -> tuple[str, int|None]:
original_on_event = PaintApp.on_event original_on_event = PaintApp.on_event
async def on_event(self: PaintApp, event: Event) -> None: async def on_event(self: PaintApp, event: Event) -> None:
await original_on_event(self, event) await original_on_event(self, event)
if replaying:
return
if isinstance(event, (MouseDown, MouseMove, MouseUp)): if isinstance(event, (MouseDown, MouseMove, MouseUp)):
try: try:
widget, _ = self.get_widget_at(*event.screen_offset) widget, _ = self.get_widget_at(*event.screen_offset)
@ -64,7 +70,7 @@ async def on_event(self: PaintApp, event: Event) -> None:
offset = event.screen_offset - widget.region.offset offset = event.screen_offset - widget.region.offset
steps.append((event, offset, *get_selector(widget))) steps.append((event, offset, *get_selector(widget)))
elif isinstance(event, Key): elif isinstance(event, Key):
if event.key == "ctrl+z": if event.key == "ctrl+z" and steps:
steps.pop() steps.pop()
run() # restart the app to replay up to this point run() # restart the app to replay up to this point
elif event.key == "ctrl+c": elif event.key == "ctrl+c":
@ -85,7 +91,7 @@ async def async_exec(code: str, **kwargs: object) -> object:
return await scope['async_exec_code']() return await scope['async_exec_code']()
async def replay_steps(pilot: Pilot[Any]) -> None: async def replay_steps(pilot: Pilot[Any]) -> None:
global app global app, replaying
assert app is not None, "app should be set by now" assert app is not None, "app should be set by now"
# for event, offset, selector, index in steps: # for event, offset, selector, index in steps:
# ... # ...
@ -93,7 +99,13 @@ async def replay_steps(pilot: Pilot[Any]) -> None:
return return
# pilot = Pilot(app) # pilot = Pilot(app)
# await pilot._wait_for_screen() # await pilot._wait_for_screen()
replaying = True
await async_exec(get_replay_code(), pilot=pilot, Offset=Offset) await async_exec(get_replay_code(), pilot=pilot, Offset=Offset)
replaying = False
# def clear_replaying_flag() -> None:
# global replaying
# replaying = False
# app.set_timer(1.0, clear_replaying_flag)
def run() -> None: def run() -> None:
global app, next_after_exit global app, next_after_exit
@ -136,7 +148,7 @@ def get_replay_code() -> str:
steps_code += f"await pilot.press({event.key!r})\n" steps_code += f"await pilot.press({event.key!r})\n"
else: else:
raise Exception(f"Unexpected event type {type(event)}") raise Exception(f"Unexpected event type {type(event)}")
return steps_code return steps_code or "pass"
def save_replay() -> None: def save_replay() -> None: