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 unicode(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