def add_node(): """Add node dialog. :return: response """ form = AddEditNodeForm() if form.validate_on_submit(): # print form.device_type.data # print form.in_use.data to_add = Nodes(name=form.name.data, ip=form.ip.data, interface=form.interface.data, node_type=form.node_type.data if form.node_type.data != "None" else None, in_use=form.in_use.data) try: db.session.add(to_add) db.session.commit() flash("Successfully added {0}".format(str(to_add))) except sqlalchemy.exc.IntegrityError as e: db.session.rollback() print e flash("Not adding nodes - %s" % str(e.message)) return redirect("/manage_nodes") return render_template("add_edit_node.html", form=form, add_edit="add node", page_loc="node - add")
def edit_node(node_id): """Node information can be edited in this view. :param node_id: id of the node to be edited :return: response """ try: to_edit = db.session.query(Nodes).filter(Nodes.id == int(node_id)).first() except IndexError: flash("Edit not successful!") return redirect("/manage_nodes") form = AddEditNodeForm() if form.validate_on_submit(): print "Edit form Valid" to_edit.name = form.name.data to_edit.ip = form.ip.data to_edit.interface = form.interface.data to_edit.node_type = form.node_type.data if form.node_type.data != "None" else None to_edit.in_use = form.in_use.data try: db.session.commit() flash("Successfully edited {0}".format(str(to_edit))) except sqlalchemy.exc.IntegrityError as e: db.session.rollback() print e flash("Not editing nodes - %s" % str(e.message)) return redirect("/manage_nodes") form.name.data = to_edit.name form.ip.data = to_edit.ip form.interface.data = to_edit.interface form.node_type.data = to_edit.node_type form.in_use.data = to_edit.in_use form.node_id = to_edit.id return render_template("add_edit_node.html", form=form, add_edit="edit node", page_loc="nodes - edit")