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

fix: handle key error during parsing root service

Browse source
This commit is contained in:
Xyphuz 2023-01-10 22:37:51 +08:00
parent f2d0ce0d22
commit 168d034c05
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
compose_viz/parser.py
View file

@ -28,6 +28,8 @@ class Parser:
@staticmethod @staticmethod
def compile_dependencies(service_name: str, compose_data: spec.ComposeSpecification) -> List[str]: def compile_dependencies(service_name: str, compose_data: spec.ComposeSpecification) -> List[str]:
assert compose_data.services assert compose_data.services
assert service_name in compose_data.services, f"Service '{service_name}' not found in given compose file."
dependencies = [] dependencies = []
for dependency in Parser._unwrap_depends_on(compose_data.services[service_name].depends_on): for dependency in Parser._unwrap_depends_on(compose_data.services[service_name].depends_on):
if dependency: if dependency:
@ -47,7 +49,7 @@ class Parser:
assert compose_data.services is not None, "No services found, aborting." assert compose_data.services is not None, "No services found, aborting."
root_dependencies: List[str] = list() root_dependencies: List[str] = []
if root_service: if root_service:
root_dependencies = Parser.compile_dependencies(root_service, compose_data) root_dependencies = Parser.compile_dependencies(root_service, compose_data)
root_dependencies.append(root_service) root_dependencies.append(root_service)