Import setuptools when wheel is available for wheel builds

1 files changed, 5 insertions, 1 deletions

diff --git a/setup.py b/setup.py
index 695ec85..9f38eeb 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,11 @@ VERSION = "7.43.0.1"
 
 import glob, os, re, sys, subprocess
 import distutils
-from distutils.core import setup
+try:
+    import wheel
+    from setuptools import setup
+except ImportError:
+    from distutils.core import setup
 from distutils.extension import Extension
 from distutils.util import split_quoted
 from distutils.version import LooseVersion