feat: added style for label on edge

tsypuk · Sep 27, 2023 · c987f3d · c987f3d
1 parent 5a5736c
commit c987f3d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 9 deletions.
diff --git a/multicloud_diagrams/__init__.py b/multicloud_diagrams/__init__.py
@@ -432,8 +432,8 @@ def add_link(self, src_node_id, dst_node_id, action=None, layer_name=None, layer
         }
         self.add_connection(src_node_id=src_node_id, dst_node_id=dst_node_id, labels=action, edge_style=style, layer_name=layer_name, layer_id=layer_id)
 
-    def add_link_uml(self, src_node_id, dst_node_id, action=None, layer_name=None, layer_id=None, style=None):
-        self.add_connection(src_node_id=src_node_id, dst_node_id=dst_node_id, labels=action, edge_style=style, layer_name=layer_name, layer_id=layer_id, prefix=layer_name)
+    def add_link_uml(self, src_node_id, dst_node_id, action=None, layer_name=None, layer_id=None, edge_style=None, label_style=None):
+        self.add_connection(src_node_id=src_node_id, dst_node_id=dst_node_id, labels=action, edge_style=edge_style, label_style=label_style, layer_name=layer_name, layer_id=layer_id, prefix=layer_name)
 
     def add_bidirectional_link(self, src_node_id, dst_node_id, action=None):
         style = {
@@ -552,7 +552,7 @@ def export_to_file(self, file_path):
         with open(file_path, 'w', encoding="utf-8") as file:
             file.write(resulting_xml)
 
-    def read_uml_from_file(self, file_name, style=None):
+    def read_uml_from_file(self, file_name, edge_style=None, label_style=None):
         with open(file_name, 'r') as file:
             sequence_diagram = file.read()
 
@@ -565,9 +565,9 @@ def read_uml_from_file(self, file_name, style=None):
         # create Layer with UML file name
         base_name = os.path.splitext(os.path.basename(file_name))[0]
         self.add_layer(base_name)
-        self.extract_messages_from_uml(sequence_diagram, actors, layer_name=base_name, style=style)
+        self.extract_messages_from_uml(sequence_diagram, actors, layer_name=base_name, edge_style=edge_style, label_style=label_style)
 
-    def extract_messages_from_uml(self, sequence_diagram, actors, layer_name, style):
+    def extract_messages_from_uml(self, sequence_diagram, actors, layer_name, edge_style, label_style):
         lines = sequence_diagram.split('\n')
         action_id = 0
         for line in lines:
@@ -583,7 +583,7 @@ def extract_messages_from_uml(self, sequence_diagram, actors, layer_name, style)
                         self.actors_to_nodes[data[1]],
                         action=[f'{action_id}: {data[2]}'],
                         layer_name=layer_name,
-                        style=style)
+                        edge_style=edge_style, label_style=label_style)
                 except KeyError:
                     print('No such node')
 

diff --git a/samples/samples/aws_mermaid_uml.py b/samples/samples/aws_mermaid_uml.py
@@ -13,6 +13,12 @@ def main():
     prepare_end2end(mcd)
 
     mcd.read_uml_mappings('uml_mapping.yml')
+    label_style = {
+        'labelBackgroundColor': 'none',
+        'fontColor': '#FF3333',
+        'fontStyle': 0,
+        'fontSize': 14
+    }
     style1 = {
         'orthogonalLoop': '1',
         'edgeStyle': 'orthogonalEdgeStyle',
@@ -24,7 +30,7 @@ def main():
         'strokeWidth': '3',
         'fontSize': '22'
     }
-    mcd.read_uml_from_file('file_upload.mermaid', style=style1)
+    mcd.read_uml_from_file('file_upload.mermaid', edge_style=style1, label_style=label_style)
 
     style2 = {
         'orthogonalLoop': '1',
@@ -33,12 +39,18 @@ def main():
         'startArrow': 'oval',
         'endArrow': 'classicThin',
         'dashed': '1',
-        'strokeColor': '#0000FF',
+        'strokeColor': '#FF0000',
         'strokeWidth': '3',
         'fontSize': '22'
     }
 
-    mcd.read_uml_from_file('process.mermaid', style=style2)
+    label_style = {
+        'labelBackgroundColor': 'none',
+        'fontColor': '#FF0000',
+        'fontStyle': 0,
+        'fontSize': 14
+    }
+    mcd.read_uml_from_file('process.mermaid', edge_style=style2, label_style=label_style)
 
     mcd.export_to_file(result_file)