def delete_script(name): deleteForm = forms.DeleteForm() if deleteForm.validate_on_submit(): delScript = AgentScript.query.filter_by(name=name).first() if delScript: db.session.delete(delScript) db.session.commit() current_app.agent_scripts = AgentScript.get_scripts_string() flash(f"{name} successfully deleted.", "success") else: flash(f"{name} doesn't exist!", "danger") return redirect(url_for("admin.agent_config"))
def deleteHost(ip): delForm = forms.DeleteForm() if delForm.validate_on_submit(): deleted = current_app.elastic.delete_host(ip) if deleted > 0: flash(f"Successfully deleted {deleted - 1 if deleted > 1 else deleted} scans for {ip}", "success") return redirect(url_for('main.browse')) else: flash(f"Couldn't delete host: {ip}", "danger") else: flash("Couldn't validate form!") return redirect(request.referrer)
def deleteHost(ip): delForm = forms.DeleteForm() if delForm.validate_on_submit(): deleted = current_app.elastic.delete_host(ip) if deleted > 0: flash("Successfully deleted host %s" % ip, "success") return redirect(url_for('main.search')) else: flash("Couldn't delete host: %s" % ip, "danger") else: flash("Couldn't validate form!") return redirect(request.referrer)
def delete_scan(scan_id): delForm = forms.DeleteForm() redirectLoc = url_for("main.browse") if delForm.validate_on_submit(): deleted = current_app.elastic.delete_scan(scan_id) if deleted not in [1, 2]: flash(f"Couldn't delete scan {scan_id}", "danger") else: flash(f"Successfully deleted scan {scan_id}.", "success") else: flash("Couldn't validate form!") return redirect(redirectLoc)
def deleteScript(name): deleteForm = forms.DeleteForm() if deleteForm.validate_on_submit(): delScript = AgentScript.query.filter_by(name=name).first() if delScript: db.session.delete(delScript) db.session.commit() current_app.agentScripts = AgentScript.query.all() current_app.agentScriptStr = AgentScript.getScriptsString(current_app.agentScripts) flash("%s successfully deleted." % name, "success") else: flash("%s doesn't exist" % name, "danger") return redirect(request.referrer)
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 delete_host(ip): delForm = forms.DeleteForm() redirectLoc = url_for("main.browse") if delForm.validate_on_submit(): deleted = current_app.elastic.delete_host(ip) if deleted > 0: flash( f"Successfully deleted {deleted - 1 if deleted > 1 else deleted} scans for {ip}.", "success", ) return redirect(redirectLoc) else: flash(f"Couldn't delete host: {ip}!", "danger") else: flash("Couldn't validate form!") return redirect(redirectLoc)
def deleteScan(scan_id): delForm = forms.DeleteForm() if delForm.validate_on_submit(): deleted = current_app.elastic.delete_scan(scan_id) if deleted in [1, 2]: flash("Successfully deleted scan %s." % scan_id, "success") if request.referrer: if scan_id in request.referrer: redirectLoc = request.referrer.rsplit(scan_id)[0] else: redirectLoc = request.referrer else: redirectLoc = url_for('main.browse') return redirect(redirectLoc) else: flash("Could not delete scan %s." % scan_id, "danger") return redirect(request.referrer or url_for('main.browse')) else: flash("Couldn't validate form!") return redirect(request.referrer)
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, )