Re-configure to cancel futures when an exception including KeyboardInterrupt is encountered.

Author: jaraco

distributions/process.py

@@ -15,7 +15,10 @@
 @main
 def run():
     res = pypi.Distribution.unprocessed()
-    with concurrent.futures.ThreadPoolExecutor(max_workers=os.cpu_count()) as executor:
-        futures = (executor.submit(dist.process) for dist in res.dists)
-        completes = concurrent.futures.as_completed(futures)
+    executor = concurrent.futures.ThreadPoolExecutor(max_workers=os.cpu_count())
+    futures = (executor.submit(dist.process) for dist in res.dists)
+    completes = concurrent.futures.as_completed(futures)
+    try:
         consume(tqdm.tqdm(completes, total=res.count, smoothing=0.1))
+    finally:
+        executor.shutdown(wait=True, cancel_futures=True)