from os import environ

from flask import Flask
from flask_admin import Admin
from flask_security import Security, SQLAlchemyUserDatastore
from flask_security.models import fsqla_v2 as fsqla
from flask_sqlalchemy import SQLAlchemy
from flask_migrate import Migrate

from .config import set_config

app = Flask(__name__)

# Config
settings = set_config(app)

# Setup Flask-SQLAlchemy
db = SQLAlchemy(app)
fsqla.FsModels.set_db_info(db)

# Setup Flask-Migrate
migrate = Migrate(app, db)

# Setup Flask-Admin
from .custom_classes import SecureAdminIndexView, SecureAssistantIndexView

adminSpace = Admin(
    app,
    name="Admin@AdvLabDB",
    url="/admin",
    template_mode="bootstrap3",
    index_view=SecureAdminIndexView(name="Home", url="/admin", endpoint="admin"),
)
assistantSpace = Admin(
    app,
    name="Assistant@AdvLabDB",
    url="/assistant",
    template_mode="bootstrap3",
    index_view=SecureAssistantIndexView(name="Home", url="/assistant", endpoint="assistant"),
)

from . import models

user_datastore = SQLAlchemyUserDatastore(db, models.User, models.Role)
Security(app, user_datastore)

from . import routes, adminModelViews, assistantModelViews