def node_add(org_name, env_name, grp_name):
org = g.user.organizations.filter_by(name = org_name).first()
env = org.envs.filter_by(name = env_name).first()
grp = env.groups.filter_by(name = grp_name).first()
form = NodeEditForm(org, env, grp, "")
if form.validate_on_submit():
node = Node(name = form.name.data, grp =grp, timestamp = datetime.utcnow(), ip = form.ip.data)
db.session.add(node)
db.session.commit()
flash(gettext('Your settings have been saved.'))
return redirect(url_for('org_deploy', name = org.name))
return render_template('node_edit.html', form = form)
def node_edit(org_name, env_name, grp_name, node_name):
org = g.user.organizations.filter_by(name = org_name).first()
env = org.envs.filter_by(name = env_name).first()
grp = env.groups.filter_by(name = grp_name).first()
node = grp.nodes.filter_by(name = node_name).first()
form = NodeEditForm(org, env, grp, node.name)
if form.validate_on_submit():
node.name = form.name.data
node.ip = form.ip.data
db.session.add(node)
db.session.commit()
flash(gettext('Your changes have been saved.'))
return redirect(url_for('org_deploy', name = org.name))
elif request.method != "POST":
form.name.data = node.name
form.ip.data = node.ip
return render_template('node_edit.html',
form = form)
