diff options
Diffstat (limited to 'openbb_terminal/common/common_model.py')
-rw-r--r-- | openbb_terminal/common/common_model.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/openbb_terminal/common/common_model.py b/openbb_terminal/common/common_model.py index 1130469b4de..bbf1d98cabf 100644 --- a/openbb_terminal/common/common_model.py +++ b/openbb_terminal/common/common_model.py @@ -52,6 +52,7 @@ file_types = ["xlsx", "csv"] @log_start_end(log=logger) def load( file: str, + sheet_name: Optional[str] = None, data_files: Optional[Dict[Any, Any]] = None, data_examples: Optional[Dict[Any, Any]] = None, ) -> pd.DataFrame: @@ -90,7 +91,15 @@ def load( try: if file_type == ".xlsx": - data = pd.read_excel(full_file) + try: + data = ( + pd.read_excel(full_file) + if sheet_name is None + else pd.read_excel(full_file, sheet_name=sheet_name) + ) + except ValueError as err: + console.print(f"[red]{err}[/red]\n") + return pd.DataFrame() elif file_type == ".csv": data = pd.read_csv(full_file) else: |