From bb5a7fcf30e11c7b955f46572de1ef4a3b773c97 Mon Sep 17 00:00:00 2001 From: Jonathan Weth Date: Fri, 31 Jul 2020 16:07:38 +0200 Subject: Return error status if connection to RWA wasn't successful Close #10 --- session.py | 26 ++++++++++++++++---------- 1 file changed, 16 insertions(+), 10 deletions(-) (limited to 'session.py') diff --git a/session.py b/session.py index 37cc312..e7ab922 100644 --- a/session.py +++ b/session.py @@ -107,21 +107,27 @@ class Session: def _register_session(self): """Register session in RWA if not in mockup_session mode.""" if not self.mockup_session: - r = requests.post( - REGISTER_URL, - json={ - "port": self.ws_port, - "password": self.password, - "pid": self.vnc_pid, - "trigger_port": self.trigger_port, - "trigger_token": self.trigger_token, - }, - ) + try: + r = requests.post( + REGISTER_URL, + json={ + "port": self.ws_port, + "password": self.password, + "pid": self.vnc_pid, + "trigger_port": self.trigger_port, + "trigger_token": self.trigger_token, + }, + ) + except requests.exceptions.ConnectionError: + raise ConnectionError() logging.info( f"The session has been registered in RWA with status code {r.status_code} and response {r.content.decode()}." ) + if r.status_code != 200: + raise ConnectionError() + self.meta = r.json() self.session_id = self.meta["session_id"] self.web_url = self.meta["url"] -- cgit v1.2.3