From 2a084833a36481265f50ac07d41baf3d5572996e Mon Sep 17 00:00:00 2001 From: martinb-bb <105685594+martinb-bb@users.noreply.github.com> Date: Thu, 17 Nov 2022 13:48:42 -0500 Subject: [PATCH] Fix for `covid` cases `export` (#3447) * fix * fixing export to have datetime as a column not index --- openbb_terminal/alternative/covid/covid_view.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/openbb_terminal/alternative/covid/covid_view.py b/openbb_terminal/alternative/covid/covid_view.py index 68c56c877dc..cb7f016c023 100644 --- a/openbb_terminal/alternative/covid/covid_view.py +++ b/openbb_terminal/alternative/covid/covid_view.py @@ -194,11 +194,11 @@ def display_covid_stat( plot : bool Flag to plot data """ + data = covid_model.get_covid_stat(country, stat, limit) if plot: plot_covid_stat(country, stat) if raw: - data = covid_model.get_covid_stat(country, stat, limit) print_rich_table( data, headers=[stat.title()], @@ -207,6 +207,12 @@ def display_covid_stat( title=f"[bold]{country} COVID {stat}[/bold]", ) if export: + data["date"] = data.index + data = data.reset_index(drop=True) + # make sure date is first column in export + cols = data.columns.tolist() + cols = cols[-1:] + cols[:-1] + data = data[cols] export_data(export, os.path.dirname(os.path.abspath(__file__)), stat, data)