add pinned news, delete db vlaue 'is_event'
This commit is contained in:
73
fet2020/posts/managers.py
Normal file
73
fet2020/posts/managers.py
Normal file
@@ -0,0 +1,73 @@
|
||||
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()
|
||||
|
||||
|
||||
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_pinned_article(self):
|
||||
return self.get_queryset().filter(is_pinned=True)
|
||||
|
||||
|
||||
class NewsManager(models.Manager):
|
||||
"""
|
||||
Provide a query set only for "News"
|
||||
"""
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().filter(post_type='N')
|
||||
|
||||
|
||||
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
|
||||
Reference in New Issue
Block a user