diff frontend/mfrontend/views/jobcontrol.py @ 10:58a993029d9f

Dodawanie nowych zadań i dekorator sprawdzający zalogowanie.
author Michał Rudowicz <michal.rudowicz@fl9.eu>
date Mon, 07 Mar 2011 10:47:48 +0100
parents 9a0a9fa7f91d
children 79cf8ca2a0e5
line wrap: on
line diff
--- a/frontend/mfrontend/views/jobcontrol.py	Mon Feb 28 19:25:51 2011 +0100
+++ b/frontend/mfrontend/views/jobcontrol.py	Mon Mar 07 10:47:48 2011 +0100
@@ -2,11 +2,25 @@
 # -*- coding: utf-8 -*-
 
 from flask import Module, render_template, request, url_for
-from flask import escape, session, redirect, flash, abort
-from mfrontend import db, utils, exceptions
+from flask import escape, session, redirect, flash, abort, g
+from mfrontend import db, utils, exceptions, decorators
 jobcontrol = Module(__name__)
 
 @jobcontrol.route('/')
+@decorators.login_required
 def index():
     jobs = db.get_jobs(None, None, None)
     return render_template('joblist.html', jobs=jobs)
+
+@jobcontrol.route('/add/', methods=['GET', 'POST'])
+@decorators.login_required
+def add():
+    if request.method == 'POST':
+        try:
+            db.new_job(session['user_id'],
+                        request.form["label"],
+                        request.form["hash"])
+            return redirect(url_for('index'))
+        except Exception as e:
+            flash(e)
+    return render_template('newjob.html')