fix(constants): use cwd for project cache path in PyInstaller/frozen mode (#720)

Author: aleksandr-kotlyar

tests/test_constants.py

@@ -2,6 +2,7 @@
 import os
 
 from webdriver_manager.core.constants import ROOT_FOLDER_NAME, get_default_user_home_cache_path
+from webdriver_manager.core.constants import get_default_project_root_cache_path
 
 
 def test_default_user_home_cache_path_falls_back_to_temp_if_home_is_root(monkeypatch):
@@ -18,3 +19,13 @@ def test_default_user_home_cache_path_falls_back_to_temp_if_home_is_unresolved(m
     cache_path = get_default_user_home_cache_path()
 
     assert cache_path == os.path.join(tempfile.gettempdir(), ROOT_FOLDER_NAME)
+
+
+def test_default_project_root_cache_path_uses_cwd_for_frozen_app(monkeypatch):
+    monkeypatch.setattr("sys.path", ["C:/app/base_library.zip"])
+    monkeypatch.setattr("sys.frozen", True, raising=False)
+    monkeypatch.setattr("os.getcwd", lambda: "C:/app/runtime")
+
+    cache_path = get_default_project_root_cache_path()
+
+    assert cache_path == os.path.join("C:/app/runtime", ROOT_FOLDER_NAME)

webdriver_manager/core/constants.py

@@ -3,7 +3,16 @@
 import tempfile
 
 ROOT_FOLDER_NAME = ".wdm"
-DEFAULT_PROJECT_ROOT_CACHE_PATH = os.path.join(sys.path[0], ROOT_FOLDER_NAME)
+
+
+def get_default_project_root_cache_path():
+    base_path = sys.path[0]
+    if getattr(sys, "frozen", False) or base_path.endswith(".zip"):
+        base_path = os.getcwd()
+    return os.path.join(base_path, ROOT_FOLDER_NAME)
+
+
+DEFAULT_PROJECT_ROOT_CACHE_PATH = get_default_project_root_cache_path()
 
 
 def get_default_user_home_cache_path():