feat: allow overriding where base pip packages get installed from whe…

…n an index is specified
plotly · Dec 12, 2018 · e74bbdb · e74bbdb
1 parent e4307fc
commit e74bbdb
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/bin/steps/python b/bin/steps/python
@@ -107,8 +107,16 @@ if [ "$FRESH_PYTHON" ] || [[ ! $(pip --version) == *$PIP_UPDATE* ]]; then
   rm -fr /app/.heroku/python/lib/python*/site-packages/pip-*
   rm -fr /app/.heroku/python/lib/python*/site-packages/setuptools-*
 
-  /app/.heroku/python/bin/python "$ROOT_DIR/vendor/get-pip.py" pip=="$PIP_UPDATE" &> /dev/null
-  /app/.heroku/python/bin/pip install "$ROOT_DIR/vendor/setuptools-39.0.1-py2.py3-none-any.whl" &> /dev/null
+  # This will allow users to override where pip gets installed from
+  # which is useful if CI for heroku applications is run in non-heroku environments
+  # where access to pypi is strictly controlled
+  if [[ -n "$GET_PIP_PYPI_INDEX_URL" ]]; then
+    /app/.heroku/python/bin/python "$ROOT_DIR/vendor/get-pip.py" --no-index "$GET_PIP_PYPI_INDEX_URL/pip/pip-9.0.2-py2.py3-none-any.whl" "$GET_PIP_PYPI_INDEX_URL/setuptools/setuptools-39.0.1-py2.py3-none-any.whl" "$GET_PIP_PYPI_INDEX_URL/wheel/wheel-0.31.1-py2.py3-none-any.whl" &> /dev/null
+    /app/.heroku/python/bin/pip install --no-index "$GET_PIP_PYPI_INDEX_URL/setuptools/setuptools-39.0.1-py2.py3-none-any.whl" &> /dev/null
+  else
+    /app/.heroku/python/bin/python "$ROOT_DIR/vendor/get-pip.py" pip=="$PIP_UPDATE" &> /dev/null
+    /app/.heroku/python/bin/pip install "$ROOT_DIR/vendor/setuptools-39.0.1-py2.py3-none-any.whl" &> /dev/null
+  fi
 
 fi