Started adding the drawing of computer and network setups for each task

git-svn-id: https://svn.lusy.fri.uni-lj.si/kpov-public-svn/kpov-public@357 5cf9fbd1-b2bc-434c-b4b7-e852f4f63414

1 files changed, 7 insertions, 0 deletions

diff --git a/kpov_judge/web/kpov_judge/kpov_judge.py b/kpov_judge/web/kpov_judge/kpov_judge.py
index d5669d8..d02ba99 100755
--- a/kpov_judge/web/kpov_judge/kpov_judge.py
+++ b/kpov_judge/web/kpov_judge/kpov_judge.py
@@ -8,6 +8,7 @@ import random
 import settings
 import datetime
 import kpov_random_helpers
+from kpov_draw_setup import draw_setup
 import traceback
 
 app = Flask(__name__)
@@ -103,6 +104,12 @@ def results_json(task_id):
         return json.dumps(results_post(task_id, json.loads(flask.app.request.form['results'])))
     return json.dumps(results_dict(task_id))
 
+@app.route('/tasks/<task_id>/setup.svg', methods=['GET'])
+def setup_svg(task_id):
+    networks = db.networks.find({'task_id': task_id})
+    computers = db.computers_meta.find({'task_id': task_id})
+    return draw_setup(computers, networks)
+
 @app.route('/tasks/<task_id>/<lang>/results.html')
 def results_html(task_id, lang):
     return render_template('results.html', results=results_dict(task_id))