python inline if then