NixJupyter/shell.nix
2024-02-09 21:54:53 -05:00

73 lines
1.6 KiB
Nix

with import <nixpkgs> { };
pkgs.mkShell rec {
venvDir = "./.venv";
buildInputs = [
# Install Python & Packages
(python3.withPackages (ps: with python3Packages; [
jupyter
ipython
ipykernel
setuptools
virtualenv
pip
pyqt5
pandas
numpy
matplotlib
scipy
]))
# Install Go (GoNB)
go
];
shellHook = ''
# Export Environment Paths
export GOBIN=$(pwd)/.bin
export PATH=$PATH:$(pwd)/.bin
export JUPYTER_DATA_DIR=$(pwd)/.data
# Upsert Notebook Directory
mkdir -p ./notebooks
# Install GoNB
go install github.com/janpfeifer/gonb@latest
go install golang.org/x/tools/cmd/goimports@latest
go install golang.org/x/tools/gopls@latest
# # fixes libstdc++ issues and libgl.so issues
# LD_LIBRARY_PATH=${stdenv.cc.cc.lib}/lib/:/run/opengl-driver/lib/
#
# # Fix XCB Issues
# QT_PLUGIN_PATH=${qt5.qtbase}/${qt5.qtbase.qtPluginPrefix}
# SOURCE_DATE_EPOCH=$(date +%s)
# QT_XCB_GL_INTEGRATION="none"
# Upsert Virtual Environment
if [ -d "${venvDir}" ]; then
echo "Skipping venv creation, '${venvDir}' already exists"
else
echo "Creating new venv environment in path: '${venvDir}'"
${python3.interpreter} -m venv "${venvDir}"
fi
# Update Python Path
PYTHONPATH=$PWD/${venvDir}/${python3.sitePackages}/:$PYTHONPATH
# Activate Virtual Environment
source "${venvDir}/bin/activate"
# Update PIP
python -m pip install --upgrade pip
python -m ipykernel install --user --name=${venvDir}
# Install GoNB
gonb --install
'';
}