Skip to content

Commit f09b5df

Browse files
committed
Updates to the logger
1 parent a3d11cb commit f09b5df

File tree

1 file changed

+23
-9
lines changed

1 file changed

+23
-9
lines changed

src/modacor/dataclasses/messagehandler.py

Lines changed: 23 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)