57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
import os
|
|
from flask import Flask
|
|
from .extensions import db, migrate, login_manager, csrf
|
|
from .config import Config
|
|
from flask_wtf.csrf import generate_csrf
|
|
|
|
def create_app():
|
|
app = Flask(__name__, static_folder="static", template_folder="templates")
|
|
app.config.from_object(Config)
|
|
os.makedirs(app.config.get("UPLOAD_FOLDER"), exist_ok=True)
|
|
os.makedirs(os.path.join(app.config.get("UPLOAD_FOLDER"), "identity"), exist_ok=True)
|
|
os.makedirs(os.path.join(app.config.get("UPLOAD_FOLDER"), "posts"), exist_ok=True)
|
|
os.makedirs(os.path.join(app.config.get("UPLOAD_FOLDER"), "activities"), exist_ok=True)
|
|
db.init_app(app)
|
|
migrate.init_app(app, db)
|
|
login_manager.init_app(app)
|
|
csrf.init_app(app)
|
|
with app.app_context():
|
|
db.create_all()
|
|
@app.context_processor
|
|
def inject_csrf():
|
|
return dict(csrf_token=generate_csrf())
|
|
from .blueprints.auth import bp as auth_bp
|
|
from .blueprints.users import bp as users_bp
|
|
from .blueprints.posts import bp as posts_bp
|
|
from .blueprints.comments import bp as comments_bp
|
|
from .blueprints.follows import bp as follows_bp
|
|
from .blueprints.feed import bp as feed_bp
|
|
from .blueprints.activities import bp as activities_bp
|
|
from .blueprints.admin import bp as admin_bp
|
|
from .blueprints.main import bp as main_bp
|
|
from .blueprints.setup import bp as setup_bp
|
|
app.register_blueprint(auth_bp)
|
|
app.register_blueprint(main_bp)
|
|
app.register_blueprint(setup_bp)
|
|
app.register_blueprint(users_bp)
|
|
app.register_blueprint(posts_bp)
|
|
app.register_blueprint(comments_bp)
|
|
app.register_blueprint(follows_bp)
|
|
app.register_blueprint(feed_bp)
|
|
app.register_blueprint(activities_bp)
|
|
app.register_blueprint(admin_bp)
|
|
from .cli import register_cli
|
|
register_cli(app)
|
|
@app.before_request
|
|
def ensure_admin_setup():
|
|
from .models import User
|
|
from flask import request, redirect, url_for
|
|
try:
|
|
has_admin = User.query.filter_by(role="admin").first() is not None
|
|
except Exception:
|
|
has_admin = True
|
|
if not has_admin:
|
|
if not (request.path.startswith("/setup/admin") or request.path.startswith("/static")):
|
|
return redirect(url_for("setup.admin"))
|
|
return app
|