Funciones, factura de un hotel

#leccion4 de funciones (una posibilidad)
def costo_hotel(noches):
    return 140*noches
    
def costo_del_vuelo(ciudad):
    costo=0
    if(ciudad=="Córdoba"):
        costo=821
    elif(ciudad=="Iguazú"):
        costo=941
    elif(ciudad=="Ushuaia"):
        costo=1280
    elif(ciudad=="Bariloche"):
        costo=1848
    else:
        costo="none"
    
    return costo


def alquiler_de_auto(dias):
    total=dias*338
    if (dias>=7):
        total=total-100
    elif(dias>=3):
        total-=50
    
    return(total)
def costo_viaje(ciudad,dias,otros_gastos):
    return(alquiler_de_auto(dias)+costo_hotel(dias)+costo_del_vuelo(ciudad)+otros_gastos)

print (costo_viaje("Iguazú",5,600))  #imprime el total