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

fix: fix parse yaml test implement

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

7
compose_viz/parser.py
View file

@ -13,7 +13,7 @@ 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 = YAML() yaml = YAML(typ='safe', pure=True)
yaml_data = yaml.load(f) yaml_data = yaml.load(f)
except YAML.YAMLError as exc: except YAML.YAMLError as exc:
raise YAML.YAMLError raise YAML.YAMLError
@ -33,7 +33,10 @@ class Parser:
service_image = service["image"] service_image = service["image"]
#print("image: {}".format(service_image)) #print("image: {}".format(service_image))
if service.get("networks"): if service.get("networks"):
service_networks = service["networks"] if(type(service["networks"]) is list):
service_networks = service["networks"]
else:
service_networks = list(service["networks"].keys())
#print("networks: {}".format(service_networks)) #print("networks: {}".format(service_networks))
services.append(Service( services.append(Service(
name=service_name, name=service_name,