From 56d86bdd56fa0a24639570943fd5872f769f511d Mon Sep 17 00:00:00 2001 From: james Date: Tue, 15 Mar 2022 11:15:33 -0400 Subject: [PATCH] clean up yfinance index commands for errors --- gamestonk_terminal/economy/yfinance_model.py | 10 ++++++---- gamestonk_terminal/economy/yfinance_view.py | 4 +++- 2 files changed, 9 insertions(+), 5 deletions(-) diff --git a/gamestonk_terminal/economy/yfinance_model.py b/gamestonk_terminal/economy/yfinance_model.py index efe036ab442..81c3cea5ddf 100644 --- a/gamestonk_terminal/economy/yfinance_model.py +++ b/gamestonk_terminal/economy/yfinance_model.py @@ -71,14 +71,16 @@ def get_index( ) if column not in index_data.columns: - return console.print( + console.print( f"The chosen column is not available for {ticker}. Please choose " - f"between: {', '.join(index_data.columns)}" + f"between: {', '.join(index_data.columns)}\n" ) + return pd.Series() if index_data.empty or len(index_data) < 2: - return console.print( + console.print( f"The chosen index {ticker}, returns no data. Please check if " - f"there is any data available." + f"there is any data available.\n" ) + return pd.Series() return index_data[column] diff --git a/gamestonk_terminal/economy/yfinance_view.py b/gamestonk_terminal/economy/yfinance_view.py index 4b85b2cb51b..8e931c22115 100644 --- a/gamestonk_terminal/economy/yfinance_view.py +++ b/gamestonk_terminal/economy/yfinance_view.py @@ -66,7 +66,9 @@ def show_indices( indices_data: pd.DataFrame = pd.DataFrame() for index in indices: - indices_data[index] = get_index(index, interval, start_date, end_date, column) + index_data = get_index(index, interval, start_date, end_date, column) + if not index_data.empty: + indices_data[index] = index_data if not store: if external_axes is None: