From fa5d6933fe9928b78b23da6730dbece3ae8d41e6 Mon Sep 17 00:00:00 2001 From: Brandl Date: Sun, 29 Mar 2020 23:43:40 +0200 Subject: [PATCH] Update setup.py --- setup.py | 18 +++++++++++------- 1 file changed, 11 insertions(+), 7 deletions(-) diff --git a/setup.py b/setup.py index 16fd185..43f22a5 100644 --- a/setup.py +++ b/setup.py @@ -1,14 +1,18 @@ """This module contains the packaging routine for the pybook package""" from setuptools import setup, find_packages -try: - from pip.download import PipSession - from pip.req import parse_requirements -except ImportError: - # It is quick hack to support pip 10 that has changed its internal - # structure of the modules. + +import pip +pip_major_version = int(pip.__version__.split(".")[0]) +if pip_major_version >= 20: + from pip._internal.req import parse_requirements + from pip._internal.network.session import PipSession +elif pip_major_version >= 10: + from pip._internal.req import parse_requirements from pip._internal.download import PipSession - from pip._internal.req.req_file import parse_requirements +else: + from pip.req import parse_requirements + from pip.download import PipSession def get_requirements(source):