py 2.7 import compat

avanbrunt-cb · Oct 8, 2019 · 78ffca6 · 78ffca6
1 parent cf5aee7
commit 78ffca6
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cabby/__init__.py b/cabby/__init__.py
@@ -1,14 +1,18 @@
 '''
     Cabby, python library for interacting with TAXII servers.
 '''
-from urllib.parse import urlparse
-
 from ._version import __version__  # noqa: used in setup.py
 
 from .client10 import Client10
 from .client11 import Client11
 
 
+try:
+    from urllib.parse import urlparse
+except ImportError:     # Python 2.7
+    from urlparse import urlparse
+
+
 def create_client(host=None, port=None, discovery_path=None, use_https=False,
                   url=None, version="1.1", headers=None):
     '''Create a client instance (TAXII version specific).