extras

aiidaplugins · Mar 4, 2024 · 15c5a6f · 15c5a6f
1 parent 5c9fa99
commit 15c5a6f
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/aiida_flexpart/calculations/flexpart_post.py b/aiida_flexpart/calculations/flexpart_post.py
@@ -36,7 +36,7 @@ def define(cls, spec):
     def prepare_for_submission(self, folder):
 
         params  = ['-m',self.inputs.input_dir.get_remote_path(),
-                   '-r','./'
+                   '-r','./','-p'
                   ]
         if 'input_offline_dir' in self.inputs:
             params += ['-n',self.inputs.input_offline_dir.get_remote_path()]
@@ -50,6 +50,6 @@ def prepare_for_submission(self, folder):
         # Prepare a `CalcInfo` to be returned to the engine
         calcinfo = common.CalcInfo()
         calcinfo.codes_info = [codeinfo]
-        calcinfo.retrieve_list = ['grid_time_*.nc', 'boundary_sensitivity_*.nc', 'aiida.out']
+        calcinfo.retrieve_list = ['grid_time_*.nc', 'boundary_sensitivity_*.nc','*.png', 'aiida.out']
 
         return calcinfo
diff --git a/aiida_flexpart/workflows/multi_dates_workflow.py b/aiida_flexpart/workflows/multi_dates_workflow.py
@@ -28,6 +28,9 @@ def define(cls, spec):
         spec.input('check_meteo_ifs_code', valid_type=orm.AbstractCode)
         spec.input('post_processing_code', valid_type=orm.AbstractCode)
 
+        #extras
+        spec.input('name', valid_type=str, non_db=True, required=False)
+
         # Basic Inputs
         spec.input('simulation_dates',
                    valid_type=orm.List,
@@ -157,7 +160,9 @@ def setup(self):
         self.ctx.outgrid = self.inputs.outgrid
         self.ctx.species = self.inputs.species
         self.ctx.land_use = self.inputs.land_use
-        #self.base.extras.set('this',3)
+        if 'name' in self.inputs:
+            self.node.base.extras.set(self.inputs.name,
+                                      {'command': self.ctx.command.get_dict()})
 
     def prepare_meteo_folder_ifs(self):
         """prepare meteo folder"""