En una granja se desea tener 3 animales en un corral personal, pero el dueño quiere saber que combinación de 3 animales, puede tener, que no supere los 500Kg de comida a la semana. El sistema deberá mostrar un mensaje indicando al usuario si se cumple o no la restricción. Oveja=80kg Caballo=200kg Vaca=225kg Pato=50kg Cerdo=150kg

def calculo_comida(animales):
  comida = 0
  for animal in animales:
    if animal == "oveja":
      comida += 80
    elif animal == "caballo":
      comida += 200
    elif animal == "vaca":
      comida += 225
    elif animal == "pato":
      comida += 50
    elif animal == "cerdo":
      comida += 150
    else:
      comida += 0
    
  if comida > 500:
    return "error, supera los 500 kg"
  else:
    return "ok"

animales = ["oveja", "pato", "caballo"]
calculo_comida(animales)