From 28a7b2f1401c104cb78dba870f50c13b7d8a2ae7 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Fri, 1 Nov 2024 08:42:08 -0400 Subject: [PATCH] Refactor logging configuration to enable/disable logging --- simplemind/logging.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/simplemind/logging.py b/simplemind/logging.py index c18d718..06defc3 100644 --- a/simplemind/logging.py +++ b/simplemind/logging.py @@ -14,13 +14,17 @@ def logger(func: Callable[..., Any]) -> Callable[..., Any]: def wrapper(*args, **kwargs) -> Any: if not settings.logging.is_enabled: return func(*args, **kwargs) + logfire.info(f"Calling {func.__name__} with args: {args}, kwargs: {kwargs}") t1 = time.perf_counter() + try: result = func(*args, **kwargs) t2 = time.perf_counter() logfire.info(f"{func.__name__} returned: {result} in {t2-t1} seconds") + return result + except Exception as e: t2 = time.perf_counter() logfire.error(f"Error in {func.__name__}: {e} in {t2-t1} seconds")