def init_app(app): # Flask-DebugToolbar debug_toolbar.init_app(app) # Celery cel.init_app(app) # CSRF protection csrf.init_app(app) # SQLAlchemy db.init_app(app) # User user_manager.init_app(app, db, User) user_manager.email_manager = CustomEmailManager(app) # Flask-Gravatar gravatar.init_app(app) # Flask-Migrate migrate.init_app( app=app, db=db, directory=str(Path(__file__).resolve().parent / "migrations"), ) # Reverse proxy should used HTTPS if app.config["USE_REVERSE_PROXY"]: app.config["PREFERRED_URL_SCHEME"] = "https" app.wsgi_app = ProxyFix(app.wsgi_app, x_proto=1, x_host=1) # API Ratelimit limiter.init_app(app) # Flask-Admin admin = Admin( name="OpenCVE Admin", template_mode="bootstrap3", index_view=HomeView() ) admin.init_app(app) with warnings.catch_warnings(): warnings.filterwarnings("ignore", "Fields missing from ruleset") admin.add_view(UserModelView(User, db.session)) admin.add_view(CveModelView(Cve, db.session)) admin.add_view(EventModelView(Event, db.session)) admin.add_view(VendorModelView(Vendor, db.session)) admin.add_view(ProductModelView(Product, db.session)) admin.add_link(MenuLink(name="Tasks", url="/admin/tasks"))
def init_app(app): # Flask-Admin admin.init_app(app) admin.add_view(UserModelView(User, db.session)) admin.add_view(CveModelView(Cve, db.session)) admin.add_view(EventModelView(Event, db.session)) admin.add_view(VendorModelView(Vendor, db.session)) admin.add_view(ProductModelView(Product, db.session)) admin.add_link(MenuLink(name="Tasks", url="/admin/tasks")) # Flask-DebugToolbar debug_toolbar.init_app(app) # Celery cel.init_app(app) # CSRF protection csrf.init_app(app) # SQLAlchemy db.init_app(app) # User user_manager.init_app(app, db, User) user_manager.email_manager = CustomEmailManager(app) # Flask-Gravatar gravatar.init_app(app) # Flask-Migrate migrate.init_app( app=app, db=db, directory=str(Path(__file__).resolve().parent / "migrations"), ) # Reverse proxy should used HTTPS if app.config["USE_REVERSE_PROXY"]: app.config["PREFERRED_URL_SCHEME"] = "https" app.wsgi_app = ProxyFix(app.wsgi_app, x_proto=1, x_host=1)