""" Utilities for all views Ben Adida (12-30-2008) """ from django.template import Context, Template, loader from django.http import HttpResponse, Http404 from django.shortcuts import render_to_response import utils # nicely update the wrapper function from functools import update_wrapper from auth.security import get_user import helios from django.conf import settings ## ## BASICS ## SUCCESS = HttpResponse("SUCCESS") # FIXME: error code FAILURE = HttpResponse("FAILURE") ## ## template abstraction ## def prepare_vars(request, vars): vars_with_user = vars.copy() vars_with_user['user'] = get_user(request) # csrf protection if request.session.has_key('csrf_token'): vars_with_user['csrf_token'] = request.session['csrf_token'] vars_with_user['utils'] = utils vars_with_user['settings'] = settings vars_with_user['HELIOS_STATIC'] = '/static/helios/helios' vars_with_user['TEMPLATE_BASE'] = helios.TEMPLATE_BASE vars_with_user['CURRENT_URL'] = request.path vars_with_user['SECURE_URL_HOST'] = settings.SECURE_URL_HOST return vars_with_user def render_template(request, template_name, vars = {}, include_user=True): t = loader.get_template(template_name + '.html') vars_with_user = prepare_vars(request, vars) if not include_user: del vars_with_user['user'] return render_to_response('helios/templates/%s.html' % template_name, vars_with_user) def render_template_raw(request, template_name, vars={}): t = loader.get_template(template_name) # if there's a request, prep the vars, otherwise can't do it. if request: full_vars = prepare_vars(request, vars) else: full_vars = vars c = Context(full_vars) return t.render(c) def render_json(json_txt): return HttpResponse(json_txt) # decorator def json(func): """ A decorator that serializes the output to JSON before returning to the web client. """ def convert_to_json(self, *args, **kwargs): return render_json(utils.to_json(func(self, *args, **kwargs))) return update_wrapper(convert_to_json,func)