File size: 2,403 Bytes
99f00fe | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
from __future__ import annotations
from typing import Any
from leo_ui7m.schema import UIAction
class PlaywrightExecutor:
"""Execute LEO-UI7M UIAction objects on a Playwright sync Page."""
def __init__(self, page) -> None:
self.page = page
self.history: list[dict[str, Any]] = []
self.completed_ids: set[str] = set()
def apply(self, action: UIAction, values: dict[str, str] | None = None) -> dict[str, Any]:
values = values or {}
data = action.to_dict()
act = data["action"]
eid = data.get("target_element_id")
if act == "done":
event = {"type": "done", "action": "done", "target_element_id": None}
self.history.append(event)
return {"ok": True, "event": event}
if act == "wait":
self.page.wait_for_timeout(100)
event = {"type": "wait", "action": "wait", "target_element_id": None}
self.history.append(event)
return {"ok": True, "event": event}
if not eid:
return {"ok": False, "error": "missing_target_element_id", "action": data}
loc = self.page.locator(f"#{eid}")
if loc.count() < 1:
return {"ok": False, "error": "target_not_found", "target_element_id": eid, "action": data}
try:
if act == "type":
text = data.get("text") or values.get(eid, "")
loc.fill(text)
event = {"type": "type", "action": "type", "target_element_id": eid, "text": text}
elif act == "select":
value = data.get("value") or values.get(eid, "")
loc.select_option(label=value)
event = {"type": "select", "action": "select", "target_element_id": eid, "value": value}
elif act == "click":
if loc.get_attribute("type") == "checkbox":
loc.check()
else:
loc.click()
event = {"type": "click", "action": "click", "target_element_id": eid}
else:
return {"ok": False, "error": "unsupported_action", "action": data}
except Exception as e:
return {"ok": False, "error": type(e).__name__, "message": str(e), "action": data}
self.completed_ids.add(eid)
self.history.append(event)
return {"ok": True, "event": event}
|