Skip to content

Commit 2c1a1f3

Browse files
[add] Changing effective log level work (#178)
1 parent e237394 commit 2c1a1f3

File tree

2 files changed

+31
-7
lines changed

2 files changed

+31
-7
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
[tool.poetry]
22
name = "redisbench-admin"
3-
version = "0.4.4"
3+
version = "0.4.5"
44
description = "Redis benchmark run helper. A wrapper around Redis and Redis Modules benchmark tools ( ftsb_redisearch, memtier_benchmark, redis-benchmark, aibench, etc... )."
55
authors = ["filipecosta90 <[email protected]>"]
66
readme = "README.md"

redisbench_admin/cli.py

Lines changed: 30 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -115,12 +115,22 @@ def main():
115115
level=LOG_LEVEL,
116116
)
117117
else:
118-
# logging settings
119-
logging.basicConfig(
120-
format=LOG_FORMAT,
121-
level=LOG_LEVEL,
122-
datefmt=LOG_DATEFMT,
123-
)
118+
logger = logging.getLogger()
119+
logger.setLevel(LOG_LEVEL)
120+
121+
# create console handler and set level to debug
122+
ch = logging.StreamHandler()
123+
ch.setLevel(LOG_LEVEL)
124+
125+
# create formatter
126+
formatter = logging.Formatter(LOG_FORMAT)
127+
128+
# add formatter to ch
129+
ch.setFormatter(formatter)
130+
131+
# add ch to logger
132+
logger.addHandler(ch)
133+
print_stdout_effective_log_level()
124134

125135
if requested_tool == "run-local":
126136
run_local_command_logic(args, project_name, project_version)
@@ -136,6 +146,20 @@ def main():
136146
compare_command_logic(args, project_name, project_version)
137147

138148

149+
def print_stdout_effective_log_level():
150+
effective_log_level = "N/A"
151+
effective_log_level = logging.getLogger().getEffectiveLevel()
152+
if effective_log_level == logging.DEBUG:
153+
effective_log_level = "DEBUG"
154+
if effective_log_level == logging.INFO:
155+
effective_log_level = "INFO"
156+
if effective_log_level == logging.WARN:
157+
effective_log_level = "WARN"
158+
if effective_log_level == logging.ERROR:
159+
effective_log_level = "ERROR"
160+
print("Effective log level set to {}".format(effective_log_level))
161+
162+
139163
def print_invalid_tool_option(requested_tool, valid_tool_options):
140164
print(
141165
"Invalid redisbench-admin <tool>. Requested tool: {}. Available tools: {}".format(

0 commit comments

Comments
 (0)