You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
105 lines
3.6 KiB
105 lines
3.6 KiB
from django.conf import settings
|
|
from django.db import models
|
|
from django.utils import timezone
|
|
from django.utils.encoding import python_2_unicode_compatible
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
|
|
AUTH_USER_MODEL = getattr(settings, 'AUTH_USER_MODEL', 'auth.User')
|
|
|
|
|
|
class MessageManager(models.Manager):
|
|
|
|
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,
|
|
)
|
|
|
|
|
|
@python_2_unicode_compatible
|
|
class Message(models.Model):
|
|
"""
|
|
A private message from user to user
|
|
"""
|
|
subject = models.CharField(_("Subject"), max_length=140)
|
|
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 __str__(self):
|
|
return self.subject
|
|
|
|
def get_absolute_url(self):
|
|
return 'django_messages: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 and getattr(settings, 'DJANGO_MESSAGES_NOTIFY', True):
|
|
# from django_messages.utils import new_message_email
|
|
# signals.post_save.connect(new_message_email, sender=Message)
|