Fix setup for pip 10.0.0

Issue-ID: DCAEGEN2-452
Change-Id: I22c2f3234408a588ebf664d53eae4737474cd800
Signed-off-by: Lusheng Ji <lji@research.att.com>

1 files changed, 10 insertions, 2 deletions

diff --git a/setup.py b/setup.py
index 57baf52..e5d3ceb 100644
--- a/setup.py
+++ b/setup.py
@@ -18,8 +18,16 @@
 
 import os
 from setuptools import setup, find_packages
-from pip.req import parse_requirements
-from pip.download import PipSession
+#from pip.req import parse_requirements
+try: # for pip >= 10
+    from pip._internal.req import parse_requirements
+except ImportError: # for pip <= 9.0.3
+    from pip.req import parse_requirements
+#from pip.download import PipSession
+try: # for pip >= 10
+    from pip._internal.download import PipSession
+except ImportError: # for pip <= 9.0.3
+    from pip.download import PipSession
 
 setup(
     name='miss_htbt_service',