From c6b6766e6bfa25dcbe3167266cdd1a7d85e02428 Mon Sep 17 00:00:00 2001 From: James Wang Date: Fri, 1 Jan 2016 23:18:40 -0500 Subject: [PATCH] Add category to Problems --- server/api/models.py | 4 +++- server/api/problem.py | 3 ++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/server/api/models.py b/server/api/models.py index 9f6b7be..d141404 100644 --- a/server/api/models.py +++ b/server/api/models.py @@ -37,6 +37,7 @@ class Teams(db.Model): class Problems(db.Model): pid = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(128)) + category = db.Column(db.String(128)) description = db.Column(db.Text) hint = db.Column(db.Text) flag = db.Column(db.Text) @@ -44,8 +45,9 @@ class Problems(db.Model): value = db.Column(db.Integer) solves = db.Column(db.Integer) - def __init__(self, name, description, hint, flag, value): + def __init__(self, name, category, description, hint, flag, value): self.name = name + self.category = category self.description = description self.hint = hint self.flag = flag diff --git a/server/api/problem.py b/server/api/problem.py index 638d6f5..b2a3f75 100644 --- a/server/api/problem.py +++ b/server/api/problem.py @@ -11,6 +11,7 @@ blueprint = Blueprint("problem", __name__) @api_wrapper def problem_add(): name = request.form["name"] + category = request.form["category"] description = request.form["description"] hint = request.form["hint"] flag = request.form["flag"] @@ -21,7 +22,7 @@ def problem_add(): if name_exists: return { "success":0, "message": "Problem name already taken." } - problem = Problems(name, description, hint, flag, value) + problem = Problems(name, category, description, hint, flag, value) db.session.add(problem) db.session.commit()