24 lines
712 B
Python
24 lines
712 B
Python
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 |