X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/24c1d259ba4af084959d70c6a1f355d0a57f1191..0c16061e3b790f05db393963620832629ddd225a:/apps/wiki/helpers.py?ds=sidebyside diff --git a/apps/wiki/helpers.py b/apps/wiki/helpers.py index d4daf1ad..dace3d00 100644 --- a/apps/wiki/helpers.py +++ b/apps/wiki/helpers.py @@ -1,8 +1,9 @@ +from datetime import datetime +from functools import wraps + from django import http from django.utils import simplejson as json from django.utils.functional import Promise -from django.template.loader import render_to_string -from datetime import datetime class ExtendedEncoder(json.JSONEncoder): @@ -24,9 +25,8 @@ class JSONResponse(http.HttpResponse): # get rid of mimetype kwargs.pop('mimetype', None) - super(JSONResponse, self).__init__( - json.dumps(data, cls=ExtendedEncoder), - mimetype="application/json", **kwargs) + data = json.dumps(data, cls=ExtendedEncoder) + super(JSONResponse, self).__init__(data, mimetype="application/json", **kwargs) # return errors @@ -39,3 +39,23 @@ class JSONServerError(JSONResponse): def __init__(self, *args, **kwargs): kwargs['status'] = 500 super(JSONServerError, self).__init__(*args, **kwargs) + + +def ajax_login_required(view): + @wraps(view) + def authenticated_view(request, *args, **kwargs): + if not request.user.is_authenticated(): + return http.HttpResponse("Login required.", status=401, mimetype="text/plain") + return view(request, *args, **kwargs) + return authenticated_view + + +def ajax_require_permission(permission): + def decorator(view): + @wraps(view) + def authorized_view(request, *args, **kwargs): + if not request.user.has_perm(permission): + return http.HttpResponse("Access Forbidden.", status=403, mimetype="text/plain") + return view(request, *args, **kwargs) + return authorized_view + return decorator