05/31 Phase 3

app/__init__.py

@@ -8,34 +8,32 @@ from app.utils.formatters import format_currency, format_percent, format_large_n
 def create_app(config_name=None):
     if config_name is None:
         config_name = os.environ.get('FLASK_ENV', 'development')
-        if config_name == 'production':
-            config_name = 'production'
-        else:
-            config_name = 'development'
+        config_name = 'production' if config_name == 'production' else 'development'
 
     app = Flask(__name__)
     app.config.from_object(config[config_name])
 
-    # Init extensions
     db.init_app(app)
     login_manager.init_app(app)
     migrate.init_app(app, db)
     csrf.init_app(app)
 
-    # Register blueprints
     from app.routes.auth import auth_bp
     from app.routes.dashboard import dashboard_bp
     from app.routes.accounts import accounts_bp
     from app.routes.categories import categories_bp
     from app.routes.transactions import transactions_bp
+    from app.routes.budgets import budgets_bp
+    from app.routes.goals import goals_bp
 
     app.register_blueprint(auth_bp)
     app.register_blueprint(dashboard_bp)
     app.register_blueprint(accounts_bp)
     app.register_blueprint(categories_bp)
     app.register_blueprint(transactions_bp)
+    app.register_blueprint(budgets_bp)
+    app.register_blueprint(goals_bp)
 
-    # Import all models so Flask-Migrate can see them
     with app.app_context():
         from app.models import (
             User, Account, Category, Receipt, RecurringRule,
@@ -44,7 +42,6 @@ def create_app(config_name=None):
             AiInsight, FxRate
         )
 
-    # Jinja2 globals
     app.jinja_env.globals['format_currency'] = format_currency
     app.jinja_env.globals['format_percent'] = format_percent
     app.jinja_env.globals['format_large_number'] = format_large_number