From 4afbecf78c93ecd868bfb9a894bc7b54654a6e3f Mon Sep 17 00:00:00 2001
From: Evie Litherland-Smith <evie@xenia.me.uk>
Date: Mon, 22 May 2023 09:54:50 +0100
Subject: [PATCH] Add directory name to venv prompt

---
 templates/python39/shell.nix | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/templates/python39/shell.nix b/templates/python39/shell.nix
index a80de335..5b62f0fd 100644
--- a/templates/python39/shell.nix
+++ b/templates/python39/shell.nix
@@ -2,7 +2,8 @@
 pkgs.mkShell {
   nativeBuildInputs = with pkgs; [python39 poetry git];
   shellHook = ''
-    [ -e .venv/bin/python3 ] && VENV_FLAG="--upgrade" || VENV_FLAG="--clear"; python3 -m venv $VENV_FLAG .venv
+    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
   '';
 }