[Setup] Add a CleanDocs class

1 files changed, 21 insertions, 0 deletions

diff --git a/setup.py b/setup.py
index 4ba76264f..a98b2cb29 100755
--- a/setup.py
+++ b/setup.py
@@ -19,6 +19,7 @@ from distutils import cmd
 from distutils.command.build import build as _build
 from distutils.command.clean import clean as _clean
 from distutils.command.install_data import install_data as _install_data
+from shutil import rmtree
 
 from setuptools import find_packages, setup
 from setuptools.command.test import test as _test
@@ -70,6 +71,25 @@ class BuildDocs(BuildDoc):
         BuildDoc.run(self)
 
 
+class CleanDocs(cmd.Command):
+    description = 'Clean the documentation build and rst files'
+    user_options = []
+
+    def initialize_options(self):
+        pass
+
+    def finalize_options(self):
+        pass
+
+    def run(self):
+        for docs_dir in ('docs/build', 'docs/source/modules'):
+            try:
+                print('Deleting {}'.format(docs_dir))
+                rmtree(docs_dir)
+            except OSError:
+                pass
+
+
 class BuildWebUI(cmd.Command):
     description = 'Minify WebUI files'
 
@@ -324,6 +344,7 @@ cmdclass = {
     'build_docs': BuildDocs,
     'install_data': InstallData,
     'clean_plugins': CleanPlugins,
+    'clean_docs': CleanDocs,
     'clean': Clean,
     'egg_info_plugins': EggInfoPlugins,
     'test': PyTest,