from __future__ import unicode_literals from django.db import models from django.conf import settings from django.utils import timezone from django.utils.translation import gettext_lazy as _ from .settings import stored_messages_settings class Message(models.Model): """ This model represents a message on the database. Fields are the same as in `contrib.messages` """ message = models.TextField() level = models.IntegerField() tags = models.TextField() url = models.URLField(null=True, blank=True) date = models.DateTimeField(default=timezone.now) def __str__(self): return self.message class MessageArchive(models.Model): """ This model holds all the messages users received. Corresponding database table will grow indefinitely depending on messages traffic. """ user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) message = models.ForeignKey(Message, on_delete=models.CASCADE) def __str__(self): return "[%s] %s" % (self.user, self.message) class Inbox(models.Model): """ Inbox messages are stored in this model until users read them. Once read, inbox messages are deleted. Inbox messages have an expire time, after that they could be removed by a proper django command. We do not expect database table corresponding to this model to grow much. """ user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) message = models.ForeignKey(Message, on_delete=models.CASCADE) class Meta: verbose_name_plural = _('inboxes') def expired(self): expiration_date = self.message.date + timezone.timedelta( days=stored_messages_settings.INBOX_EXPIRE_DAYS) return expiration_date <= timezone.now() expired.boolean = True # show a nifty icon in the admin def __str__(self): return "[%s] %s" % (self.user, self.message)