Make Save module try to inspect visual_name attribute of visuals

klarh · Apr 2, 2022 · 2370a73 · 2370a73
1 parent 2b06362
commit 2370a73
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flowws_analysis/Save.py b/flowws_analysis/Save.py
@@ -43,6 +43,8 @@ def run(self, scope, storage):
                 modifiers = []
                 if isinstance(vis, flowws.Stage):
                     modifiers.append(type(vis).__name__)
+                elif hasattr(vis, 'visual_name'):
+                    modifiers.append(vis.visual_name)
                 modifiers.extend(self.arguments.get('file_modifiers', []))
                 self._update_modifiers(filename, modifiers)
 
@@ -75,6 +77,8 @@ def run(self, scope, storage):
                 modifiers = []
                 if isinstance(vis, flowws.Stage):
                     modifiers.append(type(vis).__name__)
+                elif hasattr(vis, 'visual_name'):
+                    modifiers.append(vis.visual_name)
                 modifiers.extend(self.arguments.get('file_modifiers', []))
                 self._update_modifiers(filename, modifiers)