104 lines
3.0 KiB
Python
104 lines
3.0 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
import time
|
|
import inspect
|
|
from types import FrameType
|
|
from typing import List, Optional
|
|
|
|
from .types import LogMessage
|
|
from lib_not_dr.types.options import Options
|
|
|
|
|
|
class BaseOutputStream(Options):
|
|
name = 'BaseOutputStream'
|
|
|
|
level: int = 20
|
|
enable: bool = True
|
|
|
|
def write_stdout(self, message: LogMessage) -> None:
|
|
raise NotImplementedError(f'{self.__class__.__name__}.write_stdout is not implemented')
|
|
|
|
def write_stderr(self, message: LogMessage) -> None:
|
|
raise NotImplementedError(f'{self.__class__.__name__}.write_stderr is not implemented')
|
|
|
|
def flush(self) -> None:
|
|
raise NotImplementedError(f'{self.__class__.__name__}.flush is not implemented')
|
|
|
|
|
|
class Logger(Options):
|
|
name = 'Logger-v2'
|
|
|
|
outputs: List[BaseOutputStream] = []
|
|
|
|
enable: bool = True
|
|
level: int = 20
|
|
|
|
def log_for(self, level: int) -> bool:
|
|
"""
|
|
Check if logging is enabled for a specific level.
|
|
|
|
Args:
|
|
level (int): The logging level to check.
|
|
|
|
Returns:
|
|
bool: True if logging is enabled for the given level, False otherwise.
|
|
"""
|
|
return self.enable and level >= self.level
|
|
|
|
def add_output(self, output: BaseOutputStream) -> None:
|
|
"""
|
|
Add an output to the list of outputs.
|
|
|
|
Args:
|
|
output (BaseOutputStream): The output to be added.
|
|
|
|
Returns:
|
|
None
|
|
"""
|
|
self.outputs.append(output)
|
|
self.level = min(self.level, output.level)
|
|
|
|
def remove_output(self, output: BaseOutputStream) -> None:
|
|
"""
|
|
Removes the specified output from the list of outputs.
|
|
|
|
Args:
|
|
output (BaseOutputStream): The output to be removed.
|
|
|
|
Returns:
|
|
None
|
|
"""
|
|
self.outputs.remove(output)
|
|
self.level = max(self.level, *[output.level for output in self.outputs])
|
|
|
|
def make_log(self,
|
|
messages: List[str],
|
|
end: str = '\n',
|
|
split: str = ' ',
|
|
flush: bool = True,
|
|
level: int = 20,
|
|
logger_name: str = 'root',
|
|
logger_tag: Optional[str] = None,
|
|
stack_trace: Optional[FrameType] = None) -> None:
|
|
if not self.log_for(level):
|
|
return
|
|
if stack_trace is None:
|
|
if (stack := inspect.currentframe()) is not None:
|
|
if (up_stack := stack.f_back) is not None:
|
|
stack_trace = up_stack
|
|
else:
|
|
stack_trace = stack
|
|
|
|
message = LogMessage(messages=messages,
|
|
end=end,
|
|
split=split,
|
|
flush=flush,
|
|
level=level,
|
|
logger_name=logger_name,
|
|
logger_tag=logger_tag,
|
|
stack_trace=stack_trace)
|