fix(cleanup): handle fromImage properly by kubernetes and builtWithinLastNHours policies

alexey-igrychev · alexey-igrychev · commit 7cffc4b86242 · 2025-03-25T16:29:12.000Z
Signed-off-by: Aleksei Igrychev &lt;aleksei.igrychev@palark.com&gt;
diff --git a/pkg/build/build_phase.go b/pkg/build/build_phase.go
@@ -1019,6 +1019,9 @@ func (phase *BuildPhase) prepareStageInstructions(ctx context.Context, img *imag
 
 		// TODO: remove this legacy logic in v3.
 		serviceLabels[imagePkg.WerfBaseImageIDLabel] = prevBuiltImage.Image.GetStageDesc().Info.ID
+	} else if img.IsBasedOnStage() {
+		baseStageImage := img.GetBaseStageImage()
+		serviceLabels[imagePkg.WerfParentStageID] = baseStageImage.Image.GetStageDesc().StageID.String()
 	}
 
 	// TODO: refactor this workaround required for the image spec stage.

Original file line number	Diff line number	Diff line change
`@@ -1019,6 +1019,9 @@ func (phase BuildPhase) prepareStageInstructions(ctx context.Context, img imag`
`1019`	`1019`
`1020`	`1020`	`// TODO: remove this legacy logic in v3.`
`1021`	`1021`	`serviceLabels[imagePkg.WerfBaseImageIDLabel] = prevBuiltImage.Image.GetStageDesc().Info.ID`
	`1022`	`+ } else if img.IsBasedOnStage() {`
	`1023`	`+ baseStageImage := img.GetBaseStageImage()`
	`1024`	`+ serviceLabels[imagePkg.WerfParentStageID] = baseStageImage.Image.GetStageDesc().StageID.String()`
`1022`	`1025`	`}`
`1023`	`1026`
`1024`	`1027`	`// TODO: refactor this workaround required for the image spec stage.`