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

fix: yaml FullLoader AttributeError when PyYAML version < 5.1

Browse Source
This commit is contained in:
debugtalk
2019-06-30 20:53:42 +08:00
parent f02ba41724
commit b6878bd9d9
3 changed files with 15 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
httprunner/loader.py
View File

@@ -10,6 +10,13 @@ import sys
import yaml
from httprunner import built_in, exceptions, logger, parser, utils, validator
try:
# PyYAML version >= 5.1
# ref: https://github.com/yaml/pyyaml/wiki/PyYAML-yaml.load(input)-Deprecation
yaml.warnings({'YAMLLoadWarning': False})
except AttributeError:
pass
###############################################################################
## file loader
###############################################################################
@@ -35,7 +42,7 @@ def load_yaml_file(yaml_file):
""" load yaml file and check file content format
"""
with io.open(yaml_file, 'r', encoding='utf-8') as stream:
yaml_content = yaml.load(stream, Loader=yaml.FullLoader)
yaml_content = yaml.load(stream)
_check_format(yaml_file, yaml_content)
return yaml_content