Refactor: scheduler: functionize updating an action's optional flag

... to make custom_action() easier to follow. This also allows us to avoid
unnecessarily clearing the flag then immediately setting it.