svthalia/concrexit

"""Models for the promotion requests database tables."""

from django.db import models
from django.dispatch import Signal
from django.utils import timezone
from django.utils.translation import gettext_lazy as _

from tinymce.models import HTMLField

from events.models import Event

updated_status = Signal()


class PromotionChannel(models.Model):
    name = models.CharField(verbose_name=_("Channel name"), max_length=100)
    publisher_reminder_email = models.EmailField(
        verbose_name="Publisher Reminder Email",
        blank=True,
        null=True,
    )
    publish_deadline = models.DurationField(
        verbose_name="Minimum time between request and publish",
        default=timezone.timedelta(weeks=2),
    )

    def __str__(self):
        return str(self.name)


class UpcomingRequestManager(models.Manager):
    def get_queryset(self):
        end_date = timezone.localdate()
        start_date = end_date - timezone.timedelta(weeks=1)
        return super().get_queryset().filter(created_at__range=(start_date, end_date))


class NewRequestManager(models.Manager):
    def get_queryset(self):
        start_date = timezone.localtime()
        end_date = start_date + timezone.timedelta(weeks=1)
        return super().get_queryset().filter(publish_date__range=(start_date, end_date))


class PromotionRequest(models.Model):
    objects = models.Manager()
    upcoming_requests = UpcomingRequestManager()
    new_requests = NewRequestManager()

    created_at = models.DateTimeField(
        verbose_name=_("created at"), auto_now_add=True, null=False, blank=False
    )
    event = models.ForeignKey(
        Event, verbose_name=_("event"), on_delete=models.CASCADE, null=True, blank=True
    )
    publish_date = models.DateField(
        verbose_name=_("Publish date"),
        null=False,
        blank=False,
    )
    channel = models.ForeignKey(
        PromotionChannel,
        verbose_name=_("channel"),
        on_delete=models.CASCADE,
        null=False,
        blank=False,
    )
    assigned_to = models.CharField(
        null=True,
        blank=True,
        max_length=50,
        verbose_name=_("Assigned to"),
    )

    NOT_STARTED = "not_started"
    STARTED = "started"
    FINISHED = "finished"
    PUBLISHED = "published"

    STATUS_TYPES = (
        (NOT_STARTED, _("Not started")),
        (STARTED, _("Started")),
        (FINISHED, _("Finished")),
        (PUBLISHED, _("Published")),
    )

    status = models.CharField(
        max_length=40,
        choices=STATUS_TYPES,
        verbose_name=_("status"),
        default=NOT_STARTED,
        null=False,
        blank=False,
    )
    drive_folder = models.URLField(
        verbose_name=_("drive folder"),
        null=True,
        blank=True,
        max_length=2000,  # This appears to be the max allowed url length
    )
    remarks = HTMLField(
        verbose_name=_("remarks"),
        null=True,
        blank=True,
    )
    status_updated = models.BooleanField(
        verbose_name=_("status updated"),
        default=False,
    )

    def __str__(self):
        if self.event:
            return _("Promotion request for ") + str(self.event)
        return _("Promotion request ") + str(self.pk)

    def save(self, **kwargs):
        if not self.publish_date and self.event:
            self.publish_date = self.event.start.date()
        oldstatus = None
        if self.pk:
            oldstatus = PromotionRequest.objects.get(pk=self.pk).status
            if not self.status_updated:
                self.status_updated = oldstatus != self.status
        ret = super().save(**kwargs)
        if oldstatus and oldstatus != self.status:
            updated_status.send(sender=None, updated_request=self)
        return ret

    class Meta:
        verbose_name = _("Promotion request")
        verbose_name_plural = _("Promotion requests")