fixed setup.py for pip 10

chompy · chompy · commit b43d60973de1 · 2018-04-26T19:24:47.000-04:00
diff --git a/setup.py b/setup.py
@@ -4,11 +4,18 @@
 
 from setuptools import setup, find_packages
 
+try: # for pip >= 10
+    from pip._internal.req import parse_requirements
+    from pip._internal.download import PipSession
+except ImportError: # for pip <= 9.0.3
+    from pip.req import parse_requirements
+    from pip.download import PipSession
+
 with open('README.rst') as readme_file:
     readme = readme_file.read()
 
-requirements = pip.req.parse_requirements(
-    "requirements.txt", session=pip.download.PipSession()
+requirements = parse_requirements(
+    "requirements.txt", session=PipSession()
 )
 pip_requirements = [str(r.req) for r in requirements]
 
