accumulations

Author: b8raoult

ecml_tools/create/functions/steps/rename.py

@@ -7,6 +7,8 @@
 # nor does it submit to any jurisdiction.
 #
 
+import re
+
 from climetlab.indexing.fieldset import FieldArray
 
 
@@ -30,20 +32,22 @@ class RenamedFieldFormat:
     def __init__(self, field, format):
         self.field = field
         self.format = format
+        self.bits = re.findall(r"{(\w+)}", format)
 
     def metadata(self, key):
         value = self.field.metadata(key)
-        # print('✅✅✅✅✅✅✅✅✅ ====> ', key, value, self.format)
         if "{" + key + "}" in self.format:
-            return self.format.format(**{key: value})
+            bits = {b: self.field.metadata(b) for b in self.bits}
+            return self.format.format(**bits)
         return value
 
     def __getattr__(self, name):
         return getattr(self.field, name)
 
 
 def execute(context, input, what="param", **kwargs):
-    if "{" in what:
-        return FieldArray([RenamedFieldFormat(fs, what) for fs in input])
+    # print('🍍🍍🍍🍍🍍🍍🍍🍍🍍🍍🍍🍍🍍 ==========', kwargs)
+    if what in kwargs:
+        return FieldArray([RenamedFieldFormat(fs, kwargs[what]) for fs in input])
 
     return FieldArray([RenamedFieldMapping(fs, what, kwargs) for fs in input])

ecml_tools/create/input.py

@@ -852,7 +852,7 @@ def step_factory(config, context, action_path, previous_step):
             raise ValueError(f"Unknown step {key}")
         cls = FunctionStepAction
         args = [key] + args
-        print("========", args)
+        # print("========", args)
 
     return cls(context, action_path, previous_step, *args, **kwargs)
 

ecml_tools/create/loaders.py

@@ -366,7 +366,7 @@ def load(self):
                 LOG.info(f" -> Skipping {igroup} total={len(self.groups)} (already done)")
                 continue
             self.print(f" -> Processing {igroup} total={len(self.groups)}")
-            print("========", group)
+            # print("========", group)
             assert isinstance(group[0], datetime.datetime), group
 
             result = self.input.select(dates=group)