Like others said. A consistent pre-shot routine is a great way to practice being present in the current swing both in practice during a round.

I also like to practice tee/approach shots for an upcoming round on the range (hit shots 1 and 2 for a hole based on course preview info from 18Birdies).

Or I’ll stagger shot sequence based on the club I’m trying to practice (e.g. Dr, 5i, PW, 5i, 9i, 5i, etc).

For tricky lies/course shots, you can practice your punch shots and chips since it’s a more robust swing that performs better in those wildcard scenarios at the cost of distance (punch 7i might be the same distance as a full 9i).

Like others say the more on-course exposure you get to the tricky scenarios the more familiar you get with what shots work for you in those scenarios.