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

fix #185: log hrun logs to specified file

Browse Source
This commit is contained in:
httprunner
2018-04-20 15:14:57 +08:00
parent a430e702d5
commit fafbf1caa0
2 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
httprunner/cli.py
View File

@@ -37,6 +37,9 @@ def main_hrun():
parser.add_argument( parser.add_argument(
'--log-level', default='INFO', '--log-level', default='INFO',
help="Specify logging level, default is INFO.") help="Specify logging level, default is INFO.")
parser.add_argument(
'--log-file',
help="Write logs to specified file path.")
parser.add_argument( parser.add_argument(
'--dot-env-path', '--dot-env-path',
help="Specify .env file path, which is useful for keeping production credentials.") help="Specify .env file path, which is useful for keeping production credentials.")
@@ -54,7 +57,7 @@ def main_hrun():
help="Prettify JSON testset format.") help="Prettify JSON testset format.")
args = parser.parse_args() args = parser.parse_args()
logger.setup_logger(args.log_level) logger.setup_logger(args.log_level, args.log_file)
if is_py2: if is_py2:
logger.log_warning(get_python2_retire_msg()) logger.log_warning(get_python2_retire_msg())

8
httprunner/logger.py
View File

@@ -16,7 +16,7 @@ log_colors_config = {
'CRITICAL': 'red', 'CRITICAL': 'red',
} }
def setup_logger(log_level): def setup_logger(log_level, log_file=None):
"""setup root logger with ColoredFormatter.""" """setup root logger with ColoredFormatter."""
level = getattr(logging, log_level.upper(), None) level = getattr(logging, log_level.upper(), None)
if not level: if not level:
@@ -34,7 +34,11 @@ def setup_logger(log_level):
log_colors=log_colors_config log_colors=log_colors_config
) )
handler = logging.StreamHandler() if log_file:
handler = logging.FileHandler(log_file)
else:
handler = logging.StreamHandler()
handler.setFormatter(formatter) handler.setFormatter(formatter)
logging.root.addHandler(handler) logging.root.addHandler(handler)
logging.root.setLevel(level) logging.root.setLevel(level)