feat: coordinates of label on edges are persisted; reload label posit…

…ion from previous version of diagram
tsypuk · Jul 24, 2023 · a630f2c · a630f2c
1 parent d697f7c
commit a630f2c
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/multicloud_diagrams/__init__.py b/multicloud_diagrams/__init__.py
@@ -264,6 +264,12 @@ def add_connection(self, src_node_id, dest_node_id, start, end, labels=[]):
                     # <mxPoint as="offset"/>
                     mx_geometry = et.SubElement(mx_geometry, 'mxPoint')
                     mx_geometry.set('as', 'offset')
+                    if f'label:{src_node_id}:to:{dest_node_id}' in self.prev_coords:
+                        if 'x' in self.prev_coords[f'label:{src_node_id}:to:{dest_node_id}']:
+                            mx_geometry.set('x', self.prev_coords[f'label:{src_node_id}:to:{dest_node_id}']['x'])
+                        if 'y' in self.prev_coords[f'label:{src_node_id}:to:{dest_node_id}']:
+                            mx_geometry.set('y', self.prev_coords[f'label:{src_node_id}:to:{dest_node_id}']['y'])
+
                     # </mxGeometry>
                     # </mxCell>
             else:
@@ -338,6 +344,16 @@ def read_coords_from_file(self, file_name: str):
                         cords['width'] = data.get('width')
                     self.prev_coords[neighbor.get('id')] = cords
 
+                elif neighbor.get('id').startswith("label:"):
+                    data = neighbor.find('mxGeometry')
+                    mx_point = data.find('mxPoint')
+                    cords = {}
+                    if mx_point.get('x') is not None:
+                        cords['x'] = mx_point.get('x')
+                    if mx_point.get('y') is not None:
+                        cords['y'] = mx_point.get('y')
+                    self.prev_coords[neighbor.get('id')] = cords
+
     def export_to_file(self, file_path):
         with open(file_path, 'wb') as file:
             tree = et.ElementTree(self.mxfile)