From f202b3762caca345e6a010db5bfac724d3969a9e Mon Sep 17 00:00:00 2001
From: Julien Palard <julien@palard.fr>
Date: Sat, 29 Apr 2017 11:15:28 +0200
Subject: [PATCH] Add 'workon .' to work on current directory.

---
 .bashrc | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/.bashrc b/.bashrc
index 10830f3..2cfa2f0 100644
--- a/.bashrc
+++ b/.bashrc
@@ -250,9 +250,15 @@ fi
 function workon
 {
     local VENVS="$HOME/.venvs"
-    [ -d "$VENVS/$1" ] || mkdir -p "$VENVS/$1"
-    [ -f "$VENVS/$1/bin/activate" ] || python3 -m venv "$VENVS/$1"
-    . "$VENVS/$1/bin/activate"
+    if [ -z "$1" -o z"$1" = z"." ]
+    then
+        local VENV_NAME="$(basename "$PWD")"
+    else
+        local VENV_NAME="$1"
+    fi
+    [ -d "$VENVS/$VENV_NAME" ] || mkdir -p "$VENVS/$VENV_NAME"
+    [ -f "$VENVS/$VENV_NAME/bin/activate" ] || python3 -m venv "$VENVS/$VENV_NAME"
+    . "$VENVS/$VENV_NAME/bin/activate"
 }
 
 function _workon