from django.db import models from django.conf import settings from django.db.models import signals from django.utils.translation import ugettext_lazy as _ from django.utils import timezone AUTH_USER_MODEL = getattr(settings, 'AUTH_USER_MODEL', 'auth.User') from accounts.models import User class MessageManager(models.Manager): def senders(self, user): users_ids = self.order_by('sender').filter(recipient=user).values_list('sender').distinct()[:10] users_ids = [item[0] for item in users_ids] return User.objects.filter(id__in=users_ids) def inbox_for(self, user): """ Returns all messages that were received by the given user and are not marked as deleted. """ return self.filter( recipient=user, recipient_deleted_at__isnull=True, ) def outbox_for(self, user): """ Returns all messages that were sent by the given user and are not marked as deleted. """ return self.filter( sender=user, sender_deleted_at__isnull=True, ) def trash_for(self, user): """ Returns all messages that were either received or sent by the given user and are marked as deleted. """ return self.filter( recipient=user, recipient_deleted_at__isnull=False, ) | self.filter( sender=user, sender_deleted_at__isnull=False, ) class Message(models.Model): """ A private message from user to user """ subject = models.CharField(_("Subject"), max_length=120) body = models.TextField(_("Body")) sender = models.ForeignKey(AUTH_USER_MODEL, related_name='sent_messages', verbose_name=_("Sender")) recipient = models.ForeignKey(AUTH_USER_MODEL, related_name='received_messages', null=True, blank=True, verbose_name=_("Recipient")) parent_msg = models.ForeignKey('self', related_name='next_messages', null=True, blank=True, verbose_name=_("Parent message")) sent_at = models.DateTimeField(_("sent at"), null=True, blank=True) read_at = models.DateTimeField(_("read at"), null=True, blank=True) replied_at = models.DateTimeField(_("replied at"), null=True, blank=True) sender_deleted_at = models.DateTimeField(_("Sender deleted at"), null=True, blank=True) recipient_deleted_at = models.DateTimeField(_("Recipient deleted at"), null=True, blank=True) objects = MessageManager() def new(self): """returns whether the recipient has read the message or not""" if self.read_at is not None: return False return True def replied(self): """returns whether the recipient has written a reply to this message""" if self.replied_at is not None: return True return False def __unicode__(self): return self.subject def get_absolute_url(self): return ('messages_detail', [self.id]) get_absolute_url = models.permalink(get_absolute_url) def save(self, **kwargs): if not self.id: self.sent_at = timezone.now() super(Message, self).save(**kwargs) class Meta: ordering = ['-sent_at'] verbose_name = _("Message") verbose_name_plural = _("Messages") def inbox_count_for(user): """ returns the number of unread messages for the given user but does not mark them seen """ return Message.objects.filter(recipient=user, read_at__isnull=True, recipient_deleted_at__isnull=True).count() # fallback for email notification if django-notification could not be found """ if "notification" not in settings.INSTALLED_APPS: from django_messages.utils import new_message_email signals.post_save.connect(new_message_email, sender=Message) """