Remove all env

poetry/console/commands/env/remove.py

@@ -1,21 +1,42 @@
 from cleo import argument
+from cleo import option
 
 from ..command import Command
 
 
 class EnvRemoveCommand(Command):
 
     name = "remove"
-    description = "Removes a specific virtualenv associated with the project."
+    description = "Removes specific virtual environments associated with the project."
 
     arguments = [
-        argument("python", "The python executable to remove the virtualenv for.")
+        argument(
+            "python",
+            "The python executable(s) to remove the virtual environment for.",
+            optional=True,
+            multiple=True,
+        )
+    ]
+    options = [
+        option(
+            "all",
+            description="Remove all managed virtual environments associated with the project.",
+        ),
     ]
 
     def handle(self):
         from poetry.utils.env import EnvManager
 
-        manager = EnvManager(self.poetry)
-        venv = manager.remove(self.argument("python"))
+        pythons = self.argument("python")
+        all = self.option("all")
+        if not (pythons or all):
+            self.line("No virtualenv provided.")
 
-        self.line("Deleted virtualenv: <comment>{}</comment>".format(venv.path))
+        manager = EnvManager(self.poetry)
+        for python in pythons:
+            venv = manager.remove(python)
+            self.line("Deleted virtualenv: <comment>{}</comment>".format(venv.path))
+        if all:
+            for venv in manager.list():
+                manager.remove_venv(venv.path)
+                self.line("Deleted virtualenv: <comment>{}</comment>".format(venv.path))

tests/console/commands/env/test_remove.py

@@ -39,3 +39,34 @@ def test_remove_by_name(tester, venvs_in_cache_dirs, venv_name, venv_cache):
         expected += "Deleted virtualenv: {}\n".format((venv_cache / name))
 
     assert expected == tester.io.fetch_output()
+
+
+def test_remove_all(tester, venvs_in_cache_dirs, venv_name, venv_cache):
+    expected = {""}
+    tester.execute("--all")
+    for name in venvs_in_cache_dirs:
+        assert not (venv_cache / name).exists()
+        expected.add("Deleted virtualenv: {}".format((venv_cache / name)))
+    assert expected == set(tester.io.fetch_output().split("\n"))
+
+
+def test_remove_all_and_version(tester, venvs_in_cache_dirs, venv_name, venv_cache):
+    expected = {""}
+    tester.execute("--all {}".format(venvs_in_cache_dirs[0]))
+    for name in venvs_in_cache_dirs:
+        assert not (venv_cache / name).exists()
+        expected.add("Deleted virtualenv: {}".format((venv_cache / name)))
+    assert expected == set(tester.io.fetch_output().split("\n"))
+
+
+def test_remove_multiple(tester, venvs_in_cache_dirs, venv_name, venv_cache):
+    expected = {""}
+    removed_envs = venvs_in_cache_dirs[0:2]
+    remaining_envs = venvs_in_cache_dirs[2:]
+    tester.execute(" ".join(removed_envs))
+    for name in removed_envs:
+        assert not (venv_cache / name).exists()
+        expected.add("Deleted virtualenv: {}".format((venv_cache / name)))
+    for name in remaining_envs:
+        assert (venv_cache / name).exists()
+    assert expected == set(tester.io.fetch_output().split("\n"))