aboutsummaryrefslogtreecommitdiff
path: root/pyaggr3g470r/decorators.py
blob: 9e8f9c0b1ea2759ee5e65d52be6f5077f497f189 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#! /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
bgstack15