ZSH Python venv shell.nix

shell.nix

{ pkgs ? import <nixpkgs> {} }:

with pkgs;

mkShell {
  name = "python-env";

  buildInputs = [
    python3
  ];

  zshinithook = writeScript "zsh-init-hook.zsh" /* zsh */ ''
    export VIRTUAL_ENV_DISABLE_PROMPT=1
    python -m venv .venv
    source .venv/bin/activate

    exec 0</dev/tty
    exec 1>/dev/tty
    exec 2>/dev/tty

    setopt zle
  '';

  shellHook =  /* sh */ ''
    exec zsh \
      --interactive \
      --hist-ignore-space \
      --stdin &>/dev/null <<< " source $zshinithook"
  '';
}