Misago/misago/apps/profiles/warnings/views.py

from datetime import timedelta
from django.core.cache import cache
from django.core.urlresolvers import reverse
from django.http import Http404
from django.shortcuts import redirect
from django.utils import timezone
from django.utils.translation import ugettext as _
from misago import messages
from misago.apps.errors import error404
from misago.apps.profiles.decorators import profile_view
from misago.apps.profiles.template import RequestContext
from misago.apps.profiles.warnings.warningstracker import WarningsTracker
from misago.apps.warnuser import alerts
from misago.decorators import block_guest, check_csrf
from misago.models import Warn
from misago.shortcuts import render_to_response
from misago.utils.pagination import make_pagination

@profile_view('user_warnings')
def warnings(request, user, page=0):
    request.acl.warnings.allow_member_warns_view(request.user, user)

    queryset = user.warning_set
    count = queryset.count()
    try:
        pagination = make_pagination(page, count, 12)
    except Http404:
        return redirect(reverse('user_warnings', kwargs={'user': user.id, 'username': user.username_slug}))

    return render_to_response('profiles/warnings.html',
                              context_instance=RequestContext(request, {
                                  'profile': user,
                                  'tab': 'warnings',
                                  'items_total': count,
                                  'warning_level': user.get_current_warning_level(),
                                  'warnings_tracker': WarningsTracker(user.warning_level - pagination['start']),
                                  'items': queryset.order_by('-id')[pagination['start']:pagination['stop']],
                                  'pagination': pagination,
                                  }));

def warning_decorator(f):
    def decorator(*args, **kwargs):
        request, user = args
        request.acl.warnings.allow_member_warns_view(request.user, user)
        warning_pk = kwargs['warning']
        try:
            warning = user.warning_set.get(pk=warning_pk)
            f(request, user, warning)
            return redirect('user_warnings', username=user.username_slug, user=user.pk)
        except Warn.DoesNotExist:
            return error404(request, _("Requested warning could not be found."))
    return decorator


@block_guest
@check_csrf
@profile_view('user_warnings_cancel')
@warning_decorator
def cancel_warning(request, user, warning):
    request.acl.warnings.allow_cancel_warning(
        request.user, user, warning)

    user.decrease_warning_level()
    warning.canceled = True
    warning.canceled_on = timezone.now()
    warning.canceler = request.user
    warning.canceler_username = request.user.username
    warning.canceler_slug = request.user.username_slug
    warning.canceler_ip = request.session.get_ip(request)
    warning.canceler_agent = request.META.get('HTTP_USER_AGENT')
    warning.save(force_update=True)

    alerts.your_warn_has_been_canceled(request.user, user)
    messages.success(request, _("Selected warning has been canceled."))


@block_guest
@check_csrf
@profile_view('user_warnings_delete')
@warning_decorator
def delete_warning(request, user, warning):
    request.acl.warnings.allow_delete_warning()

    if user.is_warning_active(warning):
        alerts.your_warn_has_been_canceled(request.user, user)
        user.decrease_warning_level()
    warning.delete()

    messages.success(request, _("Selected warning has been deleted."))