From d408b4ec3be8c5a3365df7ffceeb1357cd64816b Mon Sep 17 00:00:00 2001
From: David Rotermund <54365609+davrot@users.noreply.github.com>
Date: Sun, 14 May 2023 20:14:29 +0200
Subject: [PATCH] Add files via upload

---
 content/get_drop_targets.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 content/get_drop_targets.py

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