feat: added parsing resources from YAML file; support 3options for sr…

…c/dst edge declaration; YAML definitions can be augmented with other resources

.gitignore

@@ -3,6 +3,7 @@ venv/
 
 dist/
 /__pycache__/
+tests/__pycache__/
 multicloud_diagrams/__pycache__/
 *.iml
 

multicloud_diagrams/__init__.py

@@ -2,6 +2,8 @@
 import logging
 import os.path
 
+import yaml
+
 
 class MultiCloudDiagrams:
     def __init__(self):
@@ -297,6 +299,35 @@ def add_link_list(self, links):
             self.add_link(src_node_id=link['sourceNodeID'], dst_node_id=link['destinationNodeID'])
         return
 
+    def augment_from_yaml(self, yaml_name: str):
+        with open(yaml_name, 'r') as file:
+            data = yaml.safe_load(file)
+            for vertex in data['vertices']:
+                self.add_vertex(
+                    id=vertex['arn'],
+                    node_name=vertex['name'],
+                    arn=vertex['arn'],
+                    node_type=vertex['type'],
+                    # optional attributes
+                    metadata={},
+                    # icon
+                )
+            for edge in data['edges']:
+                self.add_link(
+                    src_node_id=self._build_vertex_id(data['vertices'], edge, 'src'),
+                    dst_node_id=self._build_vertex_id(data['vertices'], edge, 'dst'),
+                    action=[edge['label']]
+                )
+
+    def _build_vertex_id(self, vertex_details, edge, src_dst_marker: str):
+        resource = {}
+        if f'{src_dst_marker}_arn' in edge:
+            resource["type"] = edge[f"{src_dst_marker}_type"]
+            resource["arn"] = edge[f'{src_dst_marker}_arn']
+        elif src_dst_marker in edge:
+            resource = [vertex for vertex in vertex_details if vertex['name'] == edge[src_dst_marker]][0]
+        return f'{resource["type"]}:{resource["arn"]}'
+
     def read_coords_from_file(self, file_name: str):
         if os.path.isfile(file_name):
             tree = et.parse(file_name)

pyproject.toml

@@ -12,6 +12,7 @@ python = "^3.10"
 
 [tool.poetry.group.dev.dependencies]
 git-changelog = "^1.0.1"
+pyyaml = "^6.0"
 
 [build-system]
 requires = ["poetry-core"]