Skip to content

Commit f1e580e

Browse files
committed
Add a get_logger function to allow users to get same predefined type of logger
1 parent fdcb070 commit f1e580e

File tree

2 files changed

+15
-4
lines changed

2 files changed

+15
-4
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# Copyright (c) Cosmo Tech corporation.
22
# Licensed under the MIT license.
33

4-
__version__ = '0.8.0'
4+
__version__ = '0.8.1'

cosmotech/coal/utils/logger.py

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
from rich.logging import RichHandler
1212
from rich.highlighter import NullHighlighter
1313

14-
LOGGER = logging.getLogger("csm.data")
1514
HIGLIGHTER = NullHighlighter()
1615
HANDLER = RichHandler(rich_tracebacks=True,
1716
omit_repeated_times=False,
@@ -29,5 +28,17 @@
2928
)
3029

3130
HANDLER.setFormatter(FORMATTER)
32-
LOGGER.addHandler(HANDLER)
33-
LOGGER.setLevel(logging.INFO)
31+
32+
33+
def get_logger(
34+
logger_name: str,
35+
level=logging.INFO
36+
) -> logging.Logger:
37+
_logger = logging.getLogger(logger_name)
38+
if not _logger.hasHandlers():
39+
_logger.addHandler(HANDLER)
40+
_logger.setLevel(level)
41+
return _logger
42+
43+
44+
LOGGER = get_logger("csm.data")

0 commit comments

Comments
 (0)