83 lines
2.4 KiB
Python
83 lines
2.4 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()
|
|
|
|
def get_visible_articles(self):
|
|
return self.get_queryset().filter(is_hidden=False)
|
|
|
|
|
|
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')
|
|
|
|
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):
|
|
return self.get_queryset().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')
|
|
|
|
|
|
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
|