Update Dockerfile

Dockerfile

@@ -53,29 +53,47 @@ RUN echo 'export PS1="\u@kanha:\w\$ "' >> /root/.bashrc && \
     echo 'cd /data' >> /root/.bashrc && \
     echo 'cd /data' >> /root/.bash_profile
 
-# ── Intercept apt and apt-get → save packages to apt.txt ─────────────────────
-# Moves real binaries to /usr/bin/apt.real and /usr/bin/apt-get.real
-# Our wrappers sit in /usr/local/bin which is higher priority in PATH
+# ── Force pip/python to always use venv (works in ALL shell types) ────────────
+RUN printf '%s\n' \
+    '#!/bin/sh' \
+    'exec /data/persist/venv/bin/pip "$@"' \
+    > /usr/local/bin/pip && \
+    chmod +x /usr/local/bin/pip && \
+    \
+    printf '%s\n' \
+    '#!/bin/sh' \
+    'exec /data/persist/venv/bin/pip3 "$@"' \
+    > /usr/local/bin/pip3 && \
+    chmod +x /usr/local/bin/pip3 && \
+    \
+    printf '%s\n' \
+    '#!/bin/sh' \
+    'exec /data/persist/venv/bin/python "$@"' \
+    > /usr/local/bin/python && \
+    chmod +x /usr/local/bin/python && \
+    \
+    printf '%s\n' \
+    '#!/bin/sh' \
+    'exec /data/persist/venv/bin/python3 "$@"' \
+    > /usr/local/bin/python3 && \
+    chmod +x /usr/local/bin/python3
+
+# ── apt/apt-get wrapper → intercepts install and saves to apt.txt ─────────────
 RUN printf '%s\n' \
     '#!/bin/sh' \
-    '# Wrapper for apt and apt-get — intercepts install and saves to apt.txt' \
     'APT_FILE="/data/apt.txt"' \
     'REAL_BIN="$1"' \
     'shift' \
     'if [ "$1" = "install" ]; then' \
     '    shift' \
-    '    # Run the real apt/apt-get install' \
     '    "$REAL_BIN" install "$@"' \
     '    STATUS=$?' \
-    '    # Only save if install succeeded' \
     '    if [ $STATUS -eq 0 ]; then' \
     '        touch "$APT_FILE"' \
     '        for arg in "$@"; do' \
-    '            # Skip flags like -y --no-install-recommends' \
     '            case "$arg" in' \
     '                -*) continue ;;' \
     '            esac' \
-    '            # Skip if already in apt.txt' \
     '            if ! grep -qx "$arg" "$APT_FILE"; then' \
     '                echo "$arg" >> "$APT_FILE"' \
     '                echo "✓ $arg saved to apt.txt"' \
@@ -84,19 +102,16 @@ RUN printf '%s\n' \
     '    fi' \
     '    exit $STATUS' \
     'fi' \
-    '# For all other apt commands (update, remove, etc) just pass through' \
     '"$REAL_BIN" "$@"' \
     > /usr/local/bin/_apt_wrapper && \
     chmod +x /usr/local/bin/_apt_wrapper
 
-# apt wrapper
 RUN printf '%s\n' \
     '#!/bin/sh' \
     'exec /usr/local/bin/_apt_wrapper /usr/bin/apt "$@"' \
     > /usr/local/bin/apt && \
     chmod +x /usr/local/bin/apt
 
-# apt-get wrapper
 RUN printf '%s\n' \
     '#!/bin/sh' \
     'exec /usr/local/bin/_apt_wrapper /usr/bin/apt-get "$@"' \
@@ -138,14 +153,6 @@ RUN printf '%s\n' \
     '' \
     'rm -f /data/persist/venv/lib/python*/EXTERNALLY-MANAGED' \
     '' \
-    '# Write real venv path to every shell automatically' \
-    'cat > /etc/profile.d/venv.sh << EOF' \
-    'export VIRTUAL_ENV=/data/persist/venv' \
-    'export PATH="/data/persist/venv/bin:$PATH"' \
-    'EOF' \
-    '' \
-    'grep -qx "source /etc/profile.d/venv.sh" /root/.bashrc || echo "source /etc/profile.d/venv.sh" >> /root/.bashrc' \
-    '' \
     'export VIRTUAL_ENV=/data/persist/venv' \
     'export PATH="/data/persist/venv/bin:$PATH"' \
     '' \