[py] convert selectors for finding elements in shadow dom

titusfortner · titusfortner · commit d53d79e35b16 · 2021-12-29T10:19:18.000-06:00
diff --git a/py/selenium/webdriver/remote/shadowroot.py b/py/selenium/webdriver/remote/shadowroot.py
@@ -44,11 +44,35 @@ def __repr__(self):
         )
 
     def find_element(self, by: By = By.ID, value: str = None):
+        if by == By.ID:
+            by = By.CSS_SELECTOR
+            value = '[id="%s"]' % value
+        elif by == By.TAG_NAME:
+            by = By.CSS_SELECTOR
+        elif by == By.CLASS_NAME:
+            by = By.CSS_SELECTOR
+            value = ".%s" % value
+        elif by == By.NAME:
+            by = By.CSS_SELECTOR
+            value = '[name="%s"]' % value
+
         return self._execute(
             Command.FIND_ELEMENT_FROM_SHADOW_ROOT, {"using": by, "value": value}
         )["value"]
 
     def find_elements(self, by: By = By.ID, value: str = None):
+        if by == By.ID:
+            by = By.CSS_SELECTOR
+            value = '[id="%s"]' % value
+        elif by == By.TAG_NAME:
+            by = By.CSS_SELECTOR
+        elif by == By.CLASS_NAME:
+            by = By.CSS_SELECTOR
+            value = ".%s" % value
+        elif by == By.NAME:
+            by = By.CSS_SELECTOR
+            value = '[name="%s"]' % value
+
         return self._execute(
             Command.FIND_ELEMENTS_FROM_SHADOW_ROOT, {"using": by, "value": value}
         )["value"]
