Add get_parent() to the SB / CDP APIs

Author: mdmintz

seleniumbase/core/browser_launcher.py

@@ -220,6 +220,7 @@ def extend_driver(driver):
     driver.highlight_if_visible = DM.highlight_if_visible
     driver.sleep = time.sleep
     driver.get_attribute = DM.get_attribute
+    driver.get_parent = DM.get_parent
     driver.get_current_url = DM.get_current_url
     driver.get_page_source = DM.get_page_source
     driver.get_title = DM.get_title
@@ -647,6 +648,7 @@ def uc_open_with_cdp_mode(driver, url=None):
     cdp.click_if_visible = CDPM.click_if_visible
     cdp.click_visible_elements = CDPM.click_visible_elements
     cdp.mouse_click = CDPM.mouse_click
+    cdp.get_parent = CDPM.get_parent
     cdp.remove_element = CDPM.remove_element
     cdp.remove_from_dom = CDPM.remove_from_dom
     cdp.remove_elements = CDPM.remove_elements

seleniumbase/core/sb_cdp.py

@@ -91,6 +91,8 @@ def __add_sync_methods(self, element):
         element.get_attribute = (
             lambda attribute: self.__get_attribute(element, attribute)
         )
+        # element.get_parent() should come last
+        element.get_parent = lambda: self.__get_parent(element)
         return element
 
     def get(self, url):
@@ -549,6 +551,9 @@ def __get_attribute(self, element, attribute):
                 pass
         return None
 
+    def __get_parent(self, element):
+        return self.__add_sync_methods(element.parent)
+
     def __get_x_scroll_offset(self):
         x_scroll_offset = self.loop.run_until_complete(
             self.page.evaluate("window.pageXOffset")
@@ -769,6 +774,11 @@ def focus(self, selector):
     def highlight_overlay(self, selector):
         self.find_element(selector).highlight_overlay()
 
+    def get_parent(self, element):
+        if isinstance(element, str):
+            element = self.select(element)
+        return self.__add_sync_methods(element.parent)
+
     def remove_element(self, selector):
         self.select(selector).remove_from_dom()
 

seleniumbase/core/sb_driver.py

@@ -51,6 +51,13 @@ def get_attribute(self, selector, attribute, by="css selector"):
         element = self.locator(selector, by=by)
         return element.get_attribute(attribute)
 
+    def get_parent(self, element):
+        if self.__is_cdp_swap_needed():
+            return self.driver.cdp.get_parent(element)
+        if isinstance(element, str):
+            element = self.locator(element)
+        return element.find_element(by="xpath", value="..")
+
     def get_current_url(self):
         if self.__is_cdp_swap_needed():
             current_url = self.driver.cdp.get_current_url()

seleniumbase/fixtures/base_case.py

@@ -2072,6 +2072,19 @@ def internalize_links(self):
             return
         self.set_attributes('[target="_blank"]', "target", "_self")
 
+    def get_parent(self, element, by="css selector", timeout=None):
+        """Returns the parent element.
+        If element is a string, then finds element first via selector."""
+        if self.__is_cdp_swap_needed():
+            return self.cdp.get_parent(element)
+        if isinstance(element, str):
+            if not timeout:
+                timeout = settings.LARGE_TIMEOUT
+            element = self.wait_for_element_present(
+                element, by=by, timeout=timeout
+            )
+        return element.find_element(by="xpath", value="..")
+
     def get_property(
         self, selector, property, by="css selector", timeout=None
     ):