From 1496602d02efb70253bde01663f72f4b1f2f651c Mon Sep 17 00:00:00 2001
From: didier <dro.lopes@campus.fct.unl.pt>
Date: Wed, 17 Mar 2021 21:50:43 +0000
Subject: [PATCH] fix prompt-toolkit

---
 .../behavioural_analysis/ba_controller.py     | 46 +++++++++++--------
 1 file changed, 26 insertions(+), 20 deletions(-)

diff --git a/gamestonk_terminal/behavioural_analysis/ba_controller.py b/gamestonk_terminal/behavioural_analysis/ba_controller.py
index 6a204c82f26..20bb0d1c1a4 100644
--- a/gamestonk_terminal/behavioural_analysis/ba_controller.py
+++ b/gamestonk_terminal/behavioural_analysis/ba_controller.py
@@ -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,