Merge pull request #36 from chavarera/main

user agent added

Author: chavarera

s_tool/driver.py

@@ -13,6 +13,7 @@
     get_cookies,
     get_element,
     get_session,
+    get_user_agent,
     hide_show_elements,
     is_element,
     page_source,
@@ -96,6 +97,7 @@ def _load_methods(self):
         self.fill = partial(fill, self.driver)
         self.is_element = partial(is_element, self.driver)
         self.set_cookies = partial(set_cookies, self.driver)
+        self.ua = partial(get_user_agent, self.driver)
 
 
 if __name__ == "__main__":

s_tool/utils.py

@@ -311,3 +311,18 @@ def is_element(driver, locator_text, locator_type="id", wait_time=2):
         return True
     except:
         return False
+
+
+def get_user_agent(driver: webdriver) -> str:
+    """Return current user agent string
+
+    Args:
+        driver  : selenium web driver
+
+    Retuns:
+        str
+
+    Usage:
+        get_user_agent()
+    """
+    return driver.execute_script("return navigator.userAgent")

tests/test_driver.py

@@ -39,3 +39,6 @@ def test_base():
 
         # page_source
         assert len(obj.text()) > 100
+
+        # user agent type
+        assert type(obj.ua()) is str