foerderbarometer/input/models.py

90 lines
2.5 KiB
Python

from django.db import models
from .settings import ACCOUNTS
class Volunteer(models.Model):
realname = models.CharField(max_length=200, null=True)
email = models.CharField(max_length=200, null=True)
username = models.CharField(max_length=200, null=True)
granted = models.BooleanField(null=True)
@classmethod
def set_granted(cl, key, b):
obj = cl.objects.get(pk=key)
obj.granted = b
obj.save()
class Meta:
abstract = True
class Project(Volunteer):
name = models.CharField(max_length=200)
start = models.DateField('Startdatum', null=True)
end = models.DateField('Erwartetes Projektende', null=True)
account = models.CharField('Kostenstelle', max_length=5,
choices=ACCOUNTS.items(), null=True,)
# the following Fields are not supposed to be editet by users
pid = models.IntegerField(null=True, blank=True)
end_mail_send = models.BooleanField(null=True)
def save(self,*args,**kwargs):
# is there a way to call super().save() only once?
super().save(*args,*kwargs)
self.pid = 10000 + self.pk
super().save(*args,*kwargs)
def __str__(self):
return f"{self.pid} {self.name}"
class HonoraryCertificate(Volunteer):
request_url = models.CharField(max_length=2000)
project = models.ForeignKey(Project, null = True, on_delete = models.SET_NULL)
def __str__(self):
return "Certificate for " + self.realname
#abstract base class for Library, IFG, ...
class Grant(Volunteer):
cost = models.CharField(max_length=10)
notes = models.CharField(max_length=500)
class Meta:
abstract = True
TYPE_CHOICES = {'BIB': 'Bibliotheksstipendium',
'ELIT': 'eLiteraturstipendium',
'SOFT': 'Softwarestipendium',
'VIS': 'Visitenkarten',
'LIST': 'Mailingliste',
'MAIL': 'E-Mail-Adresse',
'IFG': 'Kostenübernahme IFG-Anfrage',
'LIT': 'Literaturstipendium',}
# same model is used for Library, ELitStip and Software!
class Library(Grant):
type = models.CharField(
max_length=4,
choices=TYPE_CHOICES.items(), #attention: actually only BIB, ELIT, SOFT should be used here
default='LIB',
)
library = models.CharField(max_length=200)
duration = models.CharField(max_length=100)
def __str__(self):
return self.library
class IFG(Grant):
url = models.CharField(max_length=2000)
def __str__(self):
return "IFG-Anfrage von " + self.realname