summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJames Maslek <jmaslek11@gmail.com>2023-07-24 09:57:20 -0400
committerJames Maslek <jmaslek11@gmail.com>2023-07-24 09:57:20 -0400
commit43eb52734e34636ddedfbd638065fdbc6c01928f (patch)
tree156d3099d902ffc6015ba86bd802fa01ad40ea31
parent02767ca733073bb4bcfb61ffd162598c86f844a6 (diff)
Andrew reveiw after 9 months
-rw-r--r--openbb_terminal/forecast/forecast_controller.py32
-rw-r--r--openbb_terminal/forecast/helpers.py2
2 files changed, 25 insertions, 9 deletions
diff --git a/openbb_terminal/forecast/forecast_controller.py b/openbb_terminal/forecast/forecast_controller.py
index 630d8e39e47..b4386fa6e75 100644
--- a/openbb_terminal/forecast/forecast_controller.py
+++ b/openbb_terminal/forecast/forecast_controller.py
@@ -1039,7 +1039,9 @@ class ForecastController(BaseController):
if ns_parser:
# check proper file name is provided
if not ns_parser.target_dataset:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
df = self.datasets[ns_parser.target_dataset]
@@ -1079,13 +1081,17 @@ class ForecastController(BaseController):
return
if not ns_parser.values:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
values = [x.strip() for x in ns_parser.values.split(",")]
target_df = values[0].split(".")[0]
if target_df not in self.datasets:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
for value in values:
@@ -1148,7 +1154,9 @@ class ForecastController(BaseController):
return
if not ns_parser.values:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
try:
@@ -1191,7 +1199,9 @@ class ForecastController(BaseController):
if ns_parser:
# check proper file name is provided
if not ns_parser.target_dataset:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
data = self.datasets[ns_parser.target_dataset]
@@ -1311,7 +1321,9 @@ class ForecastController(BaseController):
if ns_parser:
# check proper file name is provided
if not ns_parser.target_dataset:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
(
@@ -1412,7 +1424,9 @@ class ForecastController(BaseController):
if ns_parser:
# check proper file name is provided
if not ns_parser.target_dataset:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
df = forecast_model.add_sto(
@@ -1499,7 +1513,9 @@ class ForecastController(BaseController):
if ns_parser:
# check proper file name is provided
if not ns_parser.target_dataset:
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print(
+ "[red]Please select a valid dataset with the -d flag.\n[/red]"
+ )
return
self.datasets[ns_parser.target_dataset] = forecast_model.add_rsi(
diff --git a/openbb_terminal/forecast/helpers.py b/openbb_terminal/forecast/helpers.py
index 238b6b33f1a..7c5d4193963 100644
--- a/openbb_terminal/forecast/helpers.py
+++ b/openbb_terminal/forecast/helpers.py
@@ -1153,7 +1153,7 @@ def check_parser_input(parser: argparse.ArgumentParser, datasets, *args) -> bool
if not hasattr(parser, "target_dataset"):
return False
if not parser.target_dataset: # type: ignore
- console.print("[red]Please enter valid dataset.\n[/red]")
+ console.print("[red]Please select a valid dataset with the -d flag.\n[/red]")
return False
if "ignore_column" in args:
return True