GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-12 03:09:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

check user input value for --log-level

Browse Source
This commit is contained in:
debugtalk
2018-02-20 21:48:50 +08:00
parent 00a8d1be8a
commit 967cfeeb8f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
httprunner/cli.py
View File

@@ -65,7 +65,9 @@ def main_hrun():
print("PyUnitReport version: {}".format(pyu_version)) print("PyUnitReport version: {}".format(pyu_version))
exit(0) 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) logging.basicConfig(level=log_level)
if log_level >= 20: if log_level >= 20: