python-gallery/onlylegs/app.py

"""
Onlylegs Gallery
This is the main app file, it loads all the other files and sets up the app
"""
import os
import logging

from flask_assets import Bundle
from flask_migrate import init as migrate_init

from flask import Flask, render_template, abort
from werkzeug.exceptions import HTTPException
from werkzeug.security import generate_password_hash

from onlylegs.extensions import db, migrate, login_manager, assets, compress, cache
from onlylegs.config import INSTANCE_DIR, MIGRATIONS_DIR
from onlylegs.models import Users
from onlylegs.views import (
    index as view_index,
    image as view_image,
    group as view_group,
    settings as view_settings,
    profile as view_profile,
)
from onlylegs import api
from onlylegs import auth as view_auth
from onlylegs import filters

app = Flask(__name__, instance_path=INSTANCE_DIR)
app.config.from_pyfile("config.py")

# DATABASE
db.init_app(app)
migrate.init_app(app, db, directory=MIGRATIONS_DIR)

# If database file doesn't exist, create it
if not os.path.exists(os.path.join(INSTANCE_DIR, "gallery.sqlite3")):
    print("Creating database")
    with app.app_context():
        db.create_all()

        register_user = Users(
            username=app.config["ADMIN_CONF"]["username"],
            email=app.config["ADMIN_CONF"]["email"],
            password=generate_password_hash("changeme!", method="sha256"),
        )
        db.session.add(register_user)
        db.session.commit()

        print(
            """
####################################################
# DEFAULT ADMIN USER GENERATED WITH GIVEN USERNAME #
# THE DEFAULT PASSWORD "changeme!" HAS BEEN USED,  #
# PLEASE UPDATE IT IN THE SETTINGS!                #
####################################################
        """
        )

# Check if migrations directory exists, if not create it
with app.app_context():
    if not os.path.exists(MIGRATIONS_DIR):
        print("Creating migrations directory")
        migrate_init(directory=MIGRATIONS_DIR)

# LOGIN MANAGER
# can also set session_protection to "strong"
# this would protect against session hijacking
login_manager.init_app(app)
login_manager.login_view = "onlylegs.index"


@login_manager.user_loader
def load_user(user_id):
    return Users.query.filter_by(alt_id=user_id).first()


@login_manager.unauthorized_handler
def unauthorized():
    error = 401
    msg = "You are not authorized to view this page!!!!"
    return render_template("error.html", error=error, msg=msg), error


# ERROR HANDLERS
@app.errorhandler(Exception)
def error_page(err):
    """
    Error handlers, if the error is not a HTTP error, return 500
    """
    if not isinstance(err, HTTPException):
        abort(500)
    return (
        render_template("error.html", error=err.code, msg=err.description),
        err.code,
    )


# ASSETS
assets.init_app(app)

scripts = Bundle(
    "js/*.js", output="gen/js.js", depends="js/*.js"
)  # filter jsmin is broken :c
styles = Bundle(
    "sass/style.sass",
    filters="libsass, cssmin",
    output="gen/styles.css",
    depends="sass/**/*.sass",
)

assets.register("scripts", scripts)
assets.register("styles", styles)

# BLUEPRINTS
app.register_blueprint(view_auth.blueprint)
app.register_blueprint(view_index.blueprint)
app.register_blueprint(view_image.blueprint)
app.register_blueprint(view_group.blueprint)
app.register_blueprint(view_profile.blueprint)
app.register_blueprint(view_settings.blueprint)

app.register_blueprint(api.blueprint)

# FILTERS
app.register_blueprint(filters.blueprint)

# CACHE AND COMPRESS
cache.init_app(app)
compress.init_app(app)

# Yupee! We got there :3
print("Done!")
logging.info("Gallery started successfully!")


if __name__ == "__main__":
    app.run()