From 8aeb8d4dc2275eee0a5fc089e407e3c93e7118bd Mon Sep 17 00:00:00 2001 From: debugtalk Date: Wed, 31 Oct 2018 23:21:37 +0800 Subject: [PATCH] set logging level for locusts --- httprunner/cli.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) diff --git a/httprunner/cli.py b/httprunner/cli.py index b675ecd8..f0f1bda7 100644 --- a/httprunner/cli.py +++ b/httprunner/cli.py @@ -101,20 +101,34 @@ def main_hrun(): def main_locust(): """ Performance test with locust: parse command line options and run commands. """ - logger.setup_logger("INFO") - try: from httprunner import locusts except ImportError: msg = "Locust is not installed, install first and try again.\n" msg += "install command: pip install locustio" - logger.log_warning(msg) + print(msg) exit(1) sys.argv[0] = 'locust' if len(sys.argv) == 1: sys.argv.extend(["-h"]) + # set logging level + if "-L" in sys.argv: + loglevel_index = sys.argv.index('-L') + 1 + elif "--loglevel" in sys.argv: + loglevel_index = sys.argv.index('--loglevel') + 1 + else: + loglevel_index = None + + if loglevel_index and loglevel_index < len(sys.argv): + loglevel = sys.argv[loglevel_index] + else: + # default + loglevel = "INFO" + + logger.setup_logger(loglevel) + if sys.argv[1] in ["-h", "--help", "-V", "--version"]: locusts.main() sys.exit(0)