python alternative to if elif