mirror of
https://codeberg.org/Mo8it/AdvLabDB.git
synced 2024-11-08 21:21:06 +00:00
76 lines
2 KiB
Python
76 lines
2 KiB
Python
from os import environ, makedirs
|
|
|
|
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 dotenv import load_dotenv
|
|
|
|
from .advlabdb_independent_funs import parse_bool
|
|
|
|
|
|
def set_from_env(app, var):
|
|
app.config[var] = environ[var]
|
|
|
|
|
|
load_dotenv(".env")
|
|
|
|
app = Flask(__name__)
|
|
|
|
app.config["SECURITY_EMAIL_VALIDATOR_ARGS"] = {
|
|
"check_deliverability": parse_bool(environ["CHECK_EMAIL_DELIVERABILITY"])
|
|
}
|
|
|
|
set_from_env(app, "SERVER_NAME")
|
|
|
|
set_from_env(app, "SECRET_KEY")
|
|
set_from_env(app, "SECURITY_PASSWORD_SALT")
|
|
|
|
app.config["SQLALCHEMY_DATABASE_URI"] = f"sqlite:///../{environ['RELATIVE_DB_DIR']}/advlab.db"
|
|
makedirs(environ["RELATIVE_DB_DIR"], exist_ok=True)
|
|
|
|
app.config["SQLALCHEMY_TRACK_MODIFICATIONS"] = False
|
|
|
|
app.config["FLASK_ADMIN_FLUID_LAYOUT"] = True
|
|
|
|
db = SQLAlchemy(app)
|
|
|
|
fsqla.FsModels.set_db_info(db)
|
|
|
|
migrate = Migrate(app, db)
|
|
|
|
from . import customClasses
|
|
|
|
adminSpace = Admin(
|
|
app,
|
|
name="Admin@AdvLabDB",
|
|
url="/admin",
|
|
endpoint="adminSpace",
|
|
template_mode="bootstrap3",
|
|
index_view=customClasses.SecureAdminIndexView(name="Admin", url="/admin", endpoint="adminSpace"),
|
|
)
|
|
assistantSpace = Admin(
|
|
app,
|
|
name="Assistant@AdvLabDB",
|
|
url="/assistant",
|
|
endpoint="assistantSpace",
|
|
template_mode="bootstrap3",
|
|
index_view=customClasses.SecureAssistantIndexView(name="Assistant", url="/assistant", endpoint="assistantSpace"),
|
|
)
|
|
|
|
from . import models
|
|
|
|
# Setup Flask-Security
|
|
user_datastore = SQLAlchemyUserDatastore(db, models.User, models.Role)
|
|
Security(app, user_datastore)
|
|
|
|
try:
|
|
from . import routes, adminModelViews, assistantModelViews
|
|
except Exception as ex:
|
|
print(
|
|
"\nYou are probably initializing the database with a script. If not, then you have to worry about not being able to import in __init__.py!\n"
|
|
)
|
|
print(str(ex))
|