def addScript():
addScriptForm = forms.AddScriptForm(prefix="add-script")
if addScriptForm.validate_on_submit():
newscript = AgentScript(name=addScriptForm.scriptName.data)
db.session.add(newscript)
db.session.commit()
current_app.agentScripts = AgentScript.query.all()
current_app.agentScriptStr = AgentScript.getScriptsString(current_app.agentScripts)
flash("%s successfully added to scripts" % newscript.name, "success")
return redirect(request.referrer)
else:
flash("%s couldn't be added to scripts" % addScriptForm.scriptName.data, "danger")
return redirect(request.referrer)
def add_script():
addScriptForm = forms.AddScriptForm(prefix="add-script")
if addScriptForm.validate_on_submit():
newscript = AgentScript(name=addScriptForm.scriptName.data)
db.session.add(newscript)
db.session.commit()
current_app.agent_scripts = AgentScript.get_scripts_string()
flash(f"{newscript.name} successfully added to scripts", "success")
else:
flash(f"{addScriptForm.scriptName.data} couldn't be added to scripts",
"danger")
return redirect(url_for("admin.agent_config"))
def agentConfig(): agentConfig = AgentConfig.query.get(1) # pass the model to the form to populate agentForm = forms.AgentConfigForm(obj=agentConfig) addScriptForm = forms.AddScriptForm(prefix="add-script") delScriptForm = forms.DeleteForm(prefix="del-script") if agentForm.validate_on_submit(): # populate the object from the form data agentForm.populate_obj(agentConfig) db.session.commit() current_app.agentConfig = agentConfig.as_dict() return render_template( 'admin/agents.html', agentForm=agentForm, scripts=current_app.agentScripts, addScriptForm=addScriptForm, delScriptForm=delScriptForm)
def agent_config():
agentConfig = AgentConfig.query.get(1)
agent_scripts = AgentScript.query.all()
# pass the model to the form to populate
agentForm = forms.AgentConfigForm(obj=agentConfig)
addScriptForm = forms.AddScriptForm(prefix="add-script")
delScriptForm = forms.DeleteForm(prefix="del-script")
if agentForm.validate_on_submit():
# populate the object from the form data
agentForm.populate_obj(agentConfig)
db.session.commit()
current_app.agentConfig = agentConfig.as_dict()
flash("Successfully updated agent configuration.", "success")
return render_template(
"admin/agents.html",
agentForm=agentForm,
scripts=agent_scripts,
addScriptForm=addScriptForm,
delScriptForm=delScriptForm,
)
