33 lines
746 B
Python
33 lines
746 B
Python
from flask import Flask
|
|
from flask_migrate import Migrate
|
|
from flask_login import LoginManager
|
|
from .models import db, User
|
|
from .auth.oauth2 import config_oauth
|
|
|
|
migrate = Migrate()
|
|
login_manager = LoginManager()
|
|
login_manager.login_view = 'auth.login'
|
|
|
|
|
|
def create_app(cfg):
|
|
app = Flask(__name__)
|
|
app.config.from_object(cfg)
|
|
setup_app(app)
|
|
return app
|
|
|
|
|
|
def setup_app(app):
|
|
db.init_app(app)
|
|
config_oauth(app)
|
|
migrate.init_app(app, db)
|
|
login_manager.init_app(app)
|
|
from .auth import auth
|
|
from .home import home
|
|
app.register_blueprint(auth, url_prefix='')
|
|
app.register_blueprint(home, url_prefix='')
|
|
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return db.session.query(User).get(user_id)
|