update members view
This commit is contained in:
@@ -9,7 +9,7 @@ from datetime import timedelta
|
||||
from easy_thumbnails.fields import ThumbnailerImageField
|
||||
|
||||
|
||||
class ActiveMemberManager(models.Manager):
|
||||
class ActiveJobMemberManager(models.Manager):
|
||||
'''
|
||||
return a list of active member, and members who are still working
|
||||
'''
|
||||
@@ -18,14 +18,15 @@ class ActiveMemberManager(models.Manager):
|
||||
|
||||
def get_queryset(self):
|
||||
date_today = timezone.now().date()
|
||||
qs = super().get_queryset().order_by('member__firstname')
|
||||
|
||||
return super().get_queryset().filter(
|
||||
return qs.filter(
|
||||
Q(member__role=Member.MemberRole.ACTIVE)
|
||||
& (Q(job_end__gt=date_today) | Q(job_end__isnull=True))
|
||||
)
|
||||
|
||||
|
||||
class InactiveMemberManager(models.Manager):
|
||||
class InactiveJobMemberManager(models.Manager):
|
||||
'''
|
||||
return a list of inactive member
|
||||
'''
|
||||
@@ -34,16 +35,19 @@ class InactiveMemberManager(models.Manager):
|
||||
|
||||
def get_queryset(self):
|
||||
date_today = timezone.now().date()
|
||||
qs = super().get_queryset().order_by('member__firstname')
|
||||
|
||||
return super().get_queryset().filter(
|
||||
return qs.filter(
|
||||
Q(member__role=Member.MemberRole.PENSION)
|
||||
| (Q(job_end__lt=date_today + timedelta(days=1)) & Q(job_end__isnull=False))
|
||||
)
|
||||
|
||||
|
||||
class MemberManager(models.Manager):
|
||||
class JobMemberManager(models.Manager):
|
||||
def get_members(self, role):
|
||||
return self.get_queryset().filter(Q(member__role=role))
|
||||
qs = self.get_queryset().order_by('member__firstname')
|
||||
|
||||
return qs.filter(Q(member__role=role))
|
||||
|
||||
|
||||
class JobGroupManager(models.Manager):
|
||||
@@ -76,6 +80,11 @@ class JobGroupManager(models.Manager):
|
||||
return job_groups
|
||||
|
||||
|
||||
class MemberManager(models.Manager):
|
||||
def get_queryset(self):
|
||||
return super().get_queryset().order_by('firstname')
|
||||
|
||||
|
||||
class Member(models.Model):
|
||||
firstname = models.CharField("Vorname", max_length=128)
|
||||
surname = models.CharField("Nachname", max_length=128)
|
||||
@@ -110,7 +119,7 @@ class Member(models.Model):
|
||||
date_created = models.DateTimeField(auto_now_add=True)
|
||||
|
||||
# Managers
|
||||
all_members = models.Manager()
|
||||
all_members = MemberManager()
|
||||
|
||||
class Meta:
|
||||
verbose_name = "Mitglied"
|
||||
@@ -207,6 +216,6 @@ class JobMember(models.Model):
|
||||
|
||||
job_role = models.CharField(max_length=2, choices=JobRole.choices, default=JobRole.MEMBER)
|
||||
|
||||
members = MemberManager()
|
||||
active_member = ActiveMemberManager()
|
||||
inactive_member = InactiveMemberManager()
|
||||
members = JobMemberManager()
|
||||
active_member = ActiveJobMemberManager()
|
||||
inactive_member = InactiveJobMemberManager()
|
||||
|
||||
@@ -93,7 +93,7 @@ class MemberViewSet(viewsets.ModelViewSet):
|
||||
"""
|
||||
API endpoint that allows users to be viewed or edited.
|
||||
"""
|
||||
queryset = Member.all_members.all().order_by('nickname')
|
||||
queryset = Member.all_members.order_by('nickname')
|
||||
serializer_class = MemberSerializer
|
||||
|
||||
#permission_classes = [permissions.IsAuthenticated]
|
||||
|
||||
Reference in New Issue
Block a user