diff options
Diffstat (limited to 'src/pyssg/custom_logger.py')
-rw-r--r-- | src/pyssg/custom_logger.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/pyssg/custom_logger.py b/src/pyssg/custom_logger.py index 55f3e2d..4eebc4c 100644 --- a/src/pyssg/custom_logger.py +++ b/src/pyssg/custom_logger.py @@ -2,6 +2,7 @@ import sys from logging import (Logger, StreamHandler, Formatter, LogRecord, DEBUG, INFO, WARNING, ERROR, CRITICAL, getLogger) +from typing import TextIO LOG_LEVEL: int = INFO # 'pyssg' es the name of the root logger @@ -38,7 +39,7 @@ class PerLevelFormatter(Formatter): def setup_logger(name: str = LOGGER_NAME, level: int = LOG_LEVEL) -> None: logger: Logger = getLogger(name) - handler: StreamHandler = StreamHandler(sys.stdout) + handler: StreamHandler[TextIO] = StreamHandler(sys.stdout) logger.setLevel(level) handler.setLevel(level) handler.setFormatter(PerLevelFormatter()) |