From 56bc55e1207190f71f526e8a956456ae89fbe9b8 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Wed, 25 Jan 2023 11:18:13 +0000 Subject: [PATCH] handle no connection --- openbb_terminal/account/hub_model.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/openbb_terminal/account/hub_model.py b/openbb_terminal/account/hub_model.py index 5765982568e..7b75fecd7d8 100644 --- a/openbb_terminal/account/hub_model.py +++ b/openbb_terminal/account/hub_model.py @@ -41,12 +41,12 @@ def create_session( return None -def process_session_response(response: Optional[requests.Response]) -> dict: +def process_session_response(response: requests.Response) -> dict: """Process the response from the login request. Parameters ---------- - response : Optional[requests.Response] + response : requests.Response The response from the login request. Returns @@ -54,9 +54,6 @@ def process_session_response(response: Optional[requests.Response]) -> dict: dict The login info. """ - if response is None: - return {} - if response.status_code == 200: console.print("\nLogin successful.", style="green") login = response.json() @@ -87,6 +84,8 @@ def get_session(email: str, password: str) -> dict: The session info. """ response = create_session(email, password) + if response is None: + return {} return process_session_response(response)