Merge pull request #247 from jasongrout/virtualenv

Check if site.getuserbase() exists
jupyter · Oct 27, 2021 · be863e3 · be863e3
2 parents 9aee793 + 067a230
commit be863e3
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/jupyter_core/paths.py b/jupyter_core/paths.py
@@ -168,7 +168,13 @@ def jupyter_path(*subdirs):
     # Next is environment or user, depending on the JUPYTER_PREFER_ENV_PATH flag
     user = [jupyter_data_dir()]
     if site.ENABLE_USER_SITE:
-        userdir = os.path.join(site.getuserbase(), 'share', 'jupyter')
+        # Check if site.getuserbase() exists to be compatible with virtualenv,
+        # which often does not have this method.
+        if hasattr(site, 'getuserbase'):
+            userbase = site.getuserbase()
+        else:
+            userbase = site.USER_BASE
+        userdir = os.path.join(userbase, 'share', 'jupyter')
         if userdir not in user:
             user.append(userdir)
 
@@ -231,7 +237,14 @@ def jupyter_config_path():
     # Next is environment or user, depending on the JUPYTER_PREFER_ENV_PATH flag
     user = [jupyter_config_dir()]
     if site.ENABLE_USER_SITE:
-        userdir = os.path.join(site.getuserbase(), 'etc', 'jupyter')
+        # Check if site.getuserbase() exists to be compatible with virtualenv,
+        # which often does not have this method.
+        if hasattr(site, 'getuserbase'):
+            userbase = site.getuserbase()
+        else:
+            userbase = site.USER_BASE
+
+        userdir = os.path.join(userbase, 'etc', 'jupyter')
         if userdir not in user:
             user.append(userdir)