55 lines
1.5 KiB
Python
55 lines
1.5 KiB
Python
import os
|
|
from dotenv import load_dotenv
|
|
from flask import Flask, render_template
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_login import LoginManager
|
|
from models import db, User # Our database instance
|
|
from user import user_bp
|
|
from admin import admin_bp
|
|
from auth import auth_bp
|
|
from oauth import google_bp
|
|
from commands import create_admin
|
|
|
|
# Load environment variables from .env file
|
|
load_dotenv()
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = os.environ.get('SECRET_KEY', 'dev-key-for-testing')
|
|
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///helpdesk.db'
|
|
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
|
|
|
|
# Initialize the database with the Flask app
|
|
db.init_app(app)
|
|
|
|
# Initialize Flask-Login
|
|
login_manager = LoginManager(app)
|
|
login_manager.login_view = 'auth.login'
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return User.query.get(int(user_id))
|
|
|
|
# Register Blueprints
|
|
app.register_blueprint(user_bp)
|
|
app.register_blueprint(admin_bp, url_prefix='/admin')
|
|
app.register_blueprint(auth_bp, url_prefix='/auth')
|
|
app.register_blueprint(google_bp, url_prefix='/login')
|
|
|
|
# Create database tables if they don't exist
|
|
with app.app_context():
|
|
db.create_all()
|
|
|
|
# Add this after creating the app
|
|
app.cli.add_command(create_admin)
|
|
|
|
@app.errorhandler(404)
|
|
def page_not_found(e):
|
|
return render_template('error.html'), 404
|
|
|
|
@app.errorhandler(500)
|
|
def internal_server_error(e):
|
|
return render_template('error.html'), 500
|
|
|
|
if __name__ == '__main__':
|
|
app.run(debug=True)
|