Files
fet2020/fet2020/members/models.py
Patrick 5232ac5a8a in the job are shown first the active members, second the inactive members.
add short term and slug - as first version.
delete unique filter.
2020-07-14 16:44:58 +00:00

167 lines
4.6 KiB
Python

from django.core.validators import RegexValidator
from django.db import models
from django.db.models import Q
from django.utils import timezone
from django.utils.text import slugify
from django.utils.translation import gettext_lazy as _
from rest_framework import serializers
import uuid
from datetime import timedelta
class MemberManager(models.Manager):
def get_queryset(self):
return super().get_queryset().filter(role='A')
class PensionManager(models.Manager):
def get_queryset(self):
return super().get_queryset().filter(role='P')
class ActiveMemberManager(models.Manager):
'''
return a list of active member, and members who are still working
'''
def get_queryset(self):
date_today = timezone.now().date()
return super().get_queryset().filter(
Q(member__role='A') &
(
Q(job_end__gt=date_today) |
Q(job_end__isnull=True)
)
)
class InactiveMemberManager(models.Manager):
'''
return a list of inactive member
'''
def get_queryset(self):
date_today = timezone.now().date()
return super().get_queryset().filter(
Q(member__role='P') |
(
Q(job_end__lt=date_today + timedelta(days=1)) &
Q(job_end__isnull=False)
)
)
class Member(models.Model):
firstname = models.CharField(max_length=128)
surname = models.CharField(max_length=128)
nickname = models.CharField(max_length=128)
mailaccount = models.CharField(max_length=128)
__choices = [
('A', _('Active')),
('P', _('Pension'))
]
role = models.CharField(max_length=1, choices= __choices, default='A')
description = models.TextField(null=True, blank=True)
image = models.ImageField(null=True, blank=True)
birthday = models.DateField(null=True, blank=True)
phone_error_msg =_((
"Phone number must be entered in the format: "
"+999999999'. Up to 15 digits allowed."
))
phone_regex = RegexValidator(regex=r'^\+?1?\d{9,15}$', message=phone_error_msg)
phone = models.CharField(validators=[phone_regex], max_length=17, blank=True)
address = models.TextField(null=True, blank=True)
date_modified = models.DateTimeField(auto_now=True)
date_created = models.DateTimeField(auto_now_add=True)
has_jobs = models.ManyToManyField(
'Job',
through='JobMember',
through_fields=('member', 'job'),
)
# Managers
all_members = models.Manager()
active_member = MemberManager()
pension_member = PensionManager()
class Meta:
verbose_name = "Mitglied"
verbose_name_plural = "Mitglieder"
def __str__(self):
return self.firstname + " " + self.surname
class MemberSerializer(serializers.HyperlinkedModelSerializer):
class Meta:
model = Member
fields = [
'firstname',
'surname',
'nickname',
'mailaccount',
'role',
'description',
'image'
]
class Job(models.Model):
name = models.CharField(max_length=128)
shortterm = models.CharField(max_length=128)
slug = models.SlugField(unique=True, null=True, blank=True)
description = models.TextField(null=True, blank=True)
image = models.ImageField(null=True, blank=True)
has_members = models.ManyToManyField(
'Member',
through='JobMember',
through_fields=('job', 'member'),
)
class Meta:
verbose_name = "Tätigkeit"
verbose_name_plural = "Tätigkeiten"
def save(self, *args, **kwargs):
if not self.slug:
self.slug = slugify(self.shortterm)
super().save(*args, **kwargs)
def __str__(self):
return self.shortterm
class JobMember(models.Model):
member = models.ForeignKey(
Member,
on_delete=models.CASCADE,
verbose_name = "Mitglied",
)
job = models.ForeignKey(
Job,
on_delete=models.CASCADE,
verbose_name = "Tätigkeit",
)
job_start = models.DateField('Job Start')
job_end = models.DateField('Job Ende', null=True, blank=True)
__choices = [
('1V', _('VorsitzendeR')),
('2V', _('stv VorsitzendeR')),
('3V', _('2. stv VorsitzendeR')),
('M', _('Mitglied')),
('E', _('Ersatzmitglied')),
('V', _('VerantwortlicheR'))
]
job_role = models.CharField(max_length=2, choices=__choices, default='M')
jobmember = models.Manager()
active_member = ActiveMemberManager()
inactive_member = InactiveMemberManager()