Workflow code - CAD object iterate and set state after promotion

I am adding an In Work Prototype state to my lifecycle and creating a new promotion workflow for this action.

This is for approving and locking the CAD objects when building prototypes for changed production parts.

I want the version(s) that is promoted (let's say A.3) to remain in the In Work Prototype state so it is locked.

I then need to create a new iteration(s) (A.4) and set it back to the In Work state.

This could happen more than once with the final version (A.7) getting released through ECR/ECN.

I was thinking this could be done in a robot at the end of the promotion workflow.

Any ideas?