From d3977da17097c8361815d1e3b6b79fd17a396dcd Mon Sep 17 00:00:00 2001
From: Henrik Lissner <henrik@lissner.net>
Date: Mon, 1 Sep 2014 11:31:24 -0400
Subject: [PATCH] Init pyenv after install

---
 rake/pkg/pyenv.rake | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/rake/pkg/pyenv.rake b/rake/pkg/pyenv.rake
index 44cb3a7d..4f61cd63 100644
--- a/rake/pkg/pyenv.rake
+++ b/rake/pkg/pyenv.rake
@@ -17,6 +17,9 @@ namespace :pyenv do
         "yyuu/pyenv-virtualenv",
         "yyuu/pyenv-which-ext"
       ].each { |pkg| github pkg, "#{PYENV_ROOT}/plugins/" }
+
+      sh_safe "eval $(~/.pyenv/bin/pyenv init -)"
+      sh_safe "eval $(~/.pyenv/bin/pyenv virtualenv-init -)"
     end
   end
 
@@ -37,11 +40,9 @@ namespace :pyenv do
     end
   end
 
-  desc "Remove pyenv & installed pythons"
-  task :remove do
-    if pyenv_installed?
-      echo "pyenv isn't installed!"
-    else
+  if pyenv_installed?
+    desc "Remove pyenv & installed pythons"
+    task :remove do
       echo "Deleting pyenv"
       sh_safe "rm -rf #{PYENV_ROOT}"
     end