table parser added

Author: ravishankar

s_tool/parser.py

@@ -1,3 +1,4 @@
+from s_tool.exceptions import SToolException
 from s_tool.utils import get_element
 
 
@@ -14,7 +15,7 @@ def select_options(element, swap=None, text_exclude=[]):
     """
     option_dict = dict()
     if element and hasattr(element, "tag_name") and element.tag_name == "select":
-        options = get_element(element, "tag_name", "option", many=True)
+        options = get_element(element, "option", "tag_name", many=True)
         for option in options:
             func = option.get_attribute
             text, value = func("text"), func("value")
@@ -24,3 +25,31 @@ def select_options(element, swap=None, text_exclude=[]):
                 option_dict[value] = text
 
     return option_dict
+
+
+def get_table(table):
+    """Return list of rows including header and footer of given table
+
+    Args:
+        A selenium table element
+
+    Returns:
+        return list of row list,
+        return [] if table not valid
+
+    """
+    results = []
+    if table and hasattr(table, "tag_name") and table.tag_name == "table":
+        for rows in table.find_elements_by_tag_name("tr"):
+            cell_data = []
+            for cell in rows.find_elements_by_xpath("td | th"):
+                colspan = cell.get_attribute("colspan")
+                cell_text = cell.text
+                if colspan:
+                    cell_data.extend([cell_text] + [""] * (int(colspan) - 1))
+                else:
+                    cell_data.append(cell_text)
+            results.append(cell_data)
+    else:
+        raise SToolException("INVALIDTABLE")
+    return results