Skip to content

Commit bdb98f3

Browse files
committed
Changed print+exit() to raising exceptions
1 parent 8947868 commit bdb98f3

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

VTSController.py

+3-5
Original file line numberDiff line numberDiff line change
@@ -34,15 +34,13 @@ async def authentication(self) -> None:
3434
print("inside if")
3535
res = await self.send_request(message_type='AuthenticationTokenRequest', data=self.base_info)
3636
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']}")
3938
self.__update_token(res['data']['authenticationToken'])
4039
return
4140

4241
res = await self.send_request(message_type='AuthenticationRequest', data={**self.base_info, 'authenticationToken': self.vts_token})
4342
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']}")
4644

4745
async def initialise(self) -> None:
4846
self.update_dotenv()
@@ -67,7 +65,7 @@ def update_dotenv(self) -> None:
6765
load_dotenv(override=True)
6866
self.vts_token = getenv("VTS_TOKEN")
6967

70-
def __update_token(self,token) -> None:
68+
def __update_token(self,token:str) -> None:
7169
self.vts_token = token
7270
set_key('.env', 'VTS_TOKEN', token)
7371

0 commit comments

Comments
 (0)