Selaa lähdekoodia

Show additional data nodes in DAG (#1771)

* Show additional data nodes in DAG
resolves #787

* E501

* E501

---------

Co-authored-by: Fred Lefévère-Laoide <Fred.Lefevere-Laoide@Taipy.io>
Fred Lefévère-Laoide 8 kuukautta sitten
vanhempi
säilyke
aae5778e16
1 muutettua tiedostoa jossa 16 lisäystä ja 4 poistoa
  1. 16 4
      taipy/gui_core/_adapters.py

+ 16 - 4
taipy/gui_core/_adapters.py

@@ -131,16 +131,28 @@ class _GuiCoreScenarioDagAdapter(_TaipyBase):
                 if scenario := core_get(data.id):
                     dag = scenario._get_dag()
                     nodes = {}
-                    for id, node in dag.nodes.items():
-                        entityType = _GuiCoreScenarioDagAdapter.get_entity_type(node)
+                    for id, dag_node in dag.nodes.items():
+                        entityType = _GuiCoreScenarioDagAdapter.get_entity_type(dag_node)
                         cat = nodes.get(entityType)
                         if cat is None:
                             cat = {}
                             nodes[entityType] = cat
                         cat[id] = {
-                            "name": node.entity.get_simple_label(),
-                            "type": node.entity.storage_type() if hasattr(node.entity, "storage_type") else None,
+                            "name": dag_node.entity.get_simple_label(),
+                            "type": dag_node.entity.storage_type()
+                            if hasattr(dag_node.entity, "storage_type")
+                            else None,
                         }
+                    cat = nodes.get(DataNode.__name__)
+                    if cat is None:
+                        cat = {}
+                        nodes[DataNode.__name__] = cat
+                    for id, data_node in scenario.additional_data_nodes.items():
+                        cat[id] = {
+                            "name": data_node.get_simple_label(),
+                            "type": data_node.storage_type(),
+                        }
+
                     return [
                         data.id,
                         nodes,