@@ -13,26 +13,40 @@ class MessageHandler:
1313 Args:
1414 level (int): The logging level to use. Defaults to logging.INFO.
1515 """
16- def __init__ (self , level : int = logging .INFO , ** kwargs ):
16+ def __init__ (self , level : int = logging .INFO , name : str = 'MoDaCor' ** kwargs ):
1717 self .level = level
18+ self .name = name
1819
19- def log (self , message : str , level : int = None ):
20+ self .logger = logging .getLogger (name )
21+ self .logger .setLevel (level )
22+
23+ self .consoleLogHandler = logging .StreamHandler ()
24+ self .consoleLogHandler .setLevel (level )
25+
26+ formatter = logging .Formatter ('%(asctime)s - %(name)s - %(levelname)s - %(message)s' )
27+ self .consoleLogHandler .setFormatter (formatter )
28+ self .logger .addHandler (self .consoleLogHandler )
29+
30+ def log (self , message : str , level : int = None , name : str = None ):
2031 if level is None :
2132 level = self .level
22- logger .log (level , message )
33+
34+ if name is None :
35+ name = self .name
36+
37+ self .logger (message , level = level , name = name )
2338
2439 def info (self , message : str ):
25- self .log (message , level = logging .INFO )
40+ self .log (message , level = logging .INFO , name = 'MoDaCor' )
2641
2742 def warning (self , message : str ):
28- self .log (message , level = logging .WARNING )
43+ self .log (message , level = logging .WARNING , name = 'MoDaCor' )
2944
3045 def error (self , message : str ):
31- self .log (message , level = logging .ERROR )
46+ self .log (message , level = logging .ERROR , name = 'MoDaCor' )
3247
3348 def critical (self , message : str ):
34- self .log (message , level = logging .CRITICAL )
49+ self .log (message , level = logging .CRITICAL , name = 'MoDaCor' )
3550
3651 def debug (self , message : str ):
37- self .log (message , level = logging .DEBUG )
38-
52+ self .log (message , level = logging .DEBUG , name = 'MoDaCor' )
0 commit comments