From 967cfeeb8faf2af3d2dbcac75e1b6e4bed1f0696 Mon Sep 17 00:00:00 2001 From: debugtalk Date: Tue, 20 Feb 2018 21:48:50 +0800 Subject: [PATCH] check user input value for --log-level --- httprunner/cli.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/httprunner/cli.py b/httprunner/cli.py index c86bad69..3e0d023f 100644 --- a/httprunner/cli.py +++ b/httprunner/cli.py @@ -65,7 +65,9 @@ def main_hrun(): print("PyUnitReport version: {}".format(pyu_version)) exit(0) - log_level = getattr(logging, args.log_level.upper()) + log_level = getattr(logging, args.log_level.upper(), None) + if not log_level: + raise ValueError('Invalid log level: %s' % args.log_level) logging.basicConfig(level=log_level) if log_level >= 20: