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

chore: parser add some value checking

Browse source
This commit is contained in:
uccu 2022-05-22 01:57:21 +08:00
parent 473033d3f5
commit cd36de0a5f
1 changed files with 19 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
compose_viz/parser.py
View file

@ -48,14 +48,15 @@ class Parser:
if service.get("networks"): if service.get("networks"):
if type(service["networks"]) is list: if type(service["networks"]) is list:
service_networks = service["networks"] service_networks = service["networks"]
else: elif type(service["networks"]) is dict:
service_networks = list(service["networks"].keys()) service_networks = list(service["networks"].keys())
service_extends: Optional[Extends] = None service_extends: Optional[Extends] = None
if service.get("extends"): if service.get("extends"):
service_extends = Extends(service_name=service["extends"]["service"]) if service["extends"].get("service"):
service_extends = Extends(service_name=service["extends"]["service"])
service_ports: List[str] = [] service_ports: List[Port] = []
if service.get("ports"): if service.get("ports"):
if type(service["ports"]) is list: if type(service["ports"]) is list:
for port_data in service["ports"]: for port_data in service["ports"]:
@ -73,15 +74,23 @@ class Parser:
if service.get("volumes"): if service.get("volumes"):
for volume_data in service["volumes"]: for volume_data in service["volumes"]:
if type(volume_data) is dict: if type(volume_data) is dict:
volume_source = volume_data["source"] volume_source: str = None
volume_target = volume_data["target"] volume_target: str = None
volume_type = VolumeType[volume_data["type"]] volume_type: VolumeType.volume = None
service_volumes.append(Volume(source=volume_source, target=volume_target, type=volume_type)) if volume_data.get("source"):
volume_source = volume_data["source"]
if volume_data.get("target"):
volume_target = volume_data["target"]
if volume_data.get("type"):
volume_type = VolumeType[volume_data["type"]]
service_volumes.append(Volume(source=volume_source,
target=volume_target,
type=volume_type))
elif type(volume_data) is str: elif type(volume_data) is str:
if ':' not in volume_data:
raise RuntimeError("Invalid volume input, aborting.")
spilt_data = volume_data.split(":", 1) spilt_data = volume_data.split(":", 1)
volume_source = spilt_data[0] service_volumes.append(Volume(source=spilt_data[0], target=spilt_data[1]))
volume_target = spilt_data[1]
service_volumes.append(Volume(source=volume_source, target=volume_target))
service_links: List[str] = [] service_links: List[str] = []
if service.get("links"): if service.get("links"):