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