jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-28 13:01:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Andrew reveiw after 9 months

Browse Source
This commit is contained in:
James Maslek
2023-07-24 09:57:20 -04:00
committed by Igor Radovanovic
parent 6a852c8be4
commit 69b121eb71
2 changed files with 25 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
openbb_terminal/forecast/forecast_controller.py
View File

@@ -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(

2
openbb_terminal/forecast/helpers.py
View File

@@ -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