Deseo tener 3 animales en un corral pero no quiero que se supere la cantidad de 500kg de comida a la semana, los animales son, oveja=80kg, caballo=200kg, vaca=225kg, pato=50kg, cerdo=150kg. El sistema deberá mostrar un mensaje indicando al usuario si se cumple o no la restricción

def calc_food(animals):
    food=0
    for animal in animals:
        if animal == "oveja":
            food += 80
        elif animal == "caballo":
            food += 200
        elif animal == "vaca":
            food += 225
        elif animal == "pato":
            food += 50
        else:
            food += 150
    return food

animals = ["oveja", "pato"]
food = calc_food(animals)

if food > 500:
    print("No se puede comprar mas animales")
else:
    print("Esta todo bien")