Add try/except

This commit is contained in:
Ben Kulbertis
2021-03-01 17:53:13 -07:00
parent 96227baffb
commit 534432cf12

View File

@@ -310,8 +310,12 @@ def main():
b_quit = pm.pred_menu(df_stock, s_ticker, s_start, s_interval)
# If stock data is intradaily, we need to get data again as prediction techniques work on daily adjusted data
else:
ts = TimeSeries(key=cfg.API_KEY_ALPHAVANTAGE, output_format='pandas')
df_stock_pred, _ = ts.get_daily_adjusted(symbol=s_ticker, outputsize='full')
try:
ts = TimeSeries(key=cfg.API_KEY_ALPHAVANTAGE, output_format='pandas')
df_stock_pred, _ = ts.get_daily_adjusted(symbol=s_ticker, outputsize='full')
except:
print("Either the ticker or the API_KEY are invalids. Try again!")
return
df_stock_pred = df_stock_pred.sort_index(ascending=True)
df_stock_pred = df_stock_pred[s_start:]
b_quit = pm.pred_menu(df_stock_pred, s_ticker, s_start, s_interval="1440min")