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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
import contextlib
import functools
import json
from django import http
from django.contrib.auth import decorators as auth_decorators
from django.utils.http import urlquote
from .models import use_master, skip_cache
from .urlresolvers import reverse
def login_required(f=None, redirect=True):
"""
Like Django's login_required, but with to= instead of next=.
If redirect=False then we return 401 instead of redirecting to the
login page. That's nice for ajax views.
"""
def decorator(func):
@functools.wraps(func)
def wrapper(request, *args, **kw):
if request.user.is_authenticated():
return func(request, *args, **kw)
else:
if redirect:
url = reverse('users.login')
path = urlquote(request.get_full_path())
return http.HttpResponseRedirect('%s?to=%s' % (url, path))
else:
return http.HttpResponse(status=401)
return wrapper
if f:
return decorator(f)
else:
return decorator
def post_required(f):
@functools.wraps(f)
def wrapper(request, *args, **kw):
if request.method != 'POST':
return http.HttpResponseNotAllowed(['POST'])
else:
return f(request, *args, **kw)
return wrapper
def json_view(f):
@functools.wraps(f)
def wrapper(*args, **kw):
response = f(*args, **kw)
if isinstance(response, http.HttpResponse):
return response
else:
return http.HttpResponse(json.dumps(response),
content_type='application/json')
return wrapper
json_view.error = lambda s: http.HttpResponseBadRequest(
json.dumps(s), content_type='application/json')
def write(f):
@functools.wraps(f)
def wrapper(*args, **kw):
with contextlib.nested(use_master(), skip_cache()):
return f(*args, **kw)
return wrapper
|
