view_eurostat_indicator: add object to editor and expose to shell

yvdabb · yvdabb · commit 31a31accd3bb · 2023-10-10T15:44:24.000+02:00
diff --git a/larray_editor/editor.py b/larray_editor/editor.py
@@ -280,6 +280,13 @@ def view_eurostat_indicator(self, index):
              
                 arr = freq_eurostat(selected_frequencies, arr)
 
+                # Load the data into the editor
+                editor = self.parent()
+                new_data = editor.data.copy()
+                new_data[code] = arr
+                editor.update_mapping(new_data)
+                self.parent().kernel.shell.user_ns[code] = arr
+                self.accept()
             except Exception:
                 QMessageBox.critical(self, "Error", "Failed to load {}".format(code))
             self.parent().view_expr(arr, expr=code)
