Add support for --all.

Author: dave-shawley

HISTORY

@@ -5,6 +5,7 @@ Changelog
 
   - Support removal of virtual environment directories.
   - Support recursive removal of *__pycache__* directories.
+  - Add support for *--all*.
 
 * 0.0.2 (17-Nov-2014)
 

README.rst

@@ -78,6 +78,10 @@ few new command line parameters.
 ``setup.py clean --pycache``
    Recursively removes directories named *__pycache__*.
 
+``setup.py clean --all``
+   Remove all of by-products.  This is the same as using ``--dist --egg
+   --environment --pycache``.
+
 Where can I get this extension from?
 ------------------------------------
 +---------------+-----------------------------------------------------+

setupext/janitor.py

@@ -56,6 +56,10 @@ def finalize_options(self):
         if self.egg_base is None:
             self.egg_base = os.curdir
 
+        if self.all:
+            for flag in self.boolean_options:
+                setattr(self, flag, True)
+
         if self.environment and self.virtualenv_dir is None:
             self.virtualenv_dir = os.environ.get('VIRTUAL_ENV', None)
 

tests.py

@@ -84,6 +84,12 @@ def setUpClass(cls):
     def create_directory(cls, dir_name):
         return tempfile.mkdtemp(dir=cls.temp_dir, prefix=dir_name)
 
+    @staticmethod
+    def mkdirs(*dirs):
+        for d in dirs:
+            os.makedirs(d)
+        return dirs[:]
+
     def assert_path_does_not_exist(self, *trailing_segments):
         full_path = os.path.join(*trailing_segments)
         if os.path.exists(full_path):
@@ -228,11 +234,6 @@ def setUp(self):
         self.addCleanup(os.chdir, starting_dir)
         os.chdir(self.test_root)
 
-    def mkdirs(self, *dirs):
-        for d in dirs:
-            os.makedirs(d)
-        return dirs[:]
-
     def test_that_pycache_directories_are_removed(self):
         all_dirs = self.mkdirs(
             os.path.join(self.test_root, '__pycache__'),
@@ -255,3 +256,50 @@ def test_that_janitor_does_not_fail_when_cache_parent_is_removed(self):
         run_setup('clean', '--dist', '--pycache')
         for cache_dir in all_dirs:
             self.assert_path_does_not_exist(cache_dir)
+
+
+class RemoveAllTests(DirectoryCleanupMixin, unittest.TestCase):
+
+    @classmethod
+    def setUpClass(cls):
+        super(RemoveAllTests, cls).setUpClass()
+        test_root = cls.create_directory('test-root')
+        starting_dir = os.curdir
+        saved_env_dir = os.environ.get('VIRTUAL_ENV', None)
+
+        def restore():
+            os.chdir(starting_dir)
+            os.environ.pop('VIRTUAL_ENV', None)
+            if saved_env_dir is not None:
+                os.environ['VIRTUAL_ENV'] = saved_env_dir
+
+        os.chdir(test_root)
+        try:
+            all_dirs = cls.mkdirs(
+                '__pycache__',
+                'dist',
+                'foo.egg-info',
+                'env',
+            )
+            cls.pycache_dir = all_dirs[0]
+            cls.dist_dir = all_dirs[1]
+            cls.egg_dir = all_dirs[2]
+            cls.env_dir = all_dirs[3]
+            os.environ['VIRTUAL_ENV'] = cls.env_dir
+            run_setup('clean', '--all')
+            restore()
+        except:
+            restore()
+            raise
+
+    def test_that_pycache_is_removed(self):
+        self.assert_path_does_not_exist(self.pycache_dir)
+
+    def test_that_distdir_is_removed(self):
+        self.assert_path_does_not_exist(self.dist_dir)
+
+    def test_that_eggdir_is_removed(self):
+        self.assert_path_does_not_exist(self.egg_dir)
+
+    def test_that_envdir_is_removed(self):
+        self.assert_path_does_not_exist(self.env_dir)