From 9af480dc624f37e1116db2b85a06deacb6c406d3 Mon Sep 17 00:00:00 2001
From: Artem Veremey <artem@veremey.net>
Date: Wed, 10 Mar 2021 13:59:45 -0800
Subject: [PATCH] Fix the prompt behavior if terminal is not tty

---
 gamestonk_terminal/discovery/disc_menu.py | 11 +++++++----
 gamestonk_terminal/menu.py                |  6 +++++-
 2 files changed, 12 insertions(+), 5 deletions(-)

diff --git a/gamestonk_terminal/discovery/disc_menu.py b/gamestonk_terminal/discovery/disc_menu.py
index bc6c3717efc..9818c4ca1fc 100644
--- a/gamestonk_terminal/discovery/disc_menu.py
+++ b/gamestonk_terminal/discovery/disc_menu.py
@@ -73,10 +73,13 @@ def disc_menu():
     # Loop forever and ever
     while True:
         # Get input command from user
-        as_input = session.prompt(
-            f"{get_flair()} (disc)> ",
-            completer=word_completer,
-        )
+        if session:
+            as_input = session.prompt(
+                f"{get_flair()} (disc)> ",
+                completer=word_completer,
+            )
+        else:
+            as_input = input(f"{get_flair()} (disc)> ")
 
         # Parse fundamental analysis command of the list of possible commands
         try:
diff --git a/gamestonk_terminal/menu.py b/gamestonk_terminal/menu.py
index 4c898945042..c61f7587f06 100644
--- a/gamestonk_terminal/menu.py
+++ b/gamestonk_terminal/menu.py
@@ -4,4 +4,8 @@ from prompt_toolkit import PromptSession
 from prompt_toolkit.history import FileHistory
 
 history_file = os.path.join(os.path.expanduser("~"), ".gamestonk_terminal.his")
-session = PromptSession(history=FileHistory(history_file))
+
+try:
+    session = PromptSession(history=FileHistory(history_file))
+except Exception as e:
+    session = None