ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix urllib3 sends deprecation warning when setting ssl_version

Browse Source
This commit is contained in:
allegroai 2024-03-06 09:11:26 +02:00
parent a1f3279719
commit 26ed3a9b21
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
clearml/backend_api/utils.py
View File

@ -6,6 +6,7 @@ import requests
from requests.adapters import HTTPAdapter from requests.adapters import HTTPAdapter
from urllib3.util import Retry from urllib3.util import Retry
from urllib3 import PoolManager from urllib3 import PoolManager
import urllib3
import six import six
from .session.defs import ENV_HOST_VERIFY_CERT from .session.defs import ENV_HOST_VERIFY_CERT
@ -58,10 +59,21 @@ def urllib_log_warning_setup(total_retries=10, display_warning_after=5):
class TLSv1HTTPAdapter(HTTPAdapter): class TLSv1HTTPAdapter(HTTPAdapter):
def init_poolmanager(self, connections, maxsize, block=False, **pool_kwargs): def init_poolmanager(self, connections, maxsize, block=False, **pool_kwargs):
self.poolmanager = PoolManager(num_pools=connections, # noinspection PyBroadException
maxsize=maxsize, try:
block=block, if "ssl_minimum_version" in urllib3.poolmanager.SSL_KEYWORDS:
ssl_version=ssl.PROTOCOL_TLSv1_2) self.poolmanager = PoolManager(
num_pools=connections, maxsize=maxsize, block=block, ssl_minimum_version=ssl.TLSVersion.TLSv1_2
)
else:
self.poolmanager = PoolManager(
num_pools=connections, maxsize=maxsize, block=block, ssl_version=ssl.PROTOCOL_TLSv1_2
)
except AttributeError:
# just in case some attributes were not found in urrlib3 older versions
self.poolmanager = PoolManager(
num_pools=connections, maxsize=maxsize, block=block, ssl_version=ssl.PROTOCOL_TLSv1_2
)
class SessionWithTimeout(requests.Session): class SessionWithTimeout(requests.Session):