Add dashboard automation support

ceph/UI/browser.py

@@ -0,0 +1,17 @@
+from selenium import webdriver
+
+
+class Browser:
+    def __init__(self, browser_type: str = "chrome"):
+        """
+        Initializes the Browser with the specified browser type.
+        :param browser_type: The type of browser to use ('chrome', 'firefox', etc.). Defaults to 'chrome'.
+        """
+        if browser_type.lower() == "chrome":
+            self.driver = webdriver.Chrome()
+        elif browser_type.lower() == "firefox":
+            self.driver = webdriver.Firefox()
+        else:
+            raise ValueError(
+                f"Unsupported browser type: {browser_type}. Supported types are 'chrome' and 'firefox'."
+            )

ceph/UI/dashboard.py

@@ -0,0 +1,41 @@
+from selenium.webdriver.support import expected_conditions as EC
+
+# from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+
+from ceph.UI.browser import Browser
+
+# from selenium.common.exceptions import TimeoutException
+from ceph.UI.ids import ElementIDs
+
+
+class Dashboard:
+    def __init__(self, browser_type: str = "chrome"):
+        self.browser = Browser(browser_type)
+        self.driver = self.browser.driver
+        self.element_ids = None  # ElementIDs instance will be set dynamically
+
+    def load_elements(self, yaml_file_path: str):
+        self.element_ids = ElementIDs(yaml_file_path)
+
+    def open(self, url: str):
+        if not isinstance(url, str):
+            raise ValueError("A valid URL must be provided.")
+        self.driver.get(url)
+
+    def is_element_visible(self, key: str, timeout: int = 10) -> bool:
+        element = self.find_element(key, timeout)
+        return element.is_displayed() if element else False
+
+    def find_element(self, key: str, timeout: int = 10):
+        if not self.element_ids:
+            raise RuntimeError("ElementIDs not loaded. Use 'load_elements' first.")
+        return self.element_ids.get_element(key, self.driver, timeout)
+
+    def click(self, key: str, timeout: int = 10):
+        element = self.find_element(key, timeout)
+        WebDriverWait(self.driver, timeout).until(EC.element_to_be_clickable(element))
+        element.click()
+
+    def quit(self):
+        self.driver.quit()