form progress

This commit is contained in:
Xevion
2019-07-04 04:43:51 -05:00
parent 92a6f8ad49
commit d969cd5009
11 changed files with 91 additions and 24 deletions

View File

@@ -1,18 +1,9 @@
from flask import abort
from flask_login import UserMixin
from datetime import datetime
from app import db, login
from werkzeug.security import generate_password_hash, check_password_hash
def require_role(roles=["User"]):
def wrap(func):
@wraps(func)
def decorated_view(*args, **kwargs):
if current_user.is_authenticated:
if current_user.has_roles(roles):
return func(*args, **kwargs)
return abort(401)
return decorated_view
return wrap
@login.user_loader
class User(UserMixin, db.Model):
id = db.Column(db.Integer, primary_key=True)