Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,372 Bytes
c968fc3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
# Copyright (c) 2024 Amphion.
#
# This source code is licensed under the MIT license found in the
# LICENSE file in the root directory of this source tree.
import functools
import logging
__all__ = [
"logger",
]
class Logger(object):
def __init__(self, name: str = None):
name = "PaddleSpeech" if not name else name
self.logger = logging.getLogger(name)
log_config = {
"DEBUG": 10,
"INFO": 20,
"TRAIN": 21,
"EVAL": 22,
"WARNING": 30,
"ERROR": 40,
"CRITICAL": 50,
"EXCEPTION": 100,
}
for key, level in log_config.items():
logging.addLevelName(level, key)
if key == "EXCEPTION":
self.__dict__[key.lower()] = self.logger.exception
else:
self.__dict__[key.lower()] = functools.partial(self.__call__, level)
self.format = logging.Formatter(
fmt="[%(asctime)-15s] [%(levelname)8s] - %(message)s"
)
self.handler = logging.StreamHandler()
self.handler.setFormatter(self.format)
self.logger.addHandler(self.handler)
self.logger.setLevel(logging.INFO)
self.logger.propagate = False
def __call__(self, log_level: str, msg: str):
self.logger.log(log_level, msg)
logger = Logger()
|