fix: use distutils.core on error

Jose Diaz-Gonzalez · Jose Diaz-Gonzalez · commit 794ccf399646 · 2021-01-01T21:30:54.000-05:00
diff --git a/setup.py b/setup.py
@@ -1,10 +1,24 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 import os
-from setuptools import setup
-
 from github_backup import __version__
 
+try:
+    from setuptools import setup
+    setup  # workaround for pyflakes issue #13
+except ImportError:
+    from distutils.core import setup
+
+# Hack to prevent stupid TypeError: 'NoneType' object is not callable error on
+# exit of python setup.py test # in multiprocessing/util.py _exit_function when
+# running python setup.py test (see
+# http://www.eby-sarna.com/pipermail/peak/2010-May/003357.html)
+try:
+    import multiprocessing
+    multiprocessing
+except ImportError:
+    pass
+
 
 def open_file(fname):
     return open(os.path.join(os.path.dirname(__file__), fname))
