Default prompt of environment name in batch shell

pypa · Feb 27, 2020 · 2b24f15 · 2b24f15
1 parent 91c80d6
commit 2b24f15
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/docs/changelog/1679.bugfix.rst b/docs/changelog/1679.bugfix.rst
@@ -0,0 +1 @@
+    fix windows batch (cmd.exe) shell prompt to display environment name by default, simliar to bash behaviour.
diff --git a/src/virtualenv/activation/batch/__init__.py b/src/virtualenv/activation/batch/__init__.py
@@ -21,3 +21,9 @@ def instantiate_template(self, replacements, template, creator):
         # ensure the text has all newlines as \r\n - required by batch
         base = super(BatchActivator, self).instantiate_template(replacements, template, creator)
         return base.replace(os.linesep, "\n").replace("\n", os.linesep)
+
+    def replacements(self, creator, dest_folder):
+        base_replacements = super(BatchActivator, self).replacements(creator, dest_folder)
+        if not base_replacements["__VIRTUAL_PROMPT__"]:
+            base_replacements["__VIRTUAL_PROMPT__"] = "(%s) " % creator.env_name
+        return base_replacements