python inline if variable assignment