nixos/templates/python39/shell.nix

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