refactor select handler (#565)

2024-07-09 02:22:16 +08:00
parent 7a75bb3741
commit bcbd738326
4 changed files with 309 additions and 113 deletions
--- a/skyvern/webeye/scraper/scraper.py
+++ b/skyvern/webeye/scraper/scraper.py
@@ -324,6 +324,12 @@ async def get_select2_options(frame: Page | Frame, element: ElementHandle) -> li
    return await frame.evaluate(js_script, element)


+async def get_combobox_options(frame: Page | Frame, element: ElementHandle) -> list[dict[str, Any]]:
+    await frame.evaluate(JS_FUNCTION_DEFS)
+    js_script = "async (element) => await getListboxOptions(element)"
+    return await frame.evaluate(js_script, element)
+
+
 async def get_interactable_element_tree_in_frame(
    frames: list[Frame],
    elements: list[dict],