diff frontend/mfrontend/decorators.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
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/frontend/mfrontend/decorators.py	Mon Mar 07 10:47:48 2011 +0100
@@ -0,0 +1,17 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+from functools import wraps
+from flask import g, request, redirect, url_for, session, flash
+
+def login_required(f):
+    @wraps(f)
+    def decorated_function(*args, **kwargs):
+        try:
+            if not session["logged_in"]:
+                raise Exception()
+        except Exception as e:   
+            flash(u"Nie masz uprawnień do oglądania żądanej strony.")
+            return redirect(url_for('frontend.login'))
+        return f(*args, **kwargs)
+    return decorated_function