For these sorts of if statements I sometimes use ternary operator:

    return 'DONE' if self.status in self.done_statuses else 'TODO'