From 291cc7443f7ff82061a80965b8ea9b8f2fae0fa2 Mon Sep 17 00:00:00 2001 From: Colin Delahunty <72827203+colin99d@users.noreply.github.com> Date: Mon, 24 Jan 2022 12:19:28 -0500 Subject: [PATCH] Improved debugging (#1257) --- gamestonk_terminal/decorators.py | 3 +++ terminal.py | 4 ---- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/gamestonk_terminal/decorators.py b/gamestonk_terminal/decorators.py index 084961169a7..a28ef823d9a 100644 --- a/gamestonk_terminal/decorators.py +++ b/gamestonk_terminal/decorators.py @@ -4,6 +4,8 @@ import functools import logging import os +from gamestonk_terminal.rich_config import console + logger = logging.getLogger(__name__) @@ -23,6 +25,7 @@ def try_except(f): try: return f(*args, **kwargs) except Exception as e: + console.print(f"[red]Error: {e}[/red]") logger.exception("%s", type(e).__name__) return [] diff --git a/terminal.py b/terminal.py index 112eb33789d..ed778d3d92b 100644 --- a/terminal.py +++ b/terminal.py @@ -10,7 +10,6 @@ from typing import List import pytz from prompt_toolkit.completion import NestedCompleter - from gamestonk_terminal.rich_config import console from gamestonk_terminal.parent_classes import BaseController from gamestonk_terminal import feature_flags as gtff @@ -321,8 +320,6 @@ if __name__ == "__main__": if "--debug" in sys.argv: os.environ["DEBUG_MODE"] = "true" sys.argv.remove("--debug") - else: - os.environ["DEBUG_MODE"] = "false" if len(sys.argv) > 1 and ".gst" in sys.argv[1]: if os.path.isfile(sys.argv[1]): with open(sys.argv[1]) as fp: @@ -336,5 +333,4 @@ if __name__ == "__main__": else: terminal(sys.argv[1:]) else: - os.environ["DEBUG_MODE"] = "false" terminal()