33
44class LogMixin (object ):
55
6- def log (self , level , message ):
6+ def log (self , level , message , * args , ** kwargs ):
77 """
88 Custom log method that includes the trace ID.
99 """
1010 if isinstance (message , bytes ):
1111 message = message .decode ('utf-8' )
1212 trace_message = f"[{ self .trace_id } ] { message } "
13- getattr (logger , level )(trace_message )
13+ getattr (logger , level )(trace_message , * args , ** kwargs )
1414
1515 @property
1616 def logger (self ):
@@ -21,22 +21,22 @@ class LoggerAdapter:
2121 def __init__ (self , outer ):
2222 self .outer = outer
2323
24- def debug (self , message ):
25- self .outer .log ('debug' , message )
24+ def debug (self , message , * args , ** kwargs ):
25+ self .outer .log ('debug' , message , * args , ** kwargs )
2626
27- def info (self , message ):
28- self .outer .log ('info' , message )
27+ def info (self , message , * args , ** kwargs ):
28+ self .outer .log ('info' , message , * args , ** kwargs )
2929
30- def warning (self , message ):
31- self .outer .log ('warning' , message )
30+ def warning (self , message , * args , ** kwargs ):
31+ self .outer .log ('warning' , message , * args , ** kwargs )
3232
33- def error (self , message ):
34- self .outer .log ('error' , message )
33+ def error (self , message , * args , ** kwargs ):
34+ self .outer .log ('error' , message , * args , ** kwargs )
3535
36- def critical (self , message ):
37- self .outer .log ('critical' , message )
36+ def critical (self , message , * args , ** kwargs ):
37+ self .outer .log ('critical' , message , * args , ** kwargs )
3838
39- def exception (self , message ):
40- self .outer .log ('exception' , message )
39+ def exception (self , message , * args , ** kwargs ):
40+ self .outer .log ('exception' , message , * args , ** kwargs )
4141
4242 return LoggerAdapter (self )
0 commit comments