bgneal@425: """
bgneal@425: Views for the messages application.
bgneal@425: 
bgneal@425: """
bgneal@425: import datetime
bgneal@679: import json
bgneal@425: 
bgneal@425: from django.contrib.auth.decorators import login_required
bgneal@806: from django.views.decorators.http import require_POST
bgneal@428: from django.contrib.auth.models import User
bgneal@436: from django.contrib import messages as django_messages
bgneal@801: from django.core.paginator import Paginator, EmptyPage, PageNotAnInteger
bgneal@425: from django.core.urlresolvers import reverse
bgneal@425: from django.http import HttpResponse
bgneal@425: from django.http import HttpResponseForbidden
bgneal@425: from django.http import HttpResponseNotAllowed
bgneal@425: from django.shortcuts import get_object_or_404
bgneal@805: from django.shortcuts import render, redirect
bgneal@425: 
bgneal@425: from messages.models import Message, Options
bgneal@425: from messages.forms import OptionsForm, ComposeForm
bgneal@566: from messages.utils import reply_subject
bgneal@436: from messages import MSG_BOX_LIMIT
bgneal@566: from core.functions import quote_message
bgneal@425: 
bgneal@425: 
bgneal@436: MSGS_PER_PAGE = 20      # message pagination value
bgneal@425: 
bgneal@436: # This must match the jQuery UI tab control
bgneal@425: TAB_INDICES = {
bgneal@425:     'inbox': 0,
bgneal@425:     'compose': 1,
bgneal@425:     'outbox': 2,
bgneal@425:     'trash': 3,
bgneal@425:     'options': 4,
bgneal@425: }
bgneal@425: 
bgneal@425: 
bgneal@436: def _quota_check(box_name, count, request):
bgneal@436:     """
bgneal@436:     Checks the message box count against MSG_BOX_LIMIT.
bgneal@436:     Emits a message to the user if the quota is exceeded.
bgneal@441: 
bgneal@568:     Returns the percent used as an integer between 0-100.
bgneal@568: 
bgneal@436:     """
bgneal@436:     if count >= MSG_BOX_LIMIT:
bgneal@436:         django_messages.warning(request,
bgneal@436:             "Your %s is full. Please delete some messages." % box_name)
bgneal@436: 
bgneal@568:     return 100 * count / MSG_BOX_LIMIT
bgneal@568: 
bgneal@436: 
bgneal@801: def _get_page(request, qs):
bgneal@801:     """Paginates the given queryset and returns a page object"""
bgneal@801:     paginator = Paginator(qs, MSGS_PER_PAGE)
bgneal@801:     try:
bgneal@801:         page = paginator.page(request.GET.get('page', '1'))
bgneal@801:     except PageNotAnInteger:
bgneal@801:         page = paginator.page(1)
bgneal@801:     except EmptyPage:
bgneal@801:         page = paginator.page(paginator.num_pages)
bgneal@801:     return page
bgneal@801: 
bgneal@801: 
bgneal@425: @login_required
bgneal@801: def inbox(request):
bgneal@425: 
bgneal@801:     msg_list = Message.objects.inbox(request.user)
bgneal@801:     msg_count = msg_list.count()
bgneal@801:     pct_used = _quota_check('inbox', msg_count, request)
bgneal@801: 
bgneal@801:     page = _get_page(request, msg_list)
bgneal@801: 
bgneal@801:     return render(request, 'messages/inbox.html', {
bgneal@801:         'tab': 'inbox',
bgneal@801:         'page': page,
bgneal@801:         'inbox_pct': pct_used,
bgneal@801:         'outbox_pct': None,
bgneal@801:         })
bgneal@801: 
bgneal@801: 
bgneal@801: @login_required
bgneal@801: def outbox(request):
bgneal@801: 
bgneal@801:     msg_list = Message.objects.outbox(request.user)
bgneal@801:     msg_count = msg_list.count()
bgneal@801:     pct_used = _quota_check('outbox', msg_count, request)
bgneal@801: 
bgneal@801:     page = _get_page(request, msg_list)
bgneal@801: 
bgneal@801:     return render(request, 'messages/outbox.html', {
bgneal@801:         'tab': 'outbox',
bgneal@801:         'page': page,
bgneal@801:         'inbox_pct': None,
bgneal@801:         'outbox_pct': pct_used,
bgneal@425:         })
bgneal@425: 
bgneal@425: 
bgneal@428: @login_required
bgneal@802: def trash(request):
bgneal@802: 
bgneal@802:     msg_list = Message.objects.trash(request.user)
bgneal@802: 
bgneal@802:     page = _get_page(request, msg_list)
bgneal@802: 
bgneal@802:     return render(request, 'messages/trash.html', {
bgneal@802:         'tab': 'trash',
bgneal@802:         'page': page,
bgneal@802:         'inbox_pct': None,
bgneal@802:         'outbox_pct': None,
bgneal@802:         })
bgneal@802: 
bgneal@802: 
bgneal@802: @login_required
bgneal@803: def options(request):
bgneal@803:     """
bgneal@803:     This view handles the displaying and changing of private message options.
bgneal@803: 
bgneal@803:     """
bgneal@803:     if request.method == 'POST':
bgneal@803:         options = Options.objects.for_user(request.user)
bgneal@803:         form = OptionsForm(request.POST, instance=options, prefix='opts')
bgneal@803:         if form.is_valid():
bgneal@803:             form.save()
bgneal@803:             django_messages.success(request, 'Options saved.')
bgneal@803:     else:
bgneal@803:         options = Options.objects.for_user(request.user)
bgneal@803:         form = OptionsForm(instance=options, prefix='opts')
bgneal@803: 
bgneal@803:     return render(request, 'messages/options.html', {
bgneal@803:         'tab': 'options',
bgneal@803:         'form': form,
bgneal@803:         })
bgneal@803: 
bgneal@803: 
bgneal@803: @login_required
bgneal@804: def compose(request):
bgneal@428:     """
bgneal@804:     Process or prepare the compose form to create a new private message.
bgneal@428: 
bgneal@428:     """
bgneal@804:     if request.method == 'POST':
bgneal@804:         compose_form = ComposeForm(request.user, request.POST)
bgneal@804:         if compose_form.is_valid():
bgneal@804:             compose_form.save()
bgneal@804:             django_messages.success(request, 'Message sent.')
bgneal@804:             compose_form = ComposeForm(request.user)
bgneal@804:     else:
bgneal@804:         receiver = request.GET.get('to')
bgneal@804:         if receiver:
bgneal@804:             form_data = {'receiver': receiver}
bgneal@804:             compose_form = ComposeForm(request.user, initial=form_data)
bgneal@804:         else:
bgneal@804:             compose_form = ComposeForm(request.user)
bgneal@804: 
bgneal@804:         _quota_check('outbox', Message.objects.outbox(request.user).count(), request)
bgneal@804: 
bgneal@804:     return render(request, 'messages/compose.html', {
bgneal@804:         'tab': 'compose',
bgneal@804:         'compose_form': compose_form,
bgneal@428:         })
bgneal@428: 
bgneal@428: 
bgneal@804: @login_required
bgneal@804: def view(request, msg_id):
bgneal@425:     """
bgneal@805:     This view function displays a private message for reading to the user. If
bgneal@805:     the user is a recipient of the message, a reply can be composed and sent.
bgneal@425: 
bgneal@425:     """
bgneal@805:     if request.method == 'POST':
bgneal@805:         form = ComposeForm(request.user, request.POST)
bgneal@805:         if form.is_valid():
bgneal@805:             form.save()
bgneal@805:             django_messages.success(request, 'Reply sent.')
bgneal@805:             return redirect('messages-inbox')
bgneal@805:     else:
bgneal@805:         msg = get_object_or_404(Message.objects.select_related(), pk=msg_id)
bgneal@805:         if msg.sender != request.user and msg.receiver != request.user:
bgneal@805:             django_messages.error(request,
bgneal@805:                     "You don't have permission to read that message.")
bgneal@805:             return redirect('messages-inbox')
bgneal@425: 
bgneal@805:         initial_data = {
bgneal@805:             'receiver': msg.sender.username,
bgneal@805:             'subject': reply_subject(msg.subject),
bgneal@805:             'message': quote_message(msg.sender.username, msg.message),
bgneal@805:             'parent_id': msg.pk,
bgneal@805:         }
bgneal@425: 
bgneal@805:         if msg.receiver == request.user:
bgneal@805:             if msg.read_date is None:
bgneal@805:                 msg.read_date = datetime.datetime.now()
bgneal@805:                 msg.save()
bgneal@805:         else:
bgneal@805:             initial_data['receiver'] = msg.receiver.username
bgneal@425: 
bgneal@805:         form = ComposeForm(request.user, initial=initial_data)
bgneal@425: 
bgneal@805:     return render(request, 'messages/view_message.html', {
bgneal@805:         'msg': msg,
bgneal@805:         'form': form,
bgneal@805:         })
bgneal@425: 
bgneal@425: 
bgneal@806: def _delete_pms(user, pm_ids):
bgneal@425:     """
bgneal@806:     Process the request to delete the list of PM ids by user.
bgneal@425: 
bgneal@425:     """
bgneal@806:     msgs = Message.objects.filter(id__in=pm_ids)
bgneal@806:     now = datetime.datetime.now()
bgneal@425: 
bgneal@425:     for msg in msgs:
bgneal@425:         if msg.sender == user:
bgneal@806:             if msg.receiver_delete_date is not None or msg.read_date is None:
bgneal@425:                 # Both parties deleted the message or receiver hasn't read it
bgneal@425:                 # yet, we can delete it now
bgneal@425:                 msg.delete()
bgneal@425:             else:
bgneal@806:                 # receiver still has PM in their inbox
bgneal@806:                 msg.sender_delete_date = now
bgneal@425:                 msg.save()
bgneal@425:         elif msg.receiver == user:
bgneal@425:             if msg.sender_delete_date is not None:
bgneal@425:                 # both parties deleted the message, we can delete it now
bgneal@425:                 msg.delete()
bgneal@425:             else:
bgneal@806:                 # sender still has PM in their inbox
bgneal@806:                 msg.receiver_delete_date = now
bgneal@425:                 msg.save()
bgneal@425: 
bgneal@425: 
bgneal@806: @login_required
bgneal@806: @require_POST
bgneal@806: def delete(request):
bgneal@806:     """
bgneal@806:     Deletes the requested PM's. The user must be either a sender or receiver for
bgneal@806:     this to work.
bgneal@806: 
bgneal@806:     """
bgneal@806:     pm_ids = request.POST.getlist('pm_ids')
bgneal@806:     if pm_ids:
bgneal@806:         _delete_pms(request.user, pm_ids)
bgneal@806: 
bgneal@806:     # Figure out where to redirect to
bgneal@806:     src = request.POST.get('src', 'inbox')
bgneal@806:     try:
bgneal@806:         page = int(request.POST.get('page', '1'))
bgneal@806:     except ValueError:
bgneal@806:         page = 1
bgneal@806: 
bgneal@806:     view_name = 'messages-inbox' if src == inbox else 'messages-outbox'
bgneal@806:     url = reverse(view_name) + '?page={}'.format(page)
bgneal@806:     return redirect(url)
bgneal@806: 
bgneal@806: 
bgneal@425: def _undelete_msgs(user, msg_ids):
bgneal@425:     """
bgneal@425:     Attempts to "undelete" the messages given by the msg_ids list.
bgneal@425:     This will only succeed if the user is either the sender or receiver.
bgneal@425: 
bgneal@425:     """
bgneal@425:     msg_ids = _only_integers(msg_ids)
bgneal@425:     msgs = Message.objects.filter(id__in=msg_ids)
bgneal@425:     for msg in msgs:
bgneal@425:         if msg.sender == user:
bgneal@425:             msg.sender_delete_date = None
bgneal@425:             msg.save()
bgneal@425:         elif msg.receiver == user:
bgneal@425:             msg.receiver_delete_date = None
bgneal@425:             msg.save()
bgneal@425: 
bgneal@425: 
bgneal@425: def bulk(request):
bgneal@425:     """
bgneal@425:     This view processes messages in bulk. Arrays of message ids are expected in
bgneal@425:     the POST query dict: inbox_ids and outbox_ids will be deleted; trash_ids will
bgneal@425:     be undeleted.
bgneal@425: 
bgneal@425:     """
bgneal@425:     if not request.user.is_authenticated():
bgneal@425:         return HttpResponseForbidden()
bgneal@425:     if request.method != 'POST':
bgneal@425:         return HttpResponseNotAllowed(['POST'])
bgneal@425: 
bgneal@425:     delete_ids = []
bgneal@425:     if 'inbox_ids' in request.POST:
bgneal@425:         delete_ids.extend(request.POST.getlist('inbox_ids'))
bgneal@425:     if 'outbox_ids' in request.POST:
bgneal@425:         delete_ids.extend(request.POST.getlist('outbox_ids'))
bgneal@425: 
bgneal@425:     if len(delete_ids):
bgneal@425:         _delete_msgs(request.user, delete_ids)
bgneal@425: 
bgneal@425:     if 'trash_ids' in request.POST:
bgneal@425:         _undelete_msgs(request.user, request.POST.getlist('trash_ids'))
bgneal@425: 
bgneal@425:     return HttpResponse('');