build: support build/ being a symlink

I use a symlink to keep build artifacts off of my main drive, so
preserve the link if it exists.

Signed-off-by: Ben Boeckel <mathstuf@gmail.com>

1 files changed, 5 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 47f87660..a913274b 100755
--- a/setup.py
+++ b/setup.py
@@ -115,7 +115,11 @@ class CleanMore(clean):
         self.remove_pyc('obnamlib')
         self.remove_pyc('test-plugins')
         if os.path.isdir('build'):
-            shutil.rmtree('build')
+            if os.path.islink('build'):
+                for path in os.listdir('build'):
+                    shutil.rmtree('build/' + path)
+            else:
+                shutil.rmtree('build')
 
     def remove_pyc(self, rootdir):
         for dirname, subdirs, basenames in os.walk(rootdir):