summaryrefslogtreecommitdiff
path: root/scripts/install_api.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/install_api.py')
-rwxr-xr-xscripts/install_api.py9
1 files changed, 8 insertions, 1 deletions
diff --git a/scripts/install_api.py b/scripts/install_api.py
index 1237e1e5..646bd4fe 100755
--- a/scripts/install_api.py
+++ b/scripts/install_api.py
@@ -5,7 +5,14 @@
import sys
import os
-from site import getsitepackages, getusersitepackages, check_enableusersite
+from site import check_enableusersite
+
+# Handle both normal environments and virtualenvs
+from distutils.sysconfig import get_python_lib as getsitepackages
+try:
+ from site import getusersitepackages
+except ImportError:
+ getusersitepackages = getsitepackages
# Hacky command line parsing to accept a silent-install -s flag like linux-setup.sh:
INTERACTIVE = True