That's mostly what `caffeinate your-script` is for.