Add pew-style project management

poetry/console/application.py

@@ -33,6 +33,7 @@
 
 from .config import ApplicationConfig
 
+from .commands.alias import AliasCommand
 from .commands.env import EnvCommand
 
 
@@ -91,6 +92,9 @@ def get_default_commands(self):  # type: () -> list
         # Debug command
         commands += [DebugCommand()]
 
+        # Alias command
+        commands += [AliasCommand()]
+
         # Env command
         commands += [EnvCommand()]
 

poetry/console/commands/alias/__init__.py

@@ -0,0 +1 @@
+from .alias import AliasCommand

poetry/console/commands/alias/alias.py

@@ -0,0 +1,30 @@
+from ..command import Command
+
+from .clean import AliasCleanCommand
+from .go import AliasGoCommand
+from .list import AliasListCommand
+from .prune import AliasPruneCommand
+from .rm import AliasRmCommand
+from .set import AliasSetCommand
+from .show import AliasShowCommand
+
+
+class AliasCommand(Command):
+    """
+    Work with Poetry's project aliases.
+
+    alias
+    """
+
+    commands = [
+        AliasCleanCommand(),
+        AliasGoCommand(),
+        AliasListCommand(),
+        AliasPruneCommand(),
+        AliasRmCommand(),
+        AliasSetCommand(),
+        AliasShowCommand(),
+    ]
+
+    def handle(self):  # type: () -> int
+        return self.call("help", self._config.name)

poetry/console/commands/alias/clean.py

@@ -0,0 +1,23 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasCleanCommand(Command):
+    """
+    Remove all alias definitions that no longer point to a project directory.
+
+    clean
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager()
+        removed_aliases = manager.clean()
+
+        if not removed_aliases:
+            self.line("No aliases defined.")
+            return 0
+
+        self.line("<b>Removed Aliases</b>")
+        for name in sorted(removed_aliases):
+            self.line("<info>{}</info>: {}".format(name, removed_aliases[name]))

poetry/console/commands/alias/go.py

@@ -0,0 +1,40 @@
+import sys
+
+from os import chdir, environ
+from distutils.util import strtobool
+
+from poetry.utils._compat import Path
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasGoCommand(Command):
+    """
+    Activate a project's virtualenv and change to its source directory.
+
+    go
+        {name : The alias of the project to activate.}
+    """
+
+    def handle(self):  # type: () -> None
+        # Don't do anything if we are already inside a virtualenv
+        active_venv = environ.get("VIRTUAL_ENV", None)
+
+        if active_venv is not None:
+            self.line(
+                "Virtual environment already activated: "
+                "<info>{}</>".format(active_venv)
+            )
+            return
+
+        # Determine the project path for the given alias
+        manager = AliasManager()
+        name = self.argument("name")
+        project_path = manager.get_project(name)
+        project_dirname = str(project_path)
+
+        # Change directory and activate the virtualenv
+        chdir(project_dirname)
+        environ["PWD"] = project_dirname
+        self.call("shell")

poetry/console/commands/alias/list.py

@@ -0,0 +1,18 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasListCommand(Command):
+    """
+    List all defined project aliases.
+
+    list
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager()
+        aliases = manager.list()
+
+        for name in sorted(aliases):
+            self.line("<info>{}</info>: {}".format(name, aliases[name]))

poetry/console/commands/alias/prune.py

@@ -0,0 +1,23 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasPruneCommand(Command):
+    """
+    Remove all alias definitions that no longer point to a project directory.
+
+    prune
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager()
+        removed_aliases = manager.prune()
+
+        if not removed_aliases:
+            self.line("No dangling aliases found.")
+            return 0
+
+        self.line("<b>Removed Aliases</b>")
+        for name in sorted(removed_aliases):
+            self.line("<info>{}</info>: {}".format(name, removed_aliases[name]))

poetry/console/commands/alias/rm.py

@@ -0,0 +1,23 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasRmCommand(Command):
+    """
+    Remove the alias for the current project.
+
+    rm
+        {alias? : The alias to remove.  If omitted, remove the current project's alias.}
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager()
+        alias = self.argument("alias")
+
+        if alias:
+            project_path = manager.get_project(alias)
+        else:
+            project_path = self.poetry.file.parent
+
+        manager.remove(project_path)

poetry/console/commands/alias/set.py

@@ -0,0 +1,18 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasSetCommand(Command):
+    """
+    Set an alias for the current project.
+
+    set
+        {name : The alias to set for the current project.}
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager(self.poetry.config)
+        project_path = self.poetry.file.parent
+        name = self.argument("name")
+        manager.set(project_path, name)

poetry/console/commands/alias/show.py

@@ -0,0 +1,17 @@
+from poetry.utils.alias import AliasManager
+
+from ..command import Command
+
+
+class AliasShowCommand(Command):
+    """
+    Show the alias for the current project.
+
+    show
+    """
+
+    def handle(self):  # type: () -> None
+        manager = AliasManager(self.poetry.config)
+        project_path = self.poetry.file.parent
+        name = manager.get_alias(project_path)
+        self.line(name)

poetry/console/commands/init.py

@@ -8,6 +8,7 @@
 
 from .command import Command
 from .env_command import EnvCommand
+from ...utils.alias import AliasCommandError, AliasManager
 
 
 class InitCommand(Command):
@@ -23,6 +24,7 @@ class InitCommand(Command):
         {--dev-dependency=* : Package to require for development with an optional version constraint,
                               e.g. requests:^2.10.0 or requests=2.11.1}
         {--l|license= : License of the package}
+        {--alias= : Project alias}
     """
 
     help = """\
@@ -33,6 +35,7 @@ def __init__(self):
         super(InitCommand, self).__init__()
 
         self._pool = None
+        self._alias_manager = AliasManager()
 
     def handle(self):
         from poetry.layouts import layout
@@ -112,6 +115,15 @@ def handle(self):
         )
         python = self.ask(question)
 
+        alias = self.option("alias")
+        if not alias:
+            question = self.create_question(
+                "Project alias [<comment>{}</comment>, . to skip]: ".format(name),
+                default=name,
+            )
+            question.set_validator(self._validate_alias)
+            alias = self.ask(question)
+
         self.line("")
 
         requirements = {}
@@ -163,6 +175,9 @@ def handle(self):
         with (Path.cwd() / "pyproject.toml").open("w") as f:
             f.write(content)
 
+        if alias:
+            self._alias_manager.set(Path.cwd(), alias)
+
     def _determine_requirements(
         self, requires, allow_prereleases=False  # type: List[str]  # type: bool
     ):  # type: (...) -> List[str]
@@ -318,6 +333,20 @@ def _validate_license(self, license):
 
         return license
 
+    def _validate_alias(self, alias):
+        if alias == ".":
+            return
+
+        try:
+            existing_project = self._alias_manager.get_project(alias)
+        except AliasCommandError:
+            existing_project = None
+
+        if existing_project is not None:
+            raise ValueError("Alias already exists")
+
+        return alias
+
     def _get_pool(self):
         from poetry.repositories import Pool
         from poetry.repositories.pypi_repository import PyPiRepository