print python 2 retire message as warning

This commit is contained in:
debugtalk
2018-03-25 23:07:07 +08:00
parent 261169b70e
commit 01dda7b92d
3 changed files with 21 additions and 4 deletions

View File

@@ -6,10 +6,11 @@ import unittest
from httprunner import logger
from httprunner.__about__ import __version__
from httprunner.compat import is_py2
from httprunner.task import HttpRunner
from httprunner.utils import (create_scaffold, load_dot_env_file,
prettify_json_file, print_output,
validate_json_file)
from httprunner.utils import (create_scaffold, get_python2_retire_msg,
load_dot_env_file, prettify_json_file,
print_output, validate_json_file)
def main_hrun():
@@ -51,6 +52,9 @@ def main_hrun():
args = parser.parse_args()
logger.setup_logger(args.log_level)
if is_py2:
logger.log_warning(get_python2_retire_msg())
if args.version:
logger.color_print("{}".format(__version__), "GREEN")
exit(0)