`caffeinate -s <script>` should do the trick. You don't need anything else.

`caffeinate -d` will disable the shutting down of display.

`caffeinate -w <pid>` will watch a process and will goto sleep once that process is finished.