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