bgneal@81: """
bgneal@81: Views for the forums application.
bgneal@81: """
bgneal@83: from django.contrib.auth.decorators import login_required
bgneal@82: from django.http import Http404
bgneal@98: from django.http import HttpResponse
bgneal@89: from django.http import HttpResponseBadRequest
bgneal@90: from django.http import HttpResponseForbidden
bgneal@83: from django.http import HttpResponseRedirect
bgneal@83: from django.core.urlresolvers import reverse
bgneal@91: from django.core.paginator import InvalidPage
bgneal@82: from django.shortcuts import get_object_or_404
bgneal@81: from django.shortcuts import render_to_response
bgneal@97: from django.template.loader import render_to_string
bgneal@81: from django.template import RequestContext
bgneal@89: from django.views.decorators.http import require_POST
bgneal@81: 
bgneal@90: from core.paginator import DiggPaginator
bgneal@98: from core.functions import email_admins
bgneal@81: from forums.models import Forum
bgneal@83: from forums.models import Topic
bgneal@91: from forums.models import Post
bgneal@98: from forums.models import FlaggedPost
bgneal@106: from forums.forms import NewTopicForm, NewPostForm, PostForm
bgneal@81: 
bgneal@90: #######################################################################
bgneal@90: 
bgneal@93: TOPICS_PER_PAGE = 50
bgneal@90: POSTS_PER_PAGE = 2
bgneal@90: 
bgneal@93: def create_topic_paginator(topics):
bgneal@93:    return DiggPaginator(topics, TOPICS_PER_PAGE, body=5, tail=2, margin=3, padding=2)
bgneal@93: 
bgneal@93: def create_post_paginator(posts):
bgneal@93:    return DiggPaginator(posts, POSTS_PER_PAGE, body=5, tail=2, margin=3, padding=2)
bgneal@90: 
bgneal@90: #######################################################################
bgneal@81: 
bgneal@81: def index(request):
bgneal@82:     """
bgneal@82:     This view displays all the forums available, ordered in each category.
bgneal@82:     """
bgneal@100:     forums = Forum.objects.forums_for_user(request.user)
bgneal@81:     cats = {}
bgneal@81:     for forum in forums:
bgneal@81:         cat = cats.setdefault(forum.category.id, {
bgneal@81:             'cat': forum.category,
bgneal@81:             'forums': [],
bgneal@81:             })
bgneal@81:         cat['forums'].append(forum)
bgneal@81: 
bgneal@81:     cmpdef = lambda a, b: cmp(a['cat'].position, b['cat'].position)
bgneal@81:     cats = sorted(cats.values(), cmpdef)
bgneal@81: 
bgneal@81:     return render_to_response('forums/index.html', {
bgneal@81:         'cats': cats,
bgneal@81:         },
bgneal@81:         context_instance=RequestContext(request))
bgneal@81: 
bgneal@82: 
bgneal@81: def forum_index(request, slug):
bgneal@82:     """
bgneal@82:     Displays all the topics in a forum.
bgneal@82:     """
bgneal@101:     forum = get_object_or_404(Forum.objects.select_related(), slug=slug)
bgneal@100: 
bgneal@100:     if not forum.category.can_access(request.user):
bgneal@100:         return HttpResponseForbidden()
bgneal@100: 
bgneal@100:     topics = forum.topics.select_related('last_post', 'last_post__user')
bgneal@93:     paginator = create_topic_paginator(topics)
bgneal@93:     page_num = int(request.GET.get('page', 1))
bgneal@93:     try:
bgneal@93:         page = paginator.page(page_num)
bgneal@93:     except InvalidPage:
bgneal@93:         raise Http404
bgneal@97: 
bgneal@97:     # we do this for the template since it is rendered twice
bgneal@97:     page_nav = render_to_string('forums/pagination.html', {'page': page})
bgneal@82:     
bgneal@82:     return render_to_response('forums/forum_index.html', {
bgneal@82:         'forum': forum,
bgneal@93:         'page': page,
bgneal@97:         'page_nav': page_nav,
bgneal@82:         },
bgneal@82:         context_instance=RequestContext(request))
bgneal@82: 
bgneal@82: 
bgneal@82: def topic_index(request, id):
bgneal@82:     """
bgneal@82:     Displays all the posts in a topic.
bgneal@82:     """
bgneal@101:     topic = get_object_or_404(Topic.objects.select_related(), pk=id)
bgneal@100: 
bgneal@100:     if not topic.forum.category.can_access(request.user):
bgneal@100:         return HttpResponseForbidden()
bgneal@100: 
bgneal@86:     topic.view_count += 1
bgneal@86:     topic.save()
bgneal@86: 
bgneal@86:     posts = topic.posts.select_related()
bgneal@93:     paginator = create_post_paginator(posts)
bgneal@93:     page_num = int(request.GET.get('page', 1))
bgneal@90:     try:
bgneal@90:         page = paginator.page(page_num)
bgneal@90:     except InvalidPage:
bgneal@90:         raise Http404
bgneal@90: 
bgneal@90:     last_page = page_num == paginator.num_pages
bgneal@86: 
bgneal@97:     # we do this for the template since it is rendered twice
bgneal@97:     page_nav = render_to_string('forums/pagination.html', {'page': page})
bgneal@97: 
bgneal@104:     can_moderate = request.user.is_authenticated() and (
bgneal@104:         request.user.is_superuser or \
bgneal@104:         request.user in topic.forum.moderators.all())
bgneal@104: 
bgneal@104:     can_reply = request.user.is_authenticated() and (
bgneal@104:         not topic.locked or can_moderate)
bgneal@104: 
bgneal@86:     return render_to_response('forums/topic.html', {
bgneal@86:         'forum': topic.forum,
bgneal@86:         'topic': topic,
bgneal@90:         'page': page,
bgneal@97:         'page_nav': page_nav,
bgneal@87:         'last_page': last_page,
bgneal@104:         'can_moderate': can_moderate,
bgneal@104:         'can_reply': can_reply,
bgneal@106:         'form': NewPostForm(initial={'topic_id': topic.id}),
bgneal@86:         },
bgneal@86:         context_instance=RequestContext(request))
bgneal@83: 
bgneal@83: 
bgneal@83: @login_required
bgneal@83: def new_topic(request, slug):
bgneal@83:     """
bgneal@83:     This view handles the creation of new topics.
bgneal@83:     """
bgneal@101:     forum = get_object_or_404(Forum.objects.select_related(), slug=slug)
bgneal@100: 
bgneal@100:     if not forum.category.can_access(request.user):
bgneal@100:         return HttpResponseForbidden()
bgneal@100: 
bgneal@83:     if request.method == 'POST':
bgneal@102:         form = NewTopicForm(request.user, forum, request.POST)
bgneal@83:         if form.is_valid():
bgneal@102:             topic = form.save(request.META.get("REMOTE_ADDR"))
bgneal@83:             return HttpResponseRedirect(reverse('forums-new_topic_thanks',
bgneal@83:                                             kwargs={'tid': topic.pk}))
bgneal@83:     else:
bgneal@102:         form = NewTopicForm(request.user, forum)
bgneal@83:     
bgneal@83:     return render_to_response('forums/new_topic.html', {
bgneal@83:         'forum': forum,
bgneal@83:         'form': form,
bgneal@83:         },
bgneal@83:         context_instance=RequestContext(request))
bgneal@83: 
bgneal@83: 
bgneal@83: @login_required
bgneal@83: def new_topic_thanks(request, tid):
bgneal@83:     """
bgneal@83:     This view displays the success page for a newly created topic.
bgneal@83:     """
bgneal@101:     topic = get_object_or_404(Topic.objects.select_related(), pk=tid)
bgneal@83:     return render_to_response('forums/new_topic_thanks.html', {
bgneal@83:         'forum': topic.forum,
bgneal@83:         'topic': topic,
bgneal@83:         },
bgneal@83:         context_instance=RequestContext(request))
bgneal@89: 
bgneal@89: 
bgneal@89: @require_POST
bgneal@89: def quick_reply_ajax(request):
bgneal@89:     """
bgneal@89:     This function handles the quick reply to a thread function. This
bgneal@89:     function is meant to be the target of an AJAX post, and returns
bgneal@89:     the HTML for the new post, which the client-side script appends
bgneal@89:     to the document.
bgneal@89:     """
bgneal@90:     if not request.user.is_authenticated():
bgneal@90:         return HttpResponseForbidden()
bgneal@90: 
bgneal@106:     form = NewPostForm(request.POST)
bgneal@89:     if form.is_valid():
bgneal@102:         if form.topic.locked or not form.topic.forum.category.can_access(request.user):
bgneal@100:             return HttpResponseForbidden()
bgneal@100: 
bgneal@89:         post = form.save(request.user, request.META.get("REMOTE_ADDR"))
bgneal@89:         return render_to_response('forums/display_post.html', {
bgneal@89:             'post': post,
bgneal@89:             },
bgneal@89:             context_instance=RequestContext(request))
bgneal@89: 
bgneal@89:     return HttpResponseBadRequest();
bgneal@89: 
bgneal@91: 
bgneal@91: def goto_post(request, post_id):
bgneal@91:     """
bgneal@91:     This function calculates what page a given post is on, then redirects
bgneal@91:     to that URL. This function is the target of get_absolute_url() for
bgneal@91:     Post objects.
bgneal@91:     """
bgneal@101:     post = get_object_or_404(Post.objects.select_related(), pk=post_id)
bgneal@91:     count = post.topic.posts.filter(creation_date__lt=post.creation_date).count()
bgneal@91:     page = count / POSTS_PER_PAGE + 1
bgneal@91:     url = reverse('forums-topic_index', kwargs={'id': post.topic.id}) + \
bgneal@91:         '?page=%s#p%s' % (page, post.id)
bgneal@91:     return HttpResponseRedirect(url)
bgneal@91: 
bgneal@98: 
bgneal@98: @require_POST
bgneal@98: def flag_post(request):
bgneal@98:     """
bgneal@98:     This function handles the flagging of posts by users. This function should
bgneal@98:     be the target of an AJAX post.
bgneal@98:     """
bgneal@98:     if not request.user.is_authenticated():
bgneal@99:         return HttpResponseForbidden('Please login or register to flag a post.')
bgneal@98: 
bgneal@98:     id = request.POST.get('id')
bgneal@98:     if id is None:
bgneal@98:         return HttpResponseBadRequest('No post id')
bgneal@98: 
bgneal@98:     try:
bgneal@98:         post = Post.objects.get(pk=id)
bgneal@98:     except Post.DoesNotExist:
bgneal@98:         return HttpResponseBadRequest('No post with id %s' % id)
bgneal@98: 
bgneal@98:     flag = FlaggedPost(user=request.user, post=post)
bgneal@98:     flag.save()
bgneal@98:     email_admins('A Post Has Been Flagged', """Hello,
bgneal@98: 
bgneal@98: A user has flagged a forum post for review.
bgneal@98: """)
bgneal@98:     return HttpResponse('The post was flagged. A moderator will review the post shortly. ' \
bgneal@98:             'Thanks for helping to improve the discussions on this site.')
bgneal@106: 
bgneal@106: 
bgneal@106: @login_required
bgneal@106: def edit_post(request, id):
bgneal@106:     """
bgneal@106:     This view function allows authorized users to edit posts.
bgneal@106:     The superuser, forum moderators, and original author can edit posts.
bgneal@106:     """
bgneal@106:     post = get_object_or_404(Post.objects.select_related(), pk=id)
bgneal@106:     can_edit = request.user == post.user or \
bgneal@106:             request.user.is_superuser or \
bgneal@106:             request.user in post.topic.forum.moderators.all()
bgneal@106: 
bgneal@106:     if not can_edit:
bgneal@106:         return HttpResponseForbidden("You don't have permission to edit that post.")
bgneal@106: 
bgneal@106:     if request.method == "POST":
bgneal@106:         form = PostForm(request.POST, instance=post)
bgneal@106:         if form.is_valid():
bgneal@106:             form.save()
bgneal@106:             return HttpResponseRedirect(post.get_absolute_url())
bgneal@106:     else:
bgneal@106:         form = PostForm(instance=post)
bgneal@106: 
bgneal@106:     return render_to_response('forums/edit_post.html', {
bgneal@106:         'forum': post.topic.forum,
bgneal@106:         'topic': post.topic,
bgneal@106:         'post': post,
bgneal@106:         'form': form,
bgneal@106:         'can_moderate': True,
bgneal@106:         },
bgneal@106:         context_instance=RequestContext(request))