55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
# Copyright (c) 2024 Julian Müller (ChaoticByte)
|
||
|
|
|
||
|
|
from datetime import datetime as _datetime
|
||
|
|
from os import environ as _environ
|
||
|
|
from sys import stderr as _stderr
|
||
|
|
from sys import stdout as _stdout
|
||
|
|
from typing import Any as _Any
|
||
|
|
|
||
|
|
class ComponentLogger:
|
||
|
|
|
||
|
|
LEVELS = [
|
||
|
|
"DEBUG",
|
||
|
|
"INFO",
|
||
|
|
"WARN",
|
||
|
|
"ERROR",
|
||
|
|
"CRITICAL"
|
||
|
|
]
|
||
|
|
|
||
|
|
def __init__(self, component: str, level: int = 1, print_timestamp: bool = True):
|
||
|
|
'''level may be overwritten by environment variable LOGLEVEL'''
|
||
|
|
assert type(component) == str
|
||
|
|
assert type(level) == int
|
||
|
|
assert type(print_timestamp) == bool
|
||
|
|
self.component = component
|
||
|
|
self.level = level
|
||
|
|
if "LOGLEVEL" in _environ:
|
||
|
|
loglevel_ = _environ["LOGLEVEL"]
|
||
|
|
if loglevel_ in self.LEVELS:
|
||
|
|
self.level = self.LEVELS.index(loglevel_)
|
||
|
|
self.print_timestamp = print_timestamp
|
||
|
|
|
||
|
|
def _log(self, msg: _Any, level: int, file = _stdout):
|
||
|
|
assert type(level) == int
|
||
|
|
if level >= self.level:
|
||
|
|
if self.print_timestamp:
|
||
|
|
t = _datetime.now().astimezone().strftime(r'%Y-%m-%d %H:%M:%S %z')
|
||
|
|
print(f"[{t}] [{self.component}] [{self.LEVELS[level]}] {msg}", file=file)
|
||
|
|
else:
|
||
|
|
print(f"[{self.component}] [{self.LEVELS[level]}] {msg}", file=file)
|
||
|
|
|
||
|
|
def debug(self, msg: _Any):
|
||
|
|
self._log(msg, 0)
|
||
|
|
|
||
|
|
def info(self, msg: _Any):
|
||
|
|
self._log(msg, 1)
|
||
|
|
|
||
|
|
def warning(self, msg: _Any):
|
||
|
|
self._log(msg, 2, file=_stderr)
|
||
|
|
|
||
|
|
def error(self, msg: _Any):
|
||
|
|
self._log(msg, 3, file=_stderr)
|
||
|
|
|
||
|
|
def critical(self, msg: _Any):
|
||
|
|
self._log(msg, 4, file=_stderr)
|