python inline if then else