d62c545969542c1f35f1385476a3f1c3e5034c41
[redakcja.git] / src / wiki / helpers.py
1 from datetime import datetime
2 from functools import wraps
3
4 from django import http
5 import json
6 from django.utils.functional import Promise
7
8
9 class ExtendedEncoder(json.JSONEncoder):
10
11     def default(self, obj):
12         if isinstance(obj, Promise):
13             return str(obj)
14
15         if isinstance(obj, datetime):
16             return datetime.ctime(obj) + " " + (datetime.tzname(obj) or 'GMT')
17
18         return json.JSONEncoder.default(self, obj)
19
20
21 # shortcut for JSON reponses
22 class JSONResponse(http.HttpResponse):
23
24     def __init__(self, data={}, **kwargs):
25         # get rid of content_type
26         kwargs.pop('content_type', None)
27
28         data = json.dumps(data, cls=ExtendedEncoder)
29         super(JSONResponse, self).__init__(data, content_type="application/json", **kwargs)
30
31
32 # return errors
33 class JSONFormInvalid(JSONResponse):
34     def __init__(self, form):
35         super(JSONFormInvalid, self).__init__(form.errors, status=400)
36
37
38 class JSONServerError(JSONResponse):
39     def __init__(self, *args, **kwargs):
40         kwargs['status'] = 500
41         super(JSONServerError, self).__init__(*args, **kwargs)
42
43
44 def ajax_login_required(view):
45     @wraps(view)
46     def authenticated_view(request, *args, **kwargs):
47         if not request.user.is_authenticated():
48             return http.HttpResponse("Login required.", status=401, content_type="text/plain")
49         return view(request, *args, **kwargs)
50     return authenticated_view
51
52
53 def ajax_require_permission(permission):
54     def decorator(view):
55         @wraps(view)
56         def authorized_view(request, *args, **kwargs):
57             if not request.user.has_perm(permission):
58                 return http.HttpResponse("Access Forbidden.", status=403, content_type="text/plain")
59             return view(request, *args, **kwargs)
60         return authorized_view
61     return decorator