示例#1
0
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"))
示例#2
0
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)
示例#3
0
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)
示例#4
0
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)
示例#5
0
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)
示例#6
0
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)
示例#7
0
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)
示例#8
0
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)
示例#9
0
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,
    )