diff --git a/rwx/log/__init__.py b/rwx/log/__init__.py index fcf01aa..50048e5 100644 --- a/rwx/log/__init__.py +++ b/rwx/log/__init__.py @@ -18,15 +18,15 @@ def get_file_logger(name: str) -> logging.Logger: return logger -def get_stream_logger() -> logging.Logger: +def get_stream_logger(level: int) -> logging.Logger: out_handler = logging.StreamHandler(stream=sys.stdout) - out_handler.setLevel(logging.INFO) + out_handler.setLevel(level) # logger = logging.getLogger() logger.addHandler(out_handler) - logger.setLevel(logging.INFO) + logger.setLevel(level) # return logger -stream = get_stream_logger() +stream = get_stream_logger(logging.INFO)