flask shell environment
parent
303b13b0f5
commit
f5d448db24
|
@ -4,7 +4,7 @@ from flask import Flask
|
|||
from flask_login import LoginManager
|
||||
from flask_migrate import Migrate
|
||||
|
||||
from .models import db
|
||||
from . import models
|
||||
from .routes import bp
|
||||
from .ldap import LDAPUser
|
||||
from .oauth2 import config_oauth
|
||||
|
@ -18,17 +18,27 @@ def create_app():
|
|||
setup_app(app)
|
||||
return app
|
||||
|
||||
def register_shellcontext(app):
|
||||
"""Register shell context objects."""
|
||||
|
||||
def shell_context():
|
||||
"""Shell context objects."""
|
||||
return {"db": models.db, "models": models}
|
||||
|
||||
app.shell_context_processor(shell_context)
|
||||
|
||||
|
||||
def setup_app(app):
|
||||
logging.basicConfig(
|
||||
level=logging.DEBUG,
|
||||
format="[%(asctime)-15s] %(name)-10s %(levelname)7s: %(message)s",
|
||||
)
|
||||
db.init_app(app)
|
||||
migrate = Migrate(app, db)
|
||||
models.db.init_app(app)
|
||||
migrate = Migrate(app, models.db)
|
||||
login_manager = LoginManager()
|
||||
login_manager.init_app(app)
|
||||
login_manager.login_view = "/login"
|
||||
login_manager.user_loader(LDAPUser.by_login)
|
||||
config_oauth(app)
|
||||
app.register_blueprint(bp, url_prefix='')
|
||||
register_shellcontext(app)
|
||||
|
|
Loading…
Reference in New Issue