forked from mirrors/bookwyrm
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
""" admin announcements """
|
|
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
from django.utils.translation import gettext_lazy as _
|
|
|
|
from .base_model import BookWyrmModel
|
|
|
|
|
|
DisplayTypes = [
|
|
("white-ter", _("None")),
|
|
("primary-light", _("Primary")),
|
|
("success-light", _("Success")),
|
|
("link-light", _("Link")),
|
|
("warning-light", _("Warning")),
|
|
("danger-light", _("Danger")),
|
|
]
|
|
|
|
|
|
class Announcement(BookWyrmModel):
|
|
"""The admin has something to say"""
|
|
|
|
user = models.ForeignKey("User", on_delete=models.PROTECT)
|
|
preview = models.CharField(max_length=255)
|
|
content = models.TextField(null=True, blank=True)
|
|
event_date = models.DateTimeField(blank=True, null=True)
|
|
start_date = models.DateTimeField(blank=True, null=True)
|
|
end_date = models.DateTimeField(blank=True, null=True)
|
|
active = models.BooleanField(default=True)
|
|
display_type = models.CharField(
|
|
max_length=20,
|
|
blank=False,
|
|
null=False,
|
|
choices=DisplayTypes,
|
|
default="white-ter",
|
|
)
|
|
|
|
@classmethod
|
|
def active_announcements(cls):
|
|
"""announcements that should be displayed"""
|
|
now = timezone.now()
|
|
return cls.objects.filter(
|
|
Q(start_date__isnull=True) | Q(start_date__lte=now),
|
|
Q(end_date__isnull=True) | Q(end_date__gte=now),
|
|
active=True,
|
|
)
|