Added symlink command. Added --as option.

2026-06-05 23:30:18 +00:00 · 2011-03-23 01:14:39 +09:00
parent 2f3fbeca53
commit eb49f121db
6 changed files with 65 additions and 10 deletions
@@ -0,0 +1,44 @@
+import os
+from pythonbrew.basecommand import Command
+from pythonbrew.define import PATH_PYTHONS, PATH_BIN
+from pythonbrew.util import Package, symlink, unlink
+
+class SymlinkCommand(Command):
+    name = "symlink"
+    usage = "%prog"
+    summary = "Create/Remove a symbolic link to python"
+    
+    def __init__(self):
+        super(SymlinkCommand, self).__init__()
+        self.parser.add_option(
+            "-p", "--python",
+            dest="pythons",
+            action="append",
+            default=[],
+            help="Using specified python versions."
+        )
+        self.parser.add_option(
+            "-r", "--remove",
+            dest="remove",
+            action="store_true",
+            default=False,
+            help="Remove a symbolic link."
+        )
+    
+    def run_command(self, options, args):
+        pythons = self._get_pythons(options.pythons)
+        for python in pythons:
+            version = Package(python).version
+            src = os.path.join(PATH_PYTHONS, python, 'bin', 'python')
+            dst = os.path.join(PATH_BIN, 'py%s' % (version))
+            if options.remove:
+                unlink(dst)
+            else:
+                symlink(src, dst)
+    
+    def _get_pythons(self, _pythons):
+        pythons = [Package(p).name for p in _pythons]
+        return [d for d in sorted(os.listdir(PATH_PYTHONS)) 
+                if not pythons or d in pythons]
+
+SymlinkCommand()