website/app/flaskr.py

"""
Simple flask thing
"""

from flask import redirect, flash, render_template

from app import app
from app.modules.static import Static
from app.modules.backend import Backend
from app.modules.auth import Auth
from app.modules.vote import Vote
from app.modules.backend.modules.page import Backend_Page
from app.modules.backend.modules.file import Backend_File
from app.modules.backend.modules.user import Backend_User
from app.modules.backend.modules.vote import Backend_Vote

app.register_blueprint(Auth)
app.register_blueprint(Static)
app.register_blueprint(Backend, url_prefix='/backend')
app.register_blueprint(Backend_Page, url_prefix='/backend/page')
app.register_blueprint(Backend_File, url_prefix='/backend/file')
app.register_blueprint(Backend_User, url_prefix='/backend/user')
app.register_blueprint(Backend_Vote, url_prefix='/backend/vote')
app.register_blueprint(Vote, url_prefix='/vote')


@app.errorhandler(404)
def page_not_found(error):
    return render_template("error/404.html")


@app.errorhandler(401)
def login_required(error):
    flash('Please login before opening page', 'warning')
    return redirect('/login', code=302)