diff --git a/src/tools.py b/src/tools.py index 1b7784399cf0f3257cf5d6257990aa91acf4380b..657752e0bcdbfce01739865d30a96aff31764b9d 100644 --- a/src/tools.py +++ b/src/tools.py @@ -67,6 +67,16 @@ def logger_color_reset(): return f"{Style.RESET_ALL}" return "" +def logger_exception(exception): + logger.error( + f"[{Fore.RED}ERROR{Style.RESET_ALL}] " + + f"{Fore.RED}{e}{Style.RESET_ALL}" + ) + logger.debug( + f"{Fore.BLUE}{e}{Style.RESET_ALL}\n" + + f"{Fore.RED}{traceback.format_exc()}{Style.RESET_ALL}" + ) + ########## # TIMERS # ########## @@ -135,12 +145,20 @@ def timer(func): try: value = func(*args, **kwargs) except Exception as e: - logger.debug(f"[{Fore.RED}ERROR{Style.RESET_ALL}]" + - f"[{func.__module__}.{Fore.GREEN}" + - f"{func.__qualname__}" + - f"{Style.RESET_ALL}]: " + - f"{Fore.RED}{e}{Style.RESET_ALL}") - traceback.print_exc() + logger.error( + f"[{Fore.RED}ERROR{Style.RESET_ALL}] " + + f"{Fore.RED}{e}{Style.RESET_ALL}" + ) + logger.debug( + f"[{func.__module__}.{Fore.GREEN}" + + f"{func.__qualname__}" + + f"{Style.RESET_ALL}]: " + + f"{Fore.RED}{e}{Style.RESET_ALL}" + ) + logger.debug( + f"{Fore.BLUE}{e}{Style.RESET_ALL}\n" + + f"{Fore.RED}{traceback.format_exc()}{Style.RESET_ALL}" + ) end_time = time.perf_counter() run_time = end_time - start_time @@ -329,11 +347,7 @@ class SQL(object): value = self._fetch(res, fetch_one) except Exception as e: - logger.error( - f"[{Fore.RED}ERROR{Style.RESET_ALL}] " + - f"{Fore.RED}{e}{Style.RESET_ALL}" - ) - traceback.print_exc() + logger_exception(e) finally: return value