add cleanup of temporary directory

asmeurer · Jul 9, 2013 · 56f383a · 56f383a
1 parent fce7484
commit 56f383a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/conda/cli/main_build.py b/conda/cli/main_build.py
@@ -32,6 +32,7 @@ def configure_parser(sub_parsers):
 
 def execute(args, parser):
     import sys
+    import shutil
     import tarfile
     import tempfile
     from os.path import abspath, isdir, isfile
@@ -46,8 +47,11 @@ def execute(args, parser):
             t = tarfile.open(arg, 'r:*')
             t.extractall(path=recipe_dir)
             t.close()
+            need_cleanup = True
         else:
             recipe_dir = abspath(arg)
+            need_cleanup = False
+
         if not isdir(recipe_dir):
             sys.exit("Error: no such directory: %s" % recipe_dir)
 
@@ -60,6 +64,9 @@ def execute(args, parser):
         else:
             build.build(m)
 
+        if need_cleanup:
+            shutil.rmtree(recipe_dir)
+
         print """\
 # If you want to upload this package to binstar.org, type:
 #