126 lines
3.3 KiB
Python
126 lines
3.3 KiB
Python
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
|
|
|
|
class PostManager(models.Manager):
|
|
def get_queryset(self):
|
|
return super().get_queryset().order_by("-public_date")
|
|
|
|
def get_visible_articles(self):
|
|
return self.get_queryset().filter(is_hidden=False)
|
|
|
|
def all_post_with_date(self):
|
|
return (
|
|
self.get_queryset()
|
|
.filter(Q(event_start__isnull=False) & Q(event_end__isnull=False))
|
|
.order_by("-event_start")
|
|
)
|
|
|
|
|
|
class ArticleManager(models.Manager):
|
|
"""
|
|
Provide a query set only for "Article"
|
|
regular fet meetings should not be contained in the news stream
|
|
"""
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(Q(post_type="E") | Q(post_type="N"))
|
|
.order_by("-public_date")
|
|
)
|
|
|
|
def get_visible_articles(self):
|
|
return self.get_queryset().filter(is_hidden=False)
|
|
|
|
def get_pinned_article(self):
|
|
return self.get_visible_articles().filter(is_pinned=True).first()
|
|
|
|
|
|
class NewsManager(models.Manager):
|
|
"""
|
|
Provide a query set only for "News"
|
|
"""
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().filter(post_type="N").order_by("-public_date")
|
|
|
|
def get_visible_articles(self):
|
|
return self.get_queryset().filter(is_hidden=False)
|
|
|
|
|
|
class AllEventManager(models.Manager):
|
|
"""
|
|
Provide a query set for all events ("Event" and "Fet Meeting")
|
|
"""
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().filter(Q(post_type="E") | Q(post_type="F"))
|
|
|
|
def get_five_events(self):
|
|
date_today = timezone.now()
|
|
return (
|
|
self.get_queryset()
|
|
.filter(event_start__gt=date_today)
|
|
.order_by("event_start")[:5]
|
|
)
|
|
|
|
|
|
class EventManager(models.Manager):
|
|
"""
|
|
Provide a query set only for "Events"
|
|
regular fet meetings should not be contained in the news stream
|
|
"""
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().filter(post_type="E")
|
|
|
|
def get_future_events(self):
|
|
date_today = timezone.now()
|
|
return (
|
|
self.get_queryset()
|
|
.filter(event_start__gt=date_today)
|
|
.order_by("event_start")
|
|
)
|
|
|
|
def get_past_events(self):
|
|
date_today = timezone.now()
|
|
return (
|
|
self.get_queryset()
|
|
.filter(event_start__lt=date_today)
|
|
.order_by("-event_start")
|
|
)
|
|
|
|
|
|
class FetMeetingManager(models.Manager):
|
|
"""
|
|
Provide a query set only for "Fet Meeting"
|
|
"""
|
|
|
|
def get_queryset(self):
|
|
return super().get_queryset().filter(post_type="F")
|
|
|
|
def _get_future_events(self):
|
|
date_today = timezone.now()
|
|
return (
|
|
self.get_queryset()
|
|
.filter(event_start__gt=date_today)
|
|
.order_by("event_start")
|
|
)
|
|
|
|
def _get_past_events(self):
|
|
date_today = timezone.now()
|
|
return (
|
|
self.get_queryset()
|
|
.filter(event_start__lt=date_today)
|
|
.order_by("-event_start")
|
|
)
|
|
|
|
def get_meetings(self):
|
|
meetings = []
|
|
meetings.append(self._get_future_events().first())
|
|
meetings.append(self._get_past_events().first())
|
|
return meetings
|