Enforce use of a virtual environment.

Author: Gadgetoid

install.sh

@@ -10,9 +10,17 @@ USAGE="./install.sh (--unstable)"
 POSITIONAL_ARGS=()
 FORCE=false
 UNSTABLE=false
-PYTHON="/usr/bin/python3"
+PYTHON="python"
 
 
+venv_check() {
+	PYTHON_BIN=`which $PYTHON`
+	if [[ $VIRTUAL_ENV == "" ]] || [[ $PYTHON_BIN != $VIRTUAL_ENV* ]]; then
+		printf "This script should be run in a virtual Python environment.\n"
+		exit 1
+	fi	       
+}
+
 user_check() {
 	if [ $(id -u) -eq 0 ]; then
 		printf "Script should not be run as root. Try './install.sh'\n"
@@ -126,6 +134,7 @@ while [[ $# -gt 0 ]]; do
 done
 
 user_check
+venv_check
 
 if [ ! -f "$PYTHON" ]; then
 	printf "Python path $PYTHON not found!\n"
@@ -179,6 +188,7 @@ printf "It's recommended you run these steps manually.\n"
 printf "If you want to run the full script, open it in\n"
 printf "an editor and remove 'exit 1' from below.\n"
 exit 1
+source $VIRTUAL_ENV/bin/activate
 EOF
 
 if $UNSTABLE; then

uninstall.sh

@@ -3,7 +3,16 @@
 FORCE=false
 LIBRARY_NAME=`grep -m 1 name pyproject.toml | awk -F" = " '{print substr($2,2,length($2)-2)}'`
 RESOURCES_DIR=$HOME/Pimoroni/$LIBRARY_NAME
-PYTHON="/usr/bin/python3"
+PYTHON="python"
+
+
+venv_check() {
+	PYTHON_BIN=`which $PYTHON`
+	if [[ $VIRTUAL_ENV == "" ]] || [[ $PYTHON_BIN != $VIRTUAL_ENV* ]]; then
+		printf "This script should be run in a virtual Python environment.\n"
+		exit 1
+	fi	       
+}
 
 user_check() {
 	if [ $(id -u) -eq 0 ]; then
@@ -49,6 +58,7 @@ warning() {
 printf "$LIBRARY_NAME Python Library: Uninstaller\n\n"
 
 user_check
+venv_check
 
 printf "Uninstalling for Python 3...\n"
 $PYTHON -m pip uninstall $LIBRARY_NAME