From b2618e9404b84cc62d4becb02436233a0d53b375 Mon Sep 17 00:00:00 2001 From: Cédric Bonhomme Date: Wed, 25 Nov 2015 22:45:43 +0100 Subject: Rfactorization. Just a start... --- pyaggr3g470r/decorators.py | 50 ---------------------------------------------- 1 file changed, 50 deletions(-) delete mode 100644 pyaggr3g470r/decorators.py (limited to 'pyaggr3g470r/decorators.py') diff --git a/pyaggr3g470r/decorators.py b/pyaggr3g470r/decorators.py deleted file mode 100644 index 9e8f9c0b..00000000 --- a/pyaggr3g470r/decorators.py +++ /dev/null @@ -1,50 +0,0 @@ -#! /usr/bin/env python -# -*- coding: utf-8 -*- - -from threading import Thread -from functools import wraps - -from flask import g, redirect, url_for, flash -from flask.ext.babel import gettext -from flask.ext.login import login_required - -from pyaggr3g470r.models import Feed - - -def async(f): - """ - This decorator enables to launch a task (for examle sending an email or - indexing the database) in background. - This prevent the server to freeze. - """ - def wrapper(*args, **kwargs): - thr = Thread(target=f, args=args, kwargs=kwargs) - thr.start() - return wrapper - - -def feed_access_required(func): - """ - This decorator enables to check if a user has access to a feed. - The administrator of the platform is able to access to the feeds - of a normal user. - """ - @wraps(func) - def decorated(*args, **kwargs): - if kwargs.get('feed_id', None) is not None: - feed = Feed.query.filter(Feed.id == kwargs.get('feed_id', None)).first() - if feed is not None and (feed.subscriber.id == g.user.id or g.user.is_admin()): - return func(*args, **kwargs) - flash("This feed do not exist.", "danger") - return redirect(url_for('home')) - else: - return func(*args, **kwargs) - return decorated - - -def pyagg_default_decorator(func): - @login_required - @wraps(func) - def wrapper(*args, **kwargs): - return func(*args, **kwargs) - return wrapper -- cgit