attempt to clean up after an exception. Fix #413.

Author: mihneadb

pip/basecommand.py

@@ -138,6 +138,14 @@ def main(self, args, initial_options):
             logger.fatal('Exception:\n%s' % format_exc())
             store_log = True
             exit = UNKNOWN_ERROR
+        finally:
+            # try to clean eventual files that got caught up
+            if not self.done_cleaning:
+                try:
+                    self.req_set.cleanup_files(bundle=self.bundle)
+                except:
+                    pass # go unnoticed if there is nothing to clean
+
         if log_fp is not None:
             log_fp.close()
         if store_log:

pip/commands/install.py

@@ -184,6 +184,7 @@ def _build_package_finder(self, options, index_urls):
 
     def run(self, options, args):
         fix_git_urls(args)
+        self.done_cleaning = False
         if options.download_dir:
             options.no_install = True
             options.ignore_installed = True
@@ -229,6 +230,8 @@ def run(self, options, args):
         for filename in options.requirements:
             for req in parse_requirements(filename, finder=finder, options=options):
                 requirement_set.add_requirement(req)
+        # save the requirement set for cleaning in case of error
+        self.req_set = requirement_set
         if not requirement_set.has_requirements and not options.requirements:
             opts = {'name': self.name}
             if options.find_links:
@@ -274,6 +277,7 @@ def run(self, options, args):
         # Clean up
         if not options.no_install or options.download_dir:
             requirement_set.cleanup_files(bundle=self.bundle)
+        self.done_cleaning = True
         if options.target_dir:
             if not os.path.exists(options.target_dir):
                 os.makedirs(options.target_dir)