jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-30 14:02:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix prompt-toolkit

Browse Source
This commit is contained in:
didier
2021-03-17 21:50:43 +00:00
parent b4c36b8df1
commit 1496602d02
1 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
gamestonk_terminal/behavioural_analysis/ba_controller.py
View File

@@ -19,6 +19,28 @@ from prompt_toolkit.completion import NestedCompleter
class BehaviouralAnalysisController:
"""Behavioural Analysis Controller class"""
# Command choices
CHOICES = [
"help",
"q",
"quit",
"watchlist",
"spac",
"spac_c",
"wsb",
"popular",
"bullbear",
"messages",
"trending",
"stalker",
"infer",
"sentiment",
"mentions",
"regions",
"queries",
"rise",
]
def __init__(self, ticker: str, start: datetime):
"""Constructor"""
self.ticker = ticker
@@ -26,26 +48,7 @@ class BehaviouralAnalysisController:
self.ba_parser = argparse.ArgumentParser(add_help=False, prog="ba")
self.ba_parser.add_argument(
"cmd",
choices=[
"help",
"q",
"quit",
"watchlist",
"spac",
"spac_c",
"wsb",
"popular",
"bullbear",
"messages",
"trending",
"stalker",
"infer",
"sentiment",
"mentions",
"regions",
"queries",
"rise",
],
choices=self.CHOICES,
)
@staticmethod
@@ -217,6 +220,9 @@ def menu(ticker: str, start: datetime):
while True:
# Get input command from user
if session and gtff.USE_PROMPT_TOOLKIT:
completer = NestedCompleter.from_nested_dict(
{c: None for c in ba_controller.CHOICES}
)
an_input = session.prompt(
f"{get_flair()} (ba)> ",
completer=completer,