python if or in line