Finished admin console
This commit is contained in:
@@ -2,7 +2,7 @@ from .data import load
|
||||
from ..models import User
|
||||
|
||||
from flask import abort, redirect
|
||||
from flask.helpers import url_for
|
||||
from flask.helpers import flash, url_for
|
||||
from flask_login import current_user
|
||||
|
||||
from functools import wraps
|
||||
@@ -10,7 +10,9 @@ 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'))
|
||||
if User.query.count() == 0:
|
||||
flash('Please register a user account.', 'alert')
|
||||
return redirect(url_for('admin._register'))
|
||||
return function(*args, **kwargs)
|
||||
return wrapper
|
||||
|
||||
|
Reference in New Issue
Block a user