From 3c9eb14409f7fbed27ef409f3f7e7a089444187f Mon Sep 17 00:00:00 2001
From: debugtalk <mail@debugtalk.com>
Date: Sat, 7 Mar 2020 16:40:26 +0800
Subject: [PATCH] feat: configure loguru level and log file

---
 httprunner/api.py             | 8 ++++++++
 httprunner/ext/locusts/cli.py | 4 +++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/httprunner/api.py b/httprunner/api.py
index 517fee84..6a04ecce 100644
--- a/httprunner/api.py
+++ b/httprunner/api.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import unittest
 
 from loguru import logger
@@ -38,6 +39,13 @@ class HttpRunner(object):
             "failfast": failfast,
             "resultclass": report.HtmlTestResult
         }
+
+        logger.remove()
+        log_level = log_level.upper()
+        logger.add(sys.stdout, level=log_level)
+        if log_file:
+            logger.add(log_file, level=log_level)
+
         self.unittest_runner = unittest.TextTestRunner(**kwargs)
         self.test_loader = unittest.TestLoader()
         self.save_tests = save_tests
diff --git a/httprunner/ext/locusts/cli.py b/httprunner/ext/locusts/cli.py
index 616486f8..59b84d84 100644
--- a/httprunner/ext/locusts/cli.py
+++ b/httprunner/ext/locusts/cli.py
@@ -127,11 +127,13 @@ def main():
     loglevel_index = get_arg_index("-L", "--loglevel")
     if loglevel_index and loglevel_index < len(sys.argv):
         loglevel = sys.argv[loglevel_index]
+        loglevel = loglevel.upper()
     else:
         # default
         loglevel = "WARNING"
 
-    # logger.setup_logger(loglevel)
+    logger.remove()
+    logger.add(sys.stdout, level=loglevel)
 
     # get testcase file path
     try: