43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
from flask import Flask, render_template
|
|
from string import digits, ascii_lowercase
|
|
from random import sample
|
|
from fuxi.views.authenticate import login_check, authenticate
|
|
from fuxi.views.index import index
|
|
from fuxi.views.vul_scanner import vul_scanner
|
|
from fuxi.views.asset_management import asset_management
|
|
from fuxi.views.plugin_management import plugin_management
|
|
from fuxi.views.settings import settings
|
|
from fuxi.views.dashboard import dashboard
|
|
from fuxi.views.port_scanner import port_scanner
|
|
from fuxi.views.subdomain_brute import subdomain_brute
|
|
from fuxi.views.acunetix_scanner import acunetix_scanner
|
|
from fuxi.views.auth_tester import auth_tester
|
|
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = ''.join(sample(digits + ascii_lowercase, 10))
|
|
|
|
app.register_blueprint(authenticate)
|
|
app.register_blueprint(index)
|
|
app.register_blueprint(vul_scanner)
|
|
app.register_blueprint(asset_management)
|
|
app.register_blueprint(plugin_management)
|
|
app.register_blueprint(settings)
|
|
app.register_blueprint(dashboard)
|
|
app.register_blueprint(port_scanner)
|
|
app.register_blueprint(subdomain_brute)
|
|
app.register_blueprint(acunetix_scanner)
|
|
app.register_blueprint(auth_tester)
|
|
|
|
|
|
@app.errorhandler(404)
|
|
@login_check
|
|
def page_not_fount(e):
|
|
return render_template('404.html'), 404
|
|
|
|
|
|
@app.errorhandler(500)
|
|
@login_check
|
|
def internal_server_error(e):
|
|
return render_template('500.html'), 500
|