Python If One Liner