log_helper: ensure stdout handles unicode if logging there

This way, all modules that use the log_helper do not need to call ensure_stdout_handles_unicode (e.g. msodde, olevba)

log_helper: ensure stdout handles unicode if logging there
This way, all modules that use the log_helper do not need to call ensure_stdout_handles_unicode (e.g. msodde, olevba)
Christian Herdtweck
1 parent 5bf585f5
Showing 1 changed file with 4 additions and 0 deletions
oletools/common/log_helper/log_helper.py
@@ -44,6 +44,7 @@ General logging helpers
 from ._json_formatter import JsonFormatter
 from ._logger_adapter import OletoolsLoggerAdapter
 from . import _root_logger_wrapper
+from ..io_encoding import ensure_stdout_handles_unicode
 import logging
 import sys
  
@@ -92,6 +93,9 @@ class LogHelper:
         if self._is_enabled:
             raise ValueError('re-enabling logging. Not sure whether that is ok...')
  
+        if stream in (None, sys.stdout):
+            ensure_stdout_handles_unicode()
+
         log_level = LOG_LEVELS[level]
         logging.basicConfig(level=log_level, format=log_format, stream=stream)
         self._is_enabled = True