batman (working version kinda)

app/__init__.py

@@ -0,0 +1,154 @@
+from flask import Flask, g, redirect, url_for, render_template
+import datetime
+import os
+
+def create_app(config_name='development'):
+    app = Flask(__name__, 
+                template_folder='templates',
+                static_folder='static')
+    
+    # Import config
+    try:
+        from config.settings import config
+        app.config.from_object(config.get(config_name, 'default'))
+    except ImportError:
+        # Fallback configuration
+        app.config['SECRET_KEY'] = 'dev-key-placeholder'
+        app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///app.db'
+        app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
+    
+    # Create the app instance folder if it doesn't exist
+    # This is where SQLite database will be stored
+    os.makedirs(os.path.join(app.instance_path), exist_ok=True)
+    
+    # Initialize extensions
+    from app.core.extensions import db, bcrypt, limiter, login_manager, csrf
+    db.init_app(app)
+    bcrypt.init_app(app)
+    limiter.init_app(app)
+    csrf.init_app(app)
+    
+    # Initialize login manager
+    from app.core.auth import User
+    login_manager.init_app(app)
+    
+    @login_manager.user_loader
+    def load_user(user_id):
+        # Make sure we're in app context
+        with app.app_context():
+            return User.query.get(int(user_id))
+    
+    # Initialize CSRF protection
+    from flask_wtf.csrf import CSRFProtect
+    csrf = CSRFProtect()
+    csrf.init_app(app)
+    
+    # Request hooks
+    @app.before_request
+    def before_request():
+        g.user = None
+        from flask_login import current_user
+        if current_user.is_authenticated:
+            g.user = current_user
+        
+        # Add datetime to all templates
+        g.now = datetime.datetime.utcnow()
+    
+    @app.context_processor
+    def inject_now():
+        return {'now': datetime.datetime.utcnow()}
+    
+    @app.after_request
+    def add_security_headers(response):
+        # Security headers
+        response.headers['X-Content-Type-Options'] = 'nosniff'
+        response.headers['X-Frame-Options'] = 'SAMEORIGIN'
+        response.headers['X-XSS-Protection'] = '1; mode=block'
+        
+        # Update last_seen for the user
+        if hasattr(g, 'user') and g.user and g.user.is_authenticated:
+            g.user.last_seen = datetime.datetime.utcnow()
+            db.session.commit()
+        return response
+    
+    # Add a basic index route that redirects to login or dashboard
+    @app.route('/')
+    def index():
+        from flask_login import current_user
+        if current_user.is_authenticated:
+            return redirect(url_for('dashboard.dashboard_home'))
+        return redirect(url_for('auth.login'))
+    
+    # Register blueprints - order matters!
+    # First auth blueprint
+    from app.routes.auth import bp as auth_bp
+    app.register_blueprint(auth_bp)
+    print("Registered Auth blueprint")
+    
+    # Then other blueprints
+    try:
+        from app.routes.dashboard import bp as dashboard_bp
+        app.register_blueprint(dashboard_bp)
+        print("Registered Dashboard blueprint")
+    except ImportError as e:
+        print(f"Could not import dashboard blueprint: {e}")
+    
+    try:
+        from app.routes.ipam import bp as ipam_bp
+        app.register_blueprint(ipam_bp)
+        print("Registered IPAM blueprint")
+    except ImportError as e:
+        print(f"Could not import ipam blueprint: {e}")
+    
+    try:
+        from app.routes.api import bp as api_bp
+        app.register_blueprint(api_bp)
+        print("Registered API blueprint")
+    except ImportError as e:
+        print(f"Could not import API blueprint: {e}")
+    
+    # Create database tables
+    with app.app_context():
+        try:
+            db.create_all()
+            print("Database tables created successfully")
+            
+            # Check if we need to seed the database
+            from app.core.auth import User
+            if User.query.count() == 0:
+                # Run the seed database function if we have no users
+                try:
+                    from app.scripts.db_seed import seed_database
+                    seed_database()
+                    print("Database seeded with initial data")
+                    
+                    # Create an admin user
+                    admin = User(email="admin@example.com", is_admin=True)
+                    admin.set_password("admin")
+                    db.session.add(admin)
+                    db.session.commit()
+                    print("Admin user created: admin@example.com / admin")
+                except Exception as e:
+                    print(f"Error seeding database: {e}")
+        except Exception as e:
+            print(f"Error with database setup: {e}")
+    
+    # After all blueprint registrations, add error handlers
+
+    @app.errorhandler(404)
+    def page_not_found(e):
+        return render_template('errors/404.html', title='Page Not Found'), 404
+
+    @app.errorhandler(500)
+    def internal_server_error(e):
+        return render_template('errors/500.html', title='Server Error'), 500
+
+    @app.errorhandler(403)
+    def forbidden(e):
+        return render_template('errors/403.html', title='Forbidden'), 403
+
+    @app.errorhandler(401)
+    def unauthorized(e):
+        return render_template('errors/401.html', title='Unauthorized'), 401
+    
+    return app