16
16
InfuraKeyNotFound ,
17
17
)
18
18
19
- INFURA_MAINNET_DOMAIN = ' mainnet.infura.io'
20
- INFURA_ROPSTEN_DOMAIN = ' ropsten.infura.io'
21
- INFURA_GOERLI_DOMAIN = ' goerli.infura.io'
22
- INFURA_RINKEBY_DOMAIN = ' rinkeby.infura.io'
23
- INFURA_KOVAN_DOMAIN = ' kovan.infura.io'
19
+ INFURA_MAINNET_DOMAIN = " mainnet.infura.io"
20
+ INFURA_ROPSTEN_DOMAIN = " ropsten.infura.io"
21
+ INFURA_GOERLI_DOMAIN = " goerli.infura.io"
22
+ INFURA_RINKEBY_DOMAIN = " rinkeby.infura.io"
23
+ INFURA_KOVAN_DOMAIN = " kovan.infura.io"
24
24
25
- WEBSOCKET_SCHEME = ' wss'
26
- HTTP_SCHEME = ' https'
25
+ WEBSOCKET_SCHEME = " wss"
26
+ HTTP_SCHEME = " https"
27
27
28
28
29
29
def load_api_key () -> str :
30
30
# in web3py v6 remove outdated WEB3_INFURA_API_KEY
31
- key = os .environ .get ('WEB3_INFURA_PROJECT_ID' ,
32
- os .environ .get ('WEB3_INFURA_API_KEY' , '' ))
33
- if key == '' :
31
+ key = os .environ .get (
32
+ "WEB3_INFURA_PROJECT_ID" , os .environ .get ("WEB3_INFURA_API_KEY" , "" )
33
+ )
34
+ if key == "" :
34
35
raise InfuraKeyNotFound (
35
36
"No Infura Project ID found. Please ensure "
36
37
"that the environment variable WEB3_INFURA_PROJECT_ID is set."
@@ -39,25 +40,25 @@ def load_api_key() -> str:
39
40
40
41
41
42
def load_secret () -> str :
42
- return os .environ .get (' WEB3_INFURA_API_SECRET' , '' )
43
+ return os .environ .get (" WEB3_INFURA_API_SECRET" , "" )
43
44
44
45
45
46
def build_http_headers () -> Optional [Dict [str , Tuple [str , str ]]]:
46
47
secret = load_secret ()
47
48
if secret :
48
- headers = {' auth' : ('' , secret )}
49
+ headers = {" auth" : ("" , secret )}
49
50
return headers
50
51
return None
51
52
52
53
53
54
def build_infura_url (domain : str ) -> URI :
54
- scheme = os .environ .get (' WEB3_INFURA_SCHEME' , WEBSOCKET_SCHEME )
55
+ scheme = os .environ .get (" WEB3_INFURA_SCHEME" , WEBSOCKET_SCHEME )
55
56
key = load_api_key ()
56
57
secret = load_secret ()
57
58
58
- if scheme == WEBSOCKET_SCHEME and secret != '' :
59
+ if scheme == WEBSOCKET_SCHEME and secret != "" :
59
60
return URI (f"{ scheme } ://:{ secret } @{ domain } /ws/v3/{ key } " )
60
- elif scheme == WEBSOCKET_SCHEME and secret == '' :
61
+ elif scheme == WEBSOCKET_SCHEME and secret == "" :
61
62
return URI (f"{ scheme } ://{ domain } /ws/v3/{ key } " )
62
63
elif scheme == HTTP_SCHEME :
63
64
return URI (f"{ scheme } ://{ domain } /v3/{ key } " )
0 commit comments