Make backwards compatible with urllib3~=1.0 (#197)

Signed-off-by: Jesse Whitehouse <[email protected]>

Author: Jesse

pyproject.toml

@@ -29,7 +29,7 @@ numpy = [
 sqlalchemy = "^1.3.24"
 openpyxl = "^3.0.10"
 alembic = "^1.0.11"
-urllib3 = "^2.0.0"
+urllib3 = ">=1.0"
 
 [tool.poetry.dev-dependencies]
 pytest = "^7.1.2"

src/databricks/sql/auth/retry.py

@@ -4,7 +4,13 @@
 from enum import Enum
 from typing import List, Optional, Tuple, Union
 
-from urllib3 import BaseHTTPResponse  # type: ignore
+# We only use this import for type hinting
+try:
+    # If urllib3~=2.0 is installed
+    from urllib3 import BaseHTTPResponse  # type: ignore
+except ImportError:
+    # If urllib3~=1.0 is installed
+    from urllib3 import HTTPResponse as BaseHTTPResponse
 from urllib3 import Retry
 from urllib3.util.retry import RequestHistory