19 lines
600 B
Python
19 lines
600 B
Python
from typing import List, Dict, Any
|
|
|
|
def get_replay(env, trajectory: List[Dict[str, Any]]) -> None:
|
|
|
|
def parse(action):
|
|
if action["type"] == "hotkey":
|
|
keys = "', '".join(action["param"])
|
|
return f"pyautogui.hotkey('{keys}')"
|
|
|
|
if action["type"] == "typewrite":
|
|
text = action["param"]
|
|
return f"pyautogui.typewrite('{text}')"
|
|
|
|
if action["type"] == "press":
|
|
key = action["param"]
|
|
return f"pyautogui.press('{key}')"
|
|
|
|
for action in trajectory:
|
|
env.controller.execute_python_command(parse(action)) |