mirror of
https://github.com/aljazceru/CTFd.git
synced 2025-12-18 06:24:23 +01:00
Adding Awards feature which closes #84
The Awards feature allows an admin to give users points. Eventually Awards will allow for uploading icons to make the award more commemorative.
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
from flask import current_app as app, session, render_template, jsonify, Blueprint, redirect, url_for, request
|
||||
from CTFd.utils import unix_time, authed, get_config
|
||||
from CTFd.models import db, Teams, Solves, Challenges
|
||||
from CTFd.models import db, Teams, Solves, Awards, Challenges
|
||||
from sqlalchemy.sql.expression import union_all
|
||||
|
||||
scoreboard = Blueprint('scoreboard', __name__)
|
||||
|
||||
@@ -10,14 +11,26 @@ def scoreboard_view():
|
||||
if get_config('view_scoreboard_if_authed') and not authed():
|
||||
return redirect(url_for('auth.login', next=request.path))
|
||||
score = db.func.sum(Challenges.value).label('score')
|
||||
quickest = db.func.max(Solves.date).label('quickest')
|
||||
teams = db.session.query(Solves.teamid, Teams.name, score)\
|
||||
.join(Teams)\
|
||||
.join(Challenges)\
|
||||
.filter(Teams.banned == None)\
|
||||
.group_by(Solves.teamid).order_by(score.desc(), quickest)
|
||||
scores = db.session.query(Solves.teamid.label('teamid'), Teams.name.label('name'), score, Solves.date.label('date')) \
|
||||
.join(Teams) \
|
||||
.join(Challenges) \
|
||||
.filter(Teams.banned == None) \
|
||||
.group_by(Solves.teamid)
|
||||
|
||||
awards = db.session.query(Teams.id.label('teamid'), Teams.name.label('name'),
|
||||
db.func.sum(Awards.value).label('score'), Awards.date.label('date')) \
|
||||
.filter(Teams.id == Awards.teamid) \
|
||||
.group_by(Teams.id)
|
||||
|
||||
results = union_all(scores, awards)
|
||||
|
||||
standings = db.session.query(results.columns.teamid, results.columns.name,
|
||||
db.func.sum(results.columns.score).label('score')) \
|
||||
.group_by(results.columns.teamid) \
|
||||
.order_by(db.func.sum(results.columns.score).desc(), db.func.max(results.columns.date)) \
|
||||
.all()
|
||||
db.session.close()
|
||||
return render_template('scoreboard.html', teams=teams)
|
||||
return render_template('scoreboard.html', teams=standings)
|
||||
|
||||
|
||||
@scoreboard.route('/scores')
|
||||
@@ -25,15 +38,26 @@ def scores():
|
||||
if get_config('view_scoreboard_if_authed') and not authed():
|
||||
return redirect(url_for('auth.login', next=request.path))
|
||||
score = db.func.sum(Challenges.value).label('score')
|
||||
quickest = db.func.max(Solves.date).label('quickest')
|
||||
teams = db.session.query(Solves.teamid, Teams.name, score)\
|
||||
.join(Teams)\
|
||||
.join(Challenges)\
|
||||
.filter(Teams.banned == None)\
|
||||
.group_by(Solves.teamid).order_by(score.desc(), quickest)
|
||||
scores = db.session.query(Solves.teamid.label('teamid'), Teams.name.label('name'), score, Solves.date.label('date')) \
|
||||
.join(Teams) \
|
||||
.join(Challenges) \
|
||||
.filter(Teams.banned == None) \
|
||||
.group_by(Solves.teamid)
|
||||
|
||||
awards = db.session.query(Teams.id.label('teamid'), Teams.name.label('name'), db.func.sum(Awards.value).label('score'), Awards.date.label('date'))\
|
||||
.filter(Teams.id==Awards.teamid)\
|
||||
.group_by(Teams.id)
|
||||
|
||||
results = union_all(scores, awards)
|
||||
|
||||
standings = db.session.query(results.columns.teamid, results.columns.name, db.func.sum(results.columns.score).label('score'))\
|
||||
.group_by(results.columns.teamid)\
|
||||
.order_by(db.func.sum(results.columns.score).desc(), db.func.max(results.columns.date))\
|
||||
.all()
|
||||
|
||||
db.session.close()
|
||||
json = {'standings':[]}
|
||||
for i, x in enumerate(teams):
|
||||
for i, x in enumerate(standings):
|
||||
json['standings'].append({'pos':i+1, 'id':x.teamid, 'team':x.name,'score':int(x.score)})
|
||||
return jsonify(json)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user