simono41/compose-viz
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix: use ruamel instead of yaml

Browse source
This commit is contained in:
uccu 2022-05-17 00:53:29 +08:00
parent 2f274ac002
commit d53d0a77cf
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
compose_viz/parser.py
View file

@ -1,7 +1,7 @@
from re import S from re import S
from compose_viz.compose import Compose from compose_viz.compose import Compose
from compose_viz.compose import Service from compose_viz.compose import Service
import yaml from ruamel.yaml import YAML
class Parser: class Parser:
@ -13,9 +13,10 @@ class Parser:
# load the yaml file # load the yaml file
with open(file_path, "r") as f: with open(file_path, "r") as f:
try: try:
yaml_data = yaml.safe_load(f) yaml = YAML()
except yaml.YAMLError as exc: yaml_data = yaml.load(f)
raise yaml.YAMLError except YAML.YAMLError as exc:
raise YAML.YAMLError
# validate the yaml file # validate the yaml file
if not yaml_data: if not yaml_data:
print("Error: empty yaml file") print("Error: empty yaml file")