52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# Standard library imports
|
|
from functools import wraps
|
|
|
|
# Third party imports
|
|
from flask import g, flash, redirect, url_for
|
|
from flask_login import current_user
|
|
|
|
# Module imports
|
|
from amanuensis.lexicon import LexiconModel
|
|
from amanuensis.user import UserModel
|
|
|
|
|
|
def register_custom_filters(app):
|
|
"""Adds custom filters to the Flask app"""
|
|
@app.template_filter("user_attr")
|
|
def user_attr(uid, attr):
|
|
user = UserModel.by(uid=uid)
|
|
val = getattr(user, attr)
|
|
return val
|
|
|
|
def lexicon_param(route):
|
|
"""Wrapper for loading a route's lexicon"""
|
|
@wraps(route)
|
|
def with_lexicon(name):
|
|
g.lexicon = LexiconModel.by(name=name)
|
|
if g.lexicon is None:
|
|
flash("Couldn't find a lexicon with the name '{}'".format(name))
|
|
return redirect(url_for("home.home"))
|
|
return route(name)
|
|
return with_lexicon
|
|
|
|
|
|
def admin_required(route):
|
|
"""Requires the user to be an admin to load this page"""
|
|
@wraps(route)
|
|
def admin_route(*args, **kwargs):
|
|
if not current_user.is_admin:
|
|
flash("You must be an admin to view this page")
|
|
return redirect(url_for('home.home'))
|
|
return route(*args, **kwargs)
|
|
return admin_route
|
|
|
|
|
|
def player_required(route):
|
|
"""Requires the user to be a player in the lexicon to load this page"""
|
|
@wraps(route)
|
|
def player_route(*args, **kwargs):
|
|
if current_user.id not in g.lexicon.join.joined:
|
|
flash("You must be a player to view this page")
|
|
return redirect(url_for('home.home'))
|
|
return route(*args, **kwargs)
|
|
return player_route |