If multiprocessing fails, don't use it during setup.

mdboom · mdboom · commit bc2b4cfec7ed · 2013-06-03T09:09:31.000-04:00
diff --git a/setup.py b/setup.py
@@ -299,13 +299,27 @@ def run_2to3(self, files):
             filtered = [x for x in files if should_2to3(x, self.build_lib)]
             if sys.platform.startswith('win') or 'TRAVIS' in os.environ:
                 # doing this in parallel on windows may crash your computer
-                [refactor(f) for f in filtered]
+                self.run_2to3_uniprocess(filtered)
             else:
+                self.run_2to3_multiprocess(filtered)
+            print()
+
+        def run_2to3_uniprocess(self, files):
+            for i, f in enumerate(files):
+                print("Running 2to3... %.02f%%" %
+                        (float(i) / len(files) * 100.0), end='\r')
+                refactor(f)
+
+        def run_2to3_multiprocess(self, files):
+            try:
                 p = multiprocessing.Pool()
-                for i, x in enumerate(p.imap_unordered(refactor, filtered)):
+            except:
+                self.run_2to3_uniprocess(files)
+            else:
+                for i, x in enumerate(p.imap_unordered(refactor, files)):
                     print("Running 2to3... %.02f%%" %
-                          (float(i) / len(filtered) * 100.0), end='\r')
-            print()
+                        (float(i) / len(files) * 100.0), end='\r')
+
 
 print_raw("pymods %s" % py_modules)
 print_raw("packages %s" % packages)
