22 lines
631 B
Python
22 lines
631 B
Python
|
from .data import load
|
||
|
|
from ..models import User
|
||
|
|
|
||
|
|
from flask import abort, redirect
|
||
|
|
from flask.helpers import 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: return redirect(url_for('views._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
|