mirror of
https://github.com/vacp2p/status-benchmarks.git
synced 2026-01-09 15:37:54 -05:00
Add TRACE to logger.py
This commit is contained in:
@@ -1,10 +1,18 @@
|
||||
# Python Imports
|
||||
import logging
|
||||
import logging.config
|
||||
import pathlib
|
||||
import yaml
|
||||
|
||||
# Project Imports
|
||||
class TraceLogger(logging.Logger):
|
||||
TRACE = 5
|
||||
def trace(self, msg, *args, **kwargs) -> None:
|
||||
self.log(self.TRACE, msg, *args, **kwargs)
|
||||
|
||||
# Register level name and custom class BEFORE dictConfig/getLogger
|
||||
logging.addLevelName(TraceLogger.TRACE, "TRACE")
|
||||
logging.setLoggerClass(TraceLogger)
|
||||
|
||||
# Load YAML config
|
||||
with open(pathlib.Path(__file__).parent.resolve() / 'logger_config.yaml', 'r') as f:
|
||||
config = yaml.safe_load(f.read())
|
||||
config = yaml.safe_load(f)
|
||||
logging.config.dictConfig(config)
|
||||
|
||||
Reference in New Issue
Block a user