SERVER-76690 Special handling for PYTHONPATH when activate venvv6.3

(cherry picked from commit d32ee090853a65079d5c0b0b311309b918e3c361)

1 files changed, 12 insertions, 1 deletions

diff --git a/evergreen/prelude_venv.sh b/evergreen/prelude_venv.sh
index e7dd4b9d57b..a4e9318375e 100644
--- a/evergreen/prelude_venv.sh
+++ b/evergreen/prelude_venv.sh
@@ -28,7 +28,18 @@ function activate_venv {
     python_version=$($python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
     export PYTHONPATH="${workdir}/venv/lib/python${python_version}/site-packages:${PYTHONPATH}:${workdir}/src"
   else
-    export PYTHONPATH="$PYTHONPATH:${workdir}/src"
+    python_version=$($python -c 'import sys; print(f"{sys.version_info.major}.{sys.version_info.minor}")')
+    site_packages="${workdir}/venv/lib/python${python_version}/site-packages"
+    python -c "import sys; print(sys.path)"
+
+    # Check if site_packages is already in sys.path
+    in_sys_path=$($python -c "import sys; print('$site_packages' in sys.path)")
+    if [ "$in_sys_path" = "False" ]; then
+      export PYTHONPATH="${site_packages}:${PYTHONPATH}:${workdir}/src"
+    else
+      export PYTHONPATH="$PYTHONPATH:${workdir}/src"
+    fi
+    python -c "import sys; print(sys.path)"
   fi
 
   echo "python set to $(which $python)"