Merge pull request #230 from caracal-pipeline/issue-199

Issue 199

stimela/kitchen/recipe.py

@@ -211,6 +211,9 @@ def do_assign(assignments):
             # it will be in subst.recipe if it was assigned, or is an input
             if basevar in subst.recipe:
                 value = str(subst.recipe[basevar])
+            # else it may be a for-loop index that hasn't been assigned yet -- ignore
+            elif self.for_loop is not None and basevar == self.for_loop.var:
+                continue
             # else it might be an input with a default, check for that
             elif basevar in self.inputs_outputs and self.inputs_outputs[basevar].default is not UNSET:
                 value = str(self.inputs_outputs[basevar].default)

tests/stimela_tests/test_loop_recipe.yml

@@ -73,6 +73,15 @@ cubical_image_loop:
     log:
       name: log-{recipe.loop-name}-{info.fqname}
 
+  assign_based_on:
+    loop:
+      '1':
+        z: 1
+      '2':
+        z: 2
+      DEFAULT:
+        z: 3
+
   for_loop:
     var: loop
     over: [1,2,3]