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