update 0.5

pythonbrew/commands/switch.py

@@ -0,0 +1,41 @@
+import os
+import sys
+import re
+from pythonbrew.basecommand import Command
+from pythonbrew.define import PATH_PYTHONS, PATH_BIN
+from pythonbrew.util import off, symlink, Package
+from pythonbrew.log import logger
+
+class SwitchCommand(Command):
+    name = "switch"
+    usage = "%prog VERSION"
+    summary = "Switch to the given version"
+    
+    def run_command(self, options, args):
+        if not args:
+            logger.error("Unrecognized command line argument: argument not found.")
+            sys.exit(1)
+        pkg = Package(args[0])
+        pkgname = pkg.name
+        pkgdir = "%s/%s" % (PATH_PYTHONS, pkgname)
+        if not os.path.isdir(pkgdir):
+            logger.error("`%s` is not installed." % pkgname)
+            sys.exit(1)
+        self._switch_dir(pkgdir)
+        logger.info("Switched to %s" % pkgname)
+    
+    def _switch_dir(self, pkgdir):
+        off()
+        symlink(pkgdir, "%s/current" % PATH_PYTHONS)
+        for root, dirs, files in os.walk("%s/current/bin/" % PATH_PYTHONS):
+            for f in files:
+                symlink("%s%s" % (root, f), "%s/%s" % (PATH_BIN, f))
+            break
+        # I want better code
+        if not os.path.isfile("%s/python" % PATH_BIN):
+            if os.path.isfile("%s/python3" % PATH_BIN):
+                symlink(os.path.realpath("%s/python3" % PATH_BIN), "%s/python" % PATH_BIN)
+            elif os.path.isfile("%s/python3.0" % PATH_BIN):
+                symlink(os.path.realpath("%s/python3.0" % PATH_BIN), "%s/python" % PATH_BIN)
+
+SwitchCommand()