python inline if else return