nixos/templates/python39/shell.nix

{pkgs ? import <nixpkgs> {}, ...}:
pkgs.mkShell {
  nativeBuildInputs = with pkgs; [python39 poetry git];
  shellHook = ''
    export PROJECT_NAME=$(basename $(readlink -f .))
    [ -e .venv/bin/python3 ] && VENV_FLAG="--upgrade" || VENV_FLAG="--clear --prompt $PROJECT_NAME"; python3 -m venv $VENV_FLAG .venv
    [ -e .venv/bin/activate ] && source .venv/bin/activate
  '';
}