easyctf-2017/server/api/problem.py

70 lines
2 KiB
Python
Raw Normal View History

2015-12-31 02:56:00 +00:00
from flask import Blueprint, session, request
from flask import current_app as app
from models import db, Problems
from decorators import admins_only, api_wrapper
blueprint = Blueprint("problem", __name__)
@blueprint.route("/add", methods=["POST"])
@admins_only
@api_wrapper
def problem_add():
name = request.form["name"]
2016-01-02 04:18:40 +00:00
category = request.form["category"]
2015-12-31 02:56:00 +00:00
description = request.form["description"]
hint = request.form["hint"]
flag = request.form["flag"]
value = request.form["value"]
name_exists = Problems.query.filter_by(name=name).first()
if name_exists:
return { "success":0, "message": "Problem name already taken." }
2016-01-02 04:18:40 +00:00
problem = Problems(name, category, description, hint, flag, value)
2015-12-31 02:56:00 +00:00
db.session.add(problem)
db.session.commit()
return { "success": 1, "message": "Success!" }
@blueprint.route("/delete", methods=["POST"])
@admins_only
@api_wrapper
def problem_delete():
pid = request.form["pid"]
problem = Problems.query.filter_by(pid=pid).first()
if problem:
Solves.query.filter_by(pid=pid).delete()
Challenges.query.filter_by(pid=pid).delete()
db.session.commit()
return { "success": 1, "message": "Success!" }
return { "success": 0, "message": "Problem does not exist!" }
@blueprint.route("/update", methods=["POST"])
@admins_only
@api_wrapper
def problem_update():
pid = request.form["pid"]
name = request.form["name"]
description = request.form["description"]
hint = request.form["hint"]
flag = request.form["flag"]
disabled = request.form["disabled"]
value = request.form["value"]
problem = Problems.query.filter_by(pid=pid).first()
if problem:
problem.name = name
problem.description = description
problem.hint = hint
problem.flag = flag
problem.disabled = disabled
problem.value = value
db.session.add(problem)
db.session.commit()
return { "success": 1, "message": "Success!" }
return { "success": 0, "message": "Problem does not exist!" }