ska-referee-test/ref-test/app/tools/auth.py

from .data import load
from ..models import User

from flask import abort, redirect
from flask.helpers import flash, url_for
from flask_login import current_user

from functools import wraps

def require_account_creation(function):
    @wraps(function)
    def wrapper(*args, **kwargs):
        if User.query.count() == 0:
            flash('Please register a user account.', 'alert')
            return redirect(url_for('admin._register'))
        return function(*args, **kwargs)
    return wrapper

def disable_if_logged_in(function):
    @wraps(function)
    def wrapper(*args, **kwargs):
        if current_user.is_authenticated: return abort(404)
        return function(*args, **kwargs)
    return wrapper