Merge pull request #57 from chavarera/main

[#56] added functionality to execute JavaScript statement

Author: chavarera

README.md

@@ -33,6 +33,7 @@
   - Add or modify existing cookies.
   - Retrieve current user agent.
   - Check Existence of an element on the page.
+  - Execute JavaScript statment.
 - Element Parser
   - table Information.
   - Retrieve dropdown options in the dictionary.

s_tool/driver.py

@@ -18,6 +18,7 @@
     hide_show_elements,
     is_element,
     page_source,
+    run_js,
     set_cookies,
     take_screenshot,
     visit,
@@ -159,6 +160,7 @@ def _load_methods(self):
         self.is_element = partial(is_element, self.driver)
         self.set_cookies = partial(set_cookies, self.driver)
         self.ua = partial(get_user_agent, self.driver)
+        self.js = partial(run_js, self.driver)
 
 
 if __name__ == "__main__":

s_tool/utils.py

@@ -377,3 +377,20 @@ def get_user_agent(driver: webdriver) -> str:
     """
 
     return driver.execute_script("return navigator.userAgent")
+
+
+def run_js(driver: webdriver, statement: str) -> str:
+    """Execute an javascript statement and return output
+
+    Args:
+        driver (webdriver): selenium webdriver
+        statement ([str]): and variable name or an statement to run
+
+    Returns:
+        str: console output in str format
+
+    Example:
+        Fetch URL
+            >>run_js(driver,document.URL)
+    """
+    return driver.execute_script(f"return {statement}")