python inline if else assignment