feat: configure loguru level and log file

This commit is contained in:
debugtalk
2020-03-07 16:40:26 +08:00
parent 55ba82b1a6
commit 3c9eb14409
2 changed files with 11 additions and 1 deletions

View File

@@ -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

View File

@@ -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: