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