Mercurial > public > sg101
view messages/forms.py @ 919:0b6bf9c5a982
Bio app refactor.
For Django 1.7.7 upgrade.
author | Brian Neal <bgneal@gmail.com> |
---|---|
date | Mon, 06 Apr 2015 20:33:10 -0500 |
parents | 999a71b81111 |
children | 21c592cac71c |
line wrap: on
line source
""" Forms for the messages application. """ from django import forms from django.contrib.auth.models import User from django.conf import settings from django.contrib.sites.models import Site from django.core.urlresolvers import reverse from django.template.loader import render_to_string from core.functions import send_mail from core.widgets import AutoCompleteUserInput import messages from messages.models import Flag, Message, Options # Maximum size of a private message in characters MESSAGE_MAX = getattr(settings, 'MESSAGES_MAX_SIZE', 8192) class ComposeForm(forms.Form): """ This form is used to compose private messages. """ receiver = forms.CharField(label='To', max_length=30, widget=AutoCompleteUserInput()) subject = forms.CharField(max_length=120, widget=forms.TextInput(attrs={'size': 52})) message = forms.CharField(widget=forms.Textarea(attrs={'class': 'markItUp smileyTarget'})) attach_signature = forms.BooleanField(label='Attach Signature?', required=False) parent_id = forms.IntegerField(required=False, widget=forms.HiddenInput) def __init__(self, user, *args, **kwargs): forms.Form.__init__(self, *args, **kwargs) self.user = user options = Options.objects.for_user(user) self.fields['attach_signature'].initial = options.attach_signature def clean_receiver(self): receiver = self.cleaned_data['receiver'] try: self.rcvr_user = User.objects.get(username=receiver) except User.DoesNotExist: raise forms.ValidationError("That username does not exist.") if self.user == self.rcvr_user: raise forms.ValidationError("You can't send a message to yourself.") return receiver def clean_message(self): msg = self.cleaned_data['message'] if len(msg) > MESSAGE_MAX: raise forms.ValidationError("Your message is too long. Please trim some text.") return msg def clean(self): rcvr = self.cleaned_data.get('receiver') subject = self.cleaned_data.get('subject') message = self.cleaned_data.get('message') if rcvr and subject and message: # Can we send a message? Is our outbox full? count = Message.objects.outbox(self.user).count() if count >= messages.MSG_BOX_LIMIT: raise forms.ValidationError( "Your outbox is full. Please delete some messages.") # Is the receiver's inbox full? count = Message.objects.inbox(self.rcvr_user).count() if count >= messages.MSG_BOX_LIMIT: raise forms.ValidationError( "Sorry, %s's inbox is full. This message cannot be sent." % self.rcvr_user.username) return self.cleaned_data def save(self, parent_msg=None): sender = self.user receiver = self.rcvr_user subject = self.cleaned_data['subject'] message = self.cleaned_data['message'] attach_signature = self.cleaned_data['attach_signature'] new_msg = Message( sender=sender, receiver=receiver, subject=subject, message=message, signature_attached=attach_signature, ) new_msg.save() # Update the parent message (if there is one) parent_id = self.cleaned_data['parent_id'] if parent_id: try: parent_msg = Message.objects.get(pk=parent_id) except Message.DoesNotExist: parent_msg = None if parent_msg and parent_msg.receiver == self.user: parent_msg.reply_date = new_msg.send_date parent_msg.save() # Notify recipient receiver_opts = Options.objects.for_user(receiver) if receiver_opts.notify_email: notify_receiver(new_msg) class Media: css = { 'all': (settings.GPP_THIRD_PARTY_CSS['markitup'] + settings.GPP_THIRD_PARTY_CSS['jquery-ui']) } js = (settings.GPP_THIRD_PARTY_JS['markitup'] + settings.GPP_THIRD_PARTY_JS['jquery-ui']) class OptionsForm(forms.ModelForm): class Meta: model = Options fields = ['attach_signature', 'notify_email'] class ReportForm(forms.ModelForm): class Meta: model = Flag fields = ['comments'] labels = {'comments': ''} widgets = { 'comments': forms.Textarea(attrs={ 'placeholder': 'Enter a comment for the admin (optional).'}) } def notify_receiver(new_msg): """ This function creates the notification email to notify a user of a new private message. """ site = Site.objects.get_current() email_body = render_to_string('messages/notification_email.txt', { 'site': site, 'msg': new_msg, 'options_url': reverse('messages-options'), }) subject = 'New private message for %s at %s' % (new_msg.receiver.username, site.name) from_email = settings.GPP_NO_REPLY_EMAIL + '@' + site.domain send_mail(subject, email_body, from_email, [new_msg.receiver.email])