From 21d9660e649e8784917ac90e32232151ef47b82a Mon Sep 17 00:00:00 2001
From: bretello <bretello@distruzione.org>
Date: Wed, 30 Mar 2022 15:53:03 +0200
Subject: [PATCH] zsh: improve mkvenv function

---
 functions/misc.zsh | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/functions/misc.zsh b/functions/misc.zsh
index 8aa9738..80839b6 100755
--- a/functions/misc.zsh
+++ b/functions/misc.zsh
@@ -354,12 +354,15 @@ make_backup() {
 }
 
 mkvenv(){
-    python -m venv .venv && echo "created venv: .venv"
-
-    read -q source_venv
-
-    if [[ $source_venv == "y" ]]; then
-        source .venv/bin/activate && echo "sourced venv"
+    # any arguments are passed on to the `venv` module as flags/arguments
+    if [[ -e .venv ]]; then
+        echo "$(color $RED Error:) $(color $BOLD .venv) already exists." >&2
+        return 1
     fi
+    echo -n "Creating venv..." && python -m venv $@ .venv && echo -n " done. Enable? [Y/n]"
+    read source_venv
 
+    if [[ $source_venv != "n" ]]; then
+        source .venv/bin/activate && echo -e "  → $(color $BOLD Enabled!) 🐍 $(color $BOLD$PURPLE $(python --version | cut -d " " -f2 )) ($(color $BOLD$GREEN $(pip --version | cut -d " " -f -2)))"
+    fi
 }