Validate a state string

def _validate_state(state, valid_states):
    """Validate a state string"""
    if state in State:
        return state.name
    elif state in valid_states:
        return state
    else:
        raise Invalid('Invalid state')