Exemplo n.º 1
0
def create_animal(project_id):
    """Adds a new animal to a project"""
    project = ProjectService().get(project_id)
    form = AnimalService().get_form(project)

    if form.validate_on_submit():
        gs = GroupService()
        group = gs.get(form.group.data)
        animal = AnimalService().new(number=form.number.data, age=form.age.data, gender=form.gender.data)
        gs.add_animal(group, animal)
        flash("Successfully added a new animal '%s'" % animal, "success")
        return redirect(url_for("view_project", project_id=project.id))

    if request.method == "POST":
        flash("Unable to add a new animal", "danger")

    return CreateAnimal(Project).render(form)