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