Create a Jupyter Kernel for for the current virtualenv.

addvenvkernel.sh

#!/bin/bash

# This script creates a Jupyter kernel for a virtual environment.

# Get the name of the virtual environment from the command line or default to the current environment.
VENV_NAME="$1"

# Check if a virtual environment name was provided
if [ -z "$VENV_NAME" ]; then
    if [ -z "$VIRTUAL_ENV" ]; then
        echo "No virtual environment is currently activated, and none was specified."
        exit 1
    else
        VENV_NAME=$(basename "$VIRTUAL_ENV")
    fi
else
    # Check both the current directory and the .virtualenvs directory for the specified environment
    if [ -d "$VENV_NAME" ]; then
        VENV_PATH="$VENV_NAME"
    elif [ -d "$HOME/.virtualenvs/$VENV_NAME" ]; then
        VENV_PATH="$HOME/.virtualenvs/$VENV_NAME"
    else
        echo "Virtual environment '$VENV_NAME' not found."
        exit 1
    fi
    source "$VENV_PATH/bin/activate"
fi

# Install ipykernel if not already installed
if ! pip freeze | grep -q ipykernel; then
    pip install ipykernel
fi

# Create the Jupyter kernel
python -m ipykernel install --user --name="$VENV_NAME" --display-name="Python ($VENV_NAME)"

echo "Jupyter kernel for '$VENV_NAME' created successfully."