python if inline return