| | |
| | """Collects all loggers.""" |
| |
|
| | from .normal_logger import NormalLogger |
| | from .rich_logger import RichLogger |
| | from .dummy_logger import DummyLogger |
| |
|
| | __all__ = ['build_logger'] |
| |
|
| | _LOGGERS = { |
| | 'normal': NormalLogger, |
| | 'rich': RichLogger, |
| | 'dummy': DummyLogger |
| | } |
| |
|
| |
|
| | def build_logger(logger_type='normal', **kwargs): |
| | """Builds a logger. |
| | |
| | Args: |
| | logger_type: Type of logger, which is case insensitive. |
| | (default: `normal`) |
| | **kwargs: Additional arguments to build the logger. |
| | |
| | Raises: |
| | ValueError: If the `logger_type` is not supported. |
| | """ |
| | logger_type = logger_type.lower() |
| | if logger_type not in _LOGGERS: |
| | raise ValueError(f'Invalid logger type: `{logger_type}`!\n' |
| | f'Types allowed: {list(_LOGGERS)}.') |
| | return _LOGGERS[logger_type](**kwargs) |
| |
|
