From 341c00cfb023829fda96499d78bc716a177764d4 Mon Sep 17 00:00:00 2001
From: Mo8it <mo8it@proton.me>
Date: Thu, 10 Nov 2022 13:20:32 +0100
Subject: [PATCH] Use init_app for Security

---
 advlabdb/__init__.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/advlabdb/__init__.py b/advlabdb/__init__.py
index 28afe34..ba1c163 100644
--- a/advlabdb/__init__.py
+++ b/advlabdb/__init__.py
@@ -1,5 +1,6 @@
 from flask_migrate import Migrate
 from flask_security.datastore import SQLAlchemyUserDatastore
+from flask_security.core import Security
 
 from .config import get_settings, get_data_dir
 from .models import db, User, Role
@@ -11,6 +12,8 @@ settings = get_settings(data_dir)
 
 user_datastore = SQLAlchemyUserDatastore(db, User, Role)
 
+security = Security(datastore=user_datastore)
+
 
 def create_app(create_for_server: bool = True):
     from flask import Flask
@@ -23,16 +26,13 @@ def create_app(create_for_server: bool = True):
     set_config(app, data_dir)
 
     # Setup Flask-SQLAlchemy
-
     db.init_app(app)
 
     # Setup Flask-Migrate
     migrate.init_app(app, db)
 
     # Setup Flask-Security-Too
-    from flask_security.core import Security
-
-    Security(app, user_datastore)
+    security.init_app(app)
 
     if create_for_server:
         # Setup views