2
2
3
3
from botocore import __version__ as botocore_version
4
4
from botocore import session
5
- from botocore .httpsession import URLLib3Session
6
5
from botocore .session import get_session # noqa: F401
7
6
8
7
import nifcloud
16
15
ssl_context .set_ciphers (f"{ default_cipher_names } :{ extra_ciphers } " )
17
16
18
17
19
- class CustomURLLib3Session (URLLib3Session ):
20
- def __init__ (self , * args , ** kwargs ):
21
- super ().__init__ (* args , ** kwargs )
22
- self ._manager .connection_pool_kw ['ssl_context' ] = ssl_context
23
-
24
-
25
18
class Session (session .Session ):
26
-
27
19
def __init__ (self , session_vars = None , event_hooks = None ,
28
20
include_builtin_handlers = True , profile = None ):
29
21
super (Session , self ).__init__ (session_vars , event_hooks ,
@@ -37,11 +29,22 @@ def create_client(self, service_name, region_name=None, api_version=None,
37
29
nifcloud_access_key_id = None , nifcloud_secret_access_key = None ,
38
30
nifcloud_session_token = None , config = None ):
39
31
client = super (Session , self ).create_client (
40
- service_name , region_name = region_name , api_version = api_version , use_ssl = use_ssl ,
41
- verify = verify , endpoint_url = endpoint_url , aws_access_key_id = nifcloud_access_key_id ,
42
- aws_secret_access_key = nifcloud_secret_access_key , aws_session_token = nifcloud_session_token , config = config
32
+ service_name ,
33
+ region_name = region_name ,
34
+ api_version = api_version ,
35
+ use_ssl = use_ssl ,
36
+ verify = verify ,
37
+ endpoint_url = endpoint_url ,
38
+ aws_access_key_id = nifcloud_access_key_id ,
39
+ aws_secret_access_key = nifcloud_secret_access_key ,
40
+ aws_session_token = nifcloud_session_token ,
41
+ config = config
43
42
)
44
- client ._endpoint .http_session = CustomURLLib3Session ()
43
+
44
+ http_session = client ._endpoint .http_session
45
+ if hasattr (http_session , '_manager' ):
46
+ http_session ._manager .connection_pool_kw ['ssl_context' ] = ssl_context
47
+
45
48
return client
46
49
47
50
0 commit comments