File tree 1 file changed +3
-5
lines changed
1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -34,15 +34,13 @@ async def authentication(self) -> None:
34
34
print ("inside if" )
35
35
res = await self .send_request (message_type = 'AuthenticationTokenRequest' , data = self .base_info )
36
36
if res ['messageType' ] == 'APIError' :
37
- print (f"Error occured:\n \t { res ['data' ]['message' ]} " )
38
- exit ()
37
+ raise Exception (f"Error occured:\n \t { res ['data' ]['message' ]} " )
39
38
self .__update_token (res ['data' ]['authenticationToken' ])
40
39
return
41
40
42
41
res = await self .send_request (message_type = 'AuthenticationRequest' , data = {** self .base_info , 'authenticationToken' : self .vts_token })
43
42
if not res ['data' ]['authenticated' ]:
44
- print (f"Couldn't connect to the API: { res ['data' ]['reason' ]} " )
45
- exit ()
43
+ raise ConnectionError (f"Couldn't connect to the API: { res ['data' ]['reason' ]} " )
46
44
47
45
async def initialise (self ) -> None :
48
46
self .update_dotenv ()
@@ -67,7 +65,7 @@ def update_dotenv(self) -> None:
67
65
load_dotenv (override = True )
68
66
self .vts_token = getenv ("VTS_TOKEN" )
69
67
70
- def __update_token (self ,token ) -> None :
68
+ def __update_token (self ,token : str ) -> None :
71
69
self .vts_token = token
72
70
set_key ('.env' , 'VTS_TOKEN' , token )
73
71
You can’t perform that action at this time.
0 commit comments