DR_Team/lib-not-dr
5
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add use stderr option for stdio output stream

Browse Source
This commit is contained in:
shenjack 2023-11-06 21:59:55 +08:00
parent 880669b050
commit 87b2e3b00c
Signed by: shenjack
GPG Key ID: 7B1134A979775551
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib_not_dr/logger/outstream.py
View File

@ -52,6 +52,7 @@ class StdioOutputStream(BaseOutputStream):
level: int = LogLevel.info level: int = LogLevel.info
formatter: BaseFormatter = StdFormatter() formatter: BaseFormatter = StdFormatter()
use_stderr: bool = True
def write_stdout(self, message: LogMessage) -> None: def write_stdout(self, message: LogMessage) -> None:
if not self.enable: if not self.enable:
@ -66,7 +67,10 @@ class StdioOutputStream(BaseOutputStream):
return None return None
if message.level < self.level: if message.level < self.level:
return None return None
print(self.formatter.format_message(message), end='', flush=message.flush, file=sys.stderr) if self.use_stderr:
print(self.formatter.format_message(message), end='', flush=message.flush, file=sys.stderr)
else:
print(self.formatter.format_message(message), end='', flush=message.flush)
return None return None
def flush(self) -> None: def flush(self) -> None: