Code Review
/
csit.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
review
|
tree
raw
|
inline
| side by side
python3: fix wrk traffic profile YAML load
[csit.git]
/
resources
/
tools
/
wrk
/
wrk_traffic_profile_parser.py
diff --git
a/resources/tools/wrk/wrk_traffic_profile_parser.py
b/resources/tools/wrk/wrk_traffic_profile_parser.py
index
1994b61
..
f592103
100644
(file)
--- a/
resources/tools/wrk/wrk_traffic_profile_parser.py
+++ b/
resources/tools/wrk/wrk_traffic_profile_parser.py
@@
-20,7
+20,7
@@
See LLD for the structure of a wrk traffic profile.
from os.path import isfile
from pprint import pformat
from os.path import isfile
from pprint import pformat
-from yaml import load, YAMLError
+from yaml import
safe_
load, YAMLError
from robot.api import logger
from resources.tools.wrk.wrk_errors import WrkError
from robot.api import logger
from resources.tools.wrk.wrk_errors import WrkError
@@
-62,7
+62,7
@@
class WrkTrafficProfile:
try:
with open(self.profile_name, "r") as profile_file:
try:
with open(self.profile_name, "r") as profile_file:
- self.traffic_profile = load(profile_file)
+ self.traffic_profile =
safe_
load(profile_file)
except IOError as err:
raise WrkError(
msg=f"An error occurred while opening the file "
except IOError as err:
raise WrkError(
msg=f"An error occurred while opening the file "