51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import os
|
|
from flask import Flask
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_migrate import Migrate
|
|
from flask_bcrypt import Bcrypt
|
|
from flask_login import LoginManager
|
|
|
|
# Initialize Flask extensions
|
|
db = SQLAlchemy()
|
|
migrate = Migrate()
|
|
bcrypt = Bcrypt()
|
|
login_manager = LoginManager()
|
|
|
|
login_manager.login_view = 'auth.login'
|
|
login_manager.login_message_category = 'info'
|
|
|
|
def create_app():
|
|
app = Flask(__name__)
|
|
|
|
# Load configuration
|
|
config_name = os.getenv('FLASK_CONFIG', 'DevelopmentConfig') # Default to DevelopmentConfig
|
|
app.config.from_object(f'app.config.{config_name}')
|
|
|
|
# Initialize extensions
|
|
db.init_app(app)
|
|
migrate.init_app(app, db)
|
|
bcrypt.init_app(app)
|
|
login_manager.init_app(app)
|
|
|
|
# Import models here to avoid circular imports
|
|
from app.models import User # Import the User model
|
|
|
|
# Set up the user_loader function
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return User.query.get(int(user_id)) # Query the User model by ID
|
|
|
|
# Register blueprints
|
|
from app.routes.auth import auth
|
|
from app.routes.data import data
|
|
from app.routes.main import main
|
|
from app.routes.user import user
|
|
from app.routes.reading import reading
|
|
app.register_blueprint(main, url_prefix='/')
|
|
app.register_blueprint(auth, url_prefix='/auth')
|
|
app.register_blueprint(user, url_prefix='/user')
|
|
app.register_blueprint(data, url_prefix="/data")
|
|
app.register_blueprint(reading, url_prefix="/reading")
|
|
|
|
return app
|