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")
