Bug fixed. Added tests.

pythonbrew/commands/py.py

@@ -4,7 +4,7 @@ from pythonbrew.basecommand import Command
 from pythonbrew.define import PATH_PYTHONS
 from pythonbrew.util import Package
 from pythonbrew.log import logger
-from subprocess import PIPE, Popen
+from subprocess import Popen
 
 class PyCommand(Command):
     name = "py"

pythonbrew/commands/switch.py

@@ -1,8 +1,8 @@
 import os
 import sys
 from pythonbrew.basecommand import Command
-from pythonbrew.define import PATH_PYTHONS, PATH_BIN, PATH_ETC_CURRENT
-from pythonbrew.util import Package
+from pythonbrew.define import PATH_PYTHONS, PATH_BIN
+from pythonbrew.util import Package, set_current_path
 from pythonbrew.log import logger
 
 class SwitchCommand(Command):
@@ -22,14 +22,8 @@ class SwitchCommand(Command):
             sys.exit(1)
         pkgbin = os.path.join(pkgdir,'bin')
         
-        self._set_current('%s:%s' % (PATH_BIN, pkgbin))
+        set_current_path('%s:%s' % (PATH_BIN, pkgbin))
         
         logger.info("Switched to %s" % pkgname)
-        
-    def _set_current(self, path):
-        fp = open(PATH_ETC_CURRENT, 'w')
-        fp.write('PATH_PYTHONBREW="%s"\n' % (path))
-        fp.close()
-
 
 SwitchCommand()

pythonbrew/commands/update.py

@@ -22,7 +22,7 @@ class UpdateCommand(Command):
         # check for latest version
         if version <= VERSION:
             logger.info("You are already running the installed latest version of pythonbrew.")
-            sys.exit()
+            return
         
         download_url = get_pythonbrew_update_url(version)
         if not download_url: