diff --git a/setup.py b/setup.py index cbc059dc..36ca5fed 100644 --- a/setup.py +++ b/setup.py @@ -2,10 +2,16 @@ """The setup script.""" from setuptools import setup, find_packages +from pkg_resources import parse_version -try: # pip version >= 10.0 +try: from pip._internal.req import parse_requirements - from pip._internal.download import PipSession + from pip import __version__ as __pip_version + pip_version = parse_version(__pip_version) + if (pip_version >= parse_version("20")): + from pip._internal.network.session import PipSession + elif (pip_version >= parse_version("10")): + from pip._internal.download import PipSession except ImportError: # pip version < 10.0 from pip.req import parse_requirements from pip.download import PipSession @@ -15,7 +21,10 @@ # get the requirements from requirements.txt install_reqs = parse_requirements('requirements.txt', session=PipSession()) -reqs = [str(ir.req) for ir in install_reqs] +if pip_version >= parse_version("20"): + reqs = [str(ir.requirement) for ir in install_reqs] +else: + reqs = [str(ir.req) for ir in install_reqs] setup( name='neo-python',