@@ -81,7 +81,6 @@ def setup(cls, logger: logging.Logger) -> None:
8181        _handler  =  logging .StreamHandler ()
8282        _handler .setFormatter (_formatter )
8383        logger .addHandler (_handler )
84-         logger .setLevel (logging .INFO )
8584
8685
8786test_logger  =  logging .getLogger (__name__ )
@@ -95,6 +94,7 @@ def setup(cls, logger: logging.Logger) -> None:
9594_DONE_SUFFIX : Final [str ] =  " ✅" 
9695_RAISED_PREFIX : Final [str ] =  "❌❌❌ Error: " 
9796_RAISED_SUFFIX : Final [str ] =  " ❌❌❌" 
97+ _STACK_LEVEL_OFFSET : Final [int ] =  3 
9898
9999
100100@dataclass  
@@ -170,25 +170,28 @@ def log_context(
170170            _resolve (ctx_msg .starting , _STARTING_PREFIX , _STARTING_SUFFIX ),
171171            * args ,
172172            ** kwargs ,
173+             stacklevel = _STACK_LEVEL_OFFSET ,
173174        )
174175        with  _increased_logger_indent (logger ):
175176            yield  SimpleNamespace (logger = logger , messages = ctx_msg )
176177        elapsed_time  =  datetime .datetime .now (tz = datetime .UTC ) -  started_time 
177-         done_message  =  f"{ _resolve (ctx_msg .done , _DONE_PREFIX , _DONE_SUFFIX )}   ({ _timedelta_as_minute_second_ms (elapsed_time )}  )" 
178+         done_message  =  f"{ _resolve (ctx_msg .done , _DONE_PREFIX , _DONE_SUFFIX )}   (total time spent:  { _timedelta_as_minute_second_ms (elapsed_time )}  )" 
178179        logger .log (
179180            level ,
180181            done_message ,
181182            * args ,
182183            ** kwargs ,
184+             stacklevel = _STACK_LEVEL_OFFSET ,
183185        )
184186
185187    except :
186188        elapsed_time  =  datetime .datetime .now (tz = datetime .UTC ) -  started_time 
187-         error_message  =  f"{ _resolve (ctx_msg .raised , _RAISED_PREFIX , _RAISED_SUFFIX )}   ({ _timedelta_as_minute_second_ms (elapsed_time )}  )" 
189+         error_message  =  f"{ _resolve (ctx_msg .raised , _RAISED_PREFIX , _RAISED_SUFFIX )}   (total time spent:  { _timedelta_as_minute_second_ms (elapsed_time )}  )" 
188190        logger .exception (
189191            error_message ,
190192            * args ,
191193            ** kwargs ,
194+             stacklevel = _STACK_LEVEL_OFFSET ,
192195        )
193196        raise 
194197
0 commit comments