I prefer ternary operator for things like this:

    return 'DONE' if self.status in self.done else 'TODO'