Only print traceback when debug is enabled

jo1gi · Nov 29, 2023 · d4fe47d · d4fe47d
1 parent b6af29d
commit d4fe47d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/audiobookdl/__main__.py b/audiobookdl/__main__.py
@@ -9,7 +9,6 @@
 import sys
 from rich.prompt import Prompt
 from typing import List, Optional, Union
-import traceback
 
 
 def main() -> None:
@@ -32,7 +31,8 @@ def main() -> None:
             process_url(url, options, config)
     except AudiobookDLException as e:
         e.print()
-        traceback.print_exc()
+        if logging.debug_mode:
+            logging.print_traceback()
         exit(1)
 
 

diff --git a/audiobookdl/logging.py b/audiobookdl/logging.py
@@ -5,6 +5,7 @@
 from rich.progress import Progress, ProgressColumn
 from typing import Union, List
 from audiobookdl.utils import read_asset_file
+import traceback
 
 debug_mode = False
 quiet_mode = False
@@ -62,3 +63,9 @@ def simple_help() -> None:
 
 def progress(progress_format: List[Union[str, ProgressColumn]]) -> Progress:
     return Progress(*progress_format, console=console)
+
+def print_traceback() -> None:
+    """Print traceback"""
+    console.print()
+    console.print("[underline white bold]Traceback[/]")
+    traceback.print_exc()