Merge pull request #263 from Progress1/selector_fix

Fix selector for some element types
SK-CERT · Apr 19, 2024 · 0b8a9b3 · 0b8a9b3
2 parents 864679c + 04e6024
commit 0b8a9b3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/collectors/collectors/web_collector.py b/src/collectors/collectors/web_collector.py
@@ -183,13 +183,13 @@ def __find_elements_by(driver, element_selector):
         if prefix == 'id':
             elements = [ driver.find_element(By.ID, selector) ]
         if prefix == 'name':
-            elements = driver.find_element(By.NAME, selector)
+            elements = driver.find_elements(By.NAME, selector)
         elif prefix == 'xpath':
-            elements = driver.find_element(By.XPATH, selector)
+            elements = driver.find_elements(By.XPATH, selector)
         elif prefix in [ 'tag_name', 'tag' ]:
             elements = driver.find_elements(By.TAG_NAME, selector)
         elif prefix in [ 'class_name', 'class' ]:
-            elements = driver.find_element(By.CLASS_NAME, selector)
+            elements = driver.find_elements(By.CLASS_NAME, selector)
         elif prefix in [ 'css_selector', 'css' ]:
             elements = driver.find_elements(By.CSS_SELECTOR, selector)
         return elements