Files
CTFd/CTFd/users.py
2019-04-17 01:36:30 -04:00

61 lines
1.6 KiB
Python

from flask import request, render_template, Blueprint
from CTFd.models import Users
from CTFd.utils.decorators import authed_only
from CTFd.utils import config
from CTFd.utils.user import get_current_user
from CTFd.utils.decorators.visibility import check_account_visibility, check_score_visibility
users = Blueprint('users', __name__)
@users.route('/users')
@check_account_visibility
def listing():
page = abs(request.args.get('page', 1, type=int))
results_per_page = 50
page_start = results_per_page * (page - 1)
page_end = results_per_page * (page - 1) + results_per_page
count = Users.query.filter_by(banned=False, hidden=False).count()
users = Users.query.filter_by(banned=False, hidden=False).slice(page_start, page_end).all()
pages = int(count / results_per_page) + (count % results_per_page > 0)
return render_template(
'users/users.html',
users=users,
pages=pages,
curr_page=page
)
@users.route('/profile')
@users.route('/user')
@authed_only
def private():
user = get_current_user()
solves = user.get_solves()
awards = user.get_awards()
place = user.place
score = user.score
return render_template(
'users/private.html',
solves=solves,
awards=awards,
user=user,
score=score,
place=place,
score_frozen=config.is_scoreboard_frozen()
)
@users.route('/users/<int:user_id>')
@check_account_visibility
@check_score_visibility
def public(user_id):
user = Users.query.filter_by(id=user_id, banned=False, hidden=False).first_or_404()
return render_template('users/public.html', user=user)