X-Git-Url: https://git.openstreetmap.org./osqa.git/blobdiff_plain/ca88510e4dd14206f623cababfda1e8a9daf70d4..7395cbba715aeda5b5d0927f33d4949b05803865:/forum/views/decorators.py diff --git a/forum/views/decorators.py b/forum/views/decorators.py index 90bb2b2..20eb45c 100644 --- a/forum/views/decorators.py +++ b/forum/views/decorators.py @@ -2,97 +2,79 @@ from django.http import HttpResponse, HttpResponseRedirect, Http404 from django.utils import simplejson from django.core.paginator import Paginator, EmptyPage from django.shortcuts import render_to_response +from django.core.urlresolvers import reverse from django.template import RequestContext from django.utils.translation import ungettext, ugettext as _ +from forum.modules import ui, decorate +from datetime import datetime, date +from forum.settings import ONLINE_USERS import logging -def render(template=None, tab=None): - def decorator(func): - def decorated(request, *args, **kwargs): - context = func(request, *args, **kwargs) +def login_required(func, request, *args, **kwargs): + if not request.user.is_authenticated(): + return HttpResponseRedirect(reverse('auth_signin')) + else: + return func(request, *args, **kwargs) - if tab is not None: - context['tab'] = tab - - return render_to_response(context.pop('template', template), context, context_instance=RequestContext(request)) - return decorated - return decorator +def render(template=None, tab=None, tab_title='', weight=500, tabbed=True): + def decorator(func): + def decorated(context, request, *args, **kwargs): + if request.user.is_authenticated(): + ONLINE_USERS[request.user] = datetime.now() -def list(paginate, default_page_size): - def decorator(func): - def decorated(request, *args, **kwargs): - context = func(request, *args, **kwargs) + if isinstance(context, HttpResponse): + return context - pagesize = request.utils.page_size(default_page_size) - page = int(request.GET.get('page', 1)) - - big_list = context[paginate] - paginator = Paginator(big_list, pagesize) - - try: - page_obj = paginator.page(page) - except EmptyPage: - raise Http404() - - context[paginate] = page_obj.object_list.lazy() - - base_path = context.get('base_path', None) or request.path - sort = request.utils.sort_method('') + if tab is not None: + context['tab'] = tab - context["pagination_context"] = { - 'is_paginated' : True, - 'pages': paginator.num_pages, - 'page': page, - 'has_previous': page_obj.has_previous(), - 'has_next': page_obj.has_next(), - 'previous': page_obj.previous_page_number(), - 'next': page_obj.next_page_number(), - 'base_url' : "%s%ssort=%s&" % (base_path, ('?' in base_path) and '&' or '?', sort), - 'pagesize' : pagesize - } + return render_to_response(context.pop('template', template), context, + context_instance=RequestContext(request)) - context['sort_context'] = { - 'base_url': "%s%ssort=" % (base_path, ('?' in base_path) and '&' or '?'), - 'current': sort, - } + if tabbed and tab and tab_title: + ui.register(ui.PAGE_TOP_TABS, + ui.PageTab(tab, tab_title, lambda: reverse(func.__name__), weight=weight)) + + return decorate.result.withfn(decorated, needs_params=True)(func) - return context - return decorated return decorator - class CommandException(Exception): pass +class RefreshPageCommand(HttpResponse): + def __init__(self): + super(RefreshPageCommand, self).__init__( + content=simplejson.dumps({'commands': {'refresh_page': []}, 'success': True}), + mimetype="application/json") -def command(func): - def decorated(request, *args, **kwargs): - try: - response = func(request, *args, **kwargs) +def command(func, request, *args, **kwargs): + try: + response = func(request, *args, **kwargs) - if isinstance(response, HttpResponse): - return response + if isinstance(response, HttpResponse): + return response - response['success'] = True - except Exception, e: - import sys, traceback - traceback.print_exc(file=sys.stdout) + response['success'] = True + except Exception, e: + import traceback + #traceback.print_exc() - if isinstance(e, CommandException): - response = { - 'success': False, - 'error_message': str(e) - } - else: - logging.error("%s: %s" % (func.__name__, str(e))) - response = { - 'success': False, - 'error_message': _("We're sorry, but an unknown error ocurred.
Please try again in a while.") - } - - if request.is_ajax(): - return HttpResponse(simplejson.dumps(response), mimetype="application/json") + if isinstance(e, CommandException): + response = { + 'success': False, + 'error_message': e.message + } else: - return HttpResponseRedirect(request.META.get('HTTP_REFERER', '/')) + logging.error("%s: %s" % (func.__name__, str(e))) + logging.error(traceback.format_exc()) + response = { + 'success': False, + 'error_message': _("We're sorry, but an unknown error ocurred.
Please try again in a while.") + } + + if request.is_ajax(): + return HttpResponse(simplejson.dumps(response), mimetype="application/json") + else: + return HttpResponseRedirect(request.META.get('HTTP_REFERER', '/')) - return decorated \ No newline at end of file