mirror of
https://github.com/Derpy-Leggies/OnlyLegs.git
synced 2024-12-29 10:56:10 +00:00
137 lines
4.2 KiB
Python
137 lines
4.2 KiB
Python
"""
|
|
Onlylegs Gallery
|
|
This is the main app file, it loads all the other files and sets up the app
|
|
"""
|
|
import os
|
|
import logging
|
|
import platformdirs
|
|
|
|
from flask_assets import Bundle
|
|
|
|
from flask_migrate import init as migrate_init
|
|
from flask_migrate import upgrade as migrate_upgrade
|
|
from flask_migrate import migrate as migrate_migrate
|
|
|
|
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.views import index, image, group, settings, profile
|
|
from onlylegs import api
|
|
from onlylegs import auth
|
|
from onlylegs.models import User
|
|
|
|
|
|
INSTACE_DIR = os.path.join(platformdirs.user_config_dir("onlylegs"), "instance")
|
|
MIGRATIONS_DIR = os.path.join(INSTACE_DIR, "migrations")
|
|
|
|
|
|
def create_app(): # pylint: disable=R0914
|
|
"""
|
|
Create and configure the main app
|
|
"""
|
|
app = Flask(__name__, instance_path=INSTACE_DIR)
|
|
app.config.from_pyfile("config.py")
|
|
|
|
# DATABASE
|
|
db.init_app(app)
|
|
migrate.init_app(app, db)
|
|
|
|
# If database file doesn't exist, create it
|
|
if not os.path.exists(os.path.join(INSTACE_DIR, "gallery.sqlite3")):
|
|
print("Creating database")
|
|
with app.app_context():
|
|
db.create_all()
|
|
|
|
register_user = User(
|
|
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(
|
|
"""
|
|
####################################################
|
|
# DEFAULY 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)
|
|
|
|
# Check if migrations are up to date
|
|
with app.app_context():
|
|
print("Checking for schema changes...")
|
|
migrate_migrate(directory=MIGRATIONS_DIR)
|
|
migrate_upgrade(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 User.query.filter_by(alt_id=user_id).first()
|
|
|
|
@login_manager.unauthorized_handler
|
|
def unauthorized(): # skipcq: PTC-W0065
|
|
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): # skipcq: PTC-W0065
|
|
"""
|
|
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", filters="jsmin", output="gen/js.js", depends="js/*.js")
|
|
styles = Bundle(
|
|
"sass/*.sass",
|
|
filters="libsass, cssmin",
|
|
output="gen/styles.css",
|
|
depends="sass/**/*.sass",
|
|
)
|
|
|
|
assets.register("scripts", scripts)
|
|
assets.register("styles", styles)
|
|
|
|
# BLUEPRINTS
|
|
app.register_blueprint(auth.blueprint)
|
|
app.register_blueprint(api.blueprint)
|
|
app.register_blueprint(index.blueprint)
|
|
app.register_blueprint(image.blueprint)
|
|
app.register_blueprint(group.blueprint)
|
|
app.register_blueprint(profile.blueprint)
|
|
app.register_blueprint(settings.blueprint)
|
|
|
|
# CACHE AND COMPRESS
|
|
cache.init_app(app)
|
|
compress.init_app(app)
|
|
|
|
# Yupee! We got there :3
|
|
print("Done!")
|
|
logging.info("Gallery started successfully!")
|
|
return app
|