diff --git a/lib/cpy_distutils.py b/lib/cpy_distutils.py index e1dc75bb..d4b6885a 100644 --- a/lib/cpy_distutils.py +++ b/lib/cpy_distutils.py @@ -24,9 +24,18 @@ """Implements the DistUtils command 'build_ext' """ -from distutils.command.build_ext import build_ext -from distutils.command.install import install -from distutils.command.install_lib import install_lib +# Setuptools is the default choice for package management +# and pip uses it to build it. Moreover bare +# distutils doesn't implement uninstalling. +try: + from setuptools.command.build_ext import build_ext + from setuptools.command.install import install + from setuptools.command.install_lib import install_lib +except ImportError: + from distutils.command.build_ext import build_ext + from distutils.command.install import install + from distutils.command.install_lib import install_lib + from distutils.errors import DistutilsExecError from distutils.util import get_platform from distutils.dir_util import copy_tree