django-orchestra-test/orchestra/contrib/mailer/models.py

from django.db import models
from django.utils.translation import gettext_lazy as _

from . import settings


class Message(models.Model):
    QUEUED = 'QUEUED'
    SENT = 'SENT'
    DEFERRED = 'DEFERRED'
    FAILED = 'FAILED'
    STATES = (
        (QUEUED, _("Queued")),
        (SENT, _("Sent")),
        (DEFERRED, _("Deferred")),
        (FAILED, _("Failed")),
    )

    CRITICAL = 0
    HIGH = 1
    NORMAL = 2
    LOW = 3
    PRIORITIES = (
        (CRITICAL, _("Critical (not queued)")),
        (HIGH, _("High")),
        (NORMAL, _("Normal")),
        (LOW, _("Low")),
    )

    state = models.CharField(_("State"), max_length=16, choices=STATES, default=QUEUED,
        db_index=True)
    priority = models.PositiveIntegerField(_("Priority"), choices=PRIORITIES, default=NORMAL,
        db_index=True)
    to_address = models.CharField(max_length=256)
    from_address = models.CharField(max_length=256)
    subject = models.TextField(_("subject"))
    content = models.TextField(_("content"))
    created_at = models.DateTimeField(_("created at"), auto_now_add=True)
    retries = models.PositiveIntegerField(_("retries"), default=0, db_index=True)
    last_try = models.DateTimeField(_("last try"), null=True, db_index=True)

    def __str__(self):
        return '%s to %s' % (self.subject, self.to_address)

    def defer(self):
        self.state = self.DEFERRED
        # Max tries
        if self.retries >= len(settings.MAILER_DEFERE_SECONDS):
            self.state = self.FAILED
        self.save(update_fields=('state',))

    def sent(self):
        self.state = self.SENT
        self.save(update_fields=('state',))

    def log(self, error):
        result = SMTPLog.SUCCESS
        if error:
            result= SMTPLog.FAILURE
        self.logs.create(log_message=str(error), result=result)


class SMTPLog(models.Model):
    SUCCESS = 'SUCCESS'
    FAILURE = 'FAILURE'
    RESULTS = (
        (SUCCESS, _("Success")),
        (FAILURE, _("Failure")),
    )
    message = models.ForeignKey(Message, editable=False, related_name='logs', on_delete=models.CASCADE)
    result = models.CharField(max_length=16, choices=RESULTS, default=SUCCESS)
    date = models.DateTimeField(auto_now_add=True)
    log_message = models.TextField()