Merge pull request #45 from chavarera/main

local file support added

Author: chavarera

s_tool/utils.py

@@ -22,6 +22,25 @@ def get_session(driver: webdriver) -> str:
     return driver.session_id
 
 
+def is_local(path: str) -> str:
+    """Return valid URL path for file
+
+    Args:
+        path (str): normal path or URL
+
+    Returns:
+        str: Modified file path if local file
+            Returns path as it is if URL
+    """
+    import os
+
+    URL = path
+    if os.path.exists(path) or path.startswith("file"):
+        if not URL.startswith("file"):
+            URL = f"file://{URL}"
+    return URL
+
+
 def visit(driver: webdriver, url: str) -> None:
     """Visit given url
 
@@ -32,7 +51,7 @@ def visit(driver: webdriver, url: str) -> None:
     Returns:
         None
     """
-    driver.get(url)
+    driver.get(is_local(url))
 
 
 def page_source(driver: webdriver) -> str:

tests/test_utils.py

@@ -0,0 +1,17 @@
+"""
+Test related to utilities 
+"""
+import os
+
+from s_tool.utils import is_local
+
+
+def test_local_path():
+    """test local file path or URL"""
+    URL_PATH = "http://www.google.com"
+    URL_PATH1 = "www.google.com"
+    LOCAL_PATH = "tests/index.html"
+
+    assert URL_PATH == is_local(URL_PATH)
+    assert "file" in is_local(os.path.abspath(LOCAL_PATH))
+    assert URL_PATH1 == is_local(URL_PATH1)