Statistics
| Branch: | Tag: | Revision:

root / snf-cyclades-app / synnefo / plankton / util.py @ 2db7d9df

History | View | Annotate | Download (3.1 kB)

1
# Copyright 2011 GRNET S.A. All rights reserved.
2
#
3
# Redistribution and use in source and binary forms, with or
4
# without modification, are permitted provided that the following
5
# conditions are met:
6
#
7
#   1. Redistributions of source code must retain the above
8
#      copyright notice, this list of conditions and the following
9
#      disclaimer.
10
#
11
#   2. Redistributions in binary form must reproduce the above
12
#      copyright notice, this list of conditions and the following
13
#      disclaimer in the documentation and/or other materials
14
#      provided with the distribution.
15
#
16
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
17
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
19
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
20
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
21
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
22
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
23
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
24
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
26
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
27
# POSSIBILITY OF SUCH DAMAGE.
28
#
29
# The views and conclusions contained in the software and
30
# documentation are those of the authors and should not be
31
# interpreted as representing official policies, either expressed
32
# or implied, of GRNET S.A.
33

    
34
from functools import wraps
35
from logging import getLogger
36
from traceback import format_exc
37

    
38
from django.conf import settings
39
from django.http import (HttpResponse, HttpResponseBadRequest,
40
                         HttpResponseServerError, HttpResponseForbidden)
41

    
42
from synnefo.lib.astakos import get_user
43
from synnefo.plankton.backend import (ImageBackend, BackendException,
44
                                      NotAllowedError)
45

    
46
log = getLogger('synnefo.plankton')
47

    
48

    
49
def plankton_method(method):
50
    def decorator(func):
51
        @wraps(func)
52
        def wrapper(request, *args, **kwargs):
53
            try:
54
                get_user(request, settings.ASTAKOS_URL)
55
                if not request.user_uniq:
56
                    return HttpResponse(status=401)
57
                if request.method != method:
58
                    return HttpResponse(status=405)
59
                request.backend = ImageBackend(request.user_uniq)
60
                return func(request, *args, **kwargs)
61
            except (AssertionError, BackendException) as e:
62
                message = e.args[0] if e.args else ''
63
                return HttpResponseBadRequest(message)
64
            except NotAllowedError:
65
                return HttpResponseForbidden()
66
            except Exception as e:
67
                if settings.DEBUG:
68
                    message = format_exc(e)
69
                else:
70
                    message = ''
71
                log.exception(e)
72
                return HttpResponseServerError(message)
73
            finally:
74
                if hasattr(request, 'backend'):
75
                    request.backend.close()
76
        return wrapper
77
    return decorator