python3 if else one line