Skip to content

Commit

Permalink
Set debug level explicitly when parsing options
Browse files Browse the repository at this point in the history
  • Loading branch information
SeanBryan51 committed Apr 12, 2024
1 parent 8ca400f commit 26561d9
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/benchcab/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@

from benchcab.benchcab import Benchcab
from benchcab.cli import generate_parser
from benchcab.utils import get_logger
from benchcab.utils import get_logger, set_log_level


def parse_and_dispatch(parser):
Expand All @@ -26,7 +26,8 @@ def parse_and_dispatch(parser):
_ = args.pop("verbose")

# We just need to instantiate this with the desired level
get_logger(level=log_level)
logger = get_logger()
set_log_level(logger, log_level)

func = args.pop("func")
func(**args)
Expand Down
14 changes: 14 additions & 0 deletions src/benchcab/utils/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -143,3 +143,17 @@ def get_logger(name="benchcab", level="debug"):
logger.addHandler(handler)

return logger


def set_log_level(logger, level):
"""Set the log level for a logger.
Parameters
----------
logger: logging.Logger
A logger instance.
level : str, optional
Level, by default 'debug'.
"""
logger.setLevel(getattr(logging, level.upper()))

Check warning on line 159 in src/benchcab/utils/__init__.py

View check run for this annotation

Codecov / codecov/patch

src/benchcab/utils/__init__.py#L159

Added line #L159 was not covered by tests

0 comments on commit 26561d9

Please sign in to comment.