X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/1b8ab1430082a145a3e4807de837dcc1568178a3..daacf83adcf2305add277eae0081f59c4c1a7863:/apps/wiki/helpers.py?ds=inline diff --git a/apps/wiki/helpers.py b/apps/wiki/helpers.py index 7ed97b41..dace3d00 100644 --- a/apps/wiki/helpers.py +++ b/apps/wiki/helpers.py @@ -1,41 +1,61 @@ +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): def default(self, obj): if isinstance(obj, Promise): - return unicode(obj) - + return unicode(obj) + if isinstance(obj, datetime): return datetime.ctime(obj) + " " + (datetime.tzname(obj) or 'GMT') - + return json.JSONEncoder.default(self, obj) + # shortcut for JSON reponses class JSONResponse(http.HttpResponse): - - def __init__(self, data = {}, **kwargs): + + def __init__(self, data={}, **kwargs): # 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 class JSONFormInvalid(JSONResponse): - def __init__(self, form): - super(JSONFormInvalid, self).__init__(form.errors, status = 400) - -class JSONServerError(JSONResponse): + def __init__(self, form): + super(JSONFormInvalid, self).__init__(form.errors, status=400) + + +class JSONServerError(JSONResponse): def __init__(self, *args, **kwargs): kwargs['status'] = 500 super(JSONServerError, self).__init__(*args, **kwargs) - - - \ No newline at end of file + + +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