diff --git a/content/get_drop_targets.py b/content/get_drop_targets.py new file mode 100644 index 0000000..1a1d720 --- /dev/null +++ b/content/get_drop_targets.py @@ -0,0 +1,30 @@ +from selenium import webdriver +from selenium.webdriver.common.by import By + + +def get_drop_targets( + driver: webdriver.firefox.webdriver.WebDriver, +) -> tuple[list[int], list[int], list[str]]: + find_string_id_1: str = "colpos-" + find_string_image_1: str = "actions-add.svg" + + found_element_list = driver.find_elements(By.TAG_NAME, "div") + + center_x: list[int] = [] + center_y: list[int] = [] + id_name: list[str] = [] + + for i in found_element_list: + if str(i.get_dom_attribute("id")).startswith(find_string_id_1): + found_child_list = i.find_elements(By.TAG_NAME, "img") + for j in found_child_list: + src_info = j.get_dom_attribute("src") + src_info = src_info.split("/")[-1] + if str(src_info).startswith(find_string_image_1): + x = j.location["x"] + j.size["width"] // 2 + y = j.location["y"] + j.size["height"] // 2 + center_x.append(int(x)) + center_y.append(int(y)) + id_name.append(i.get_dom_attribute("id")) + + return center_x, center_y, id_name