foerderbarometer/input/models.py

80 lines
2.3 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('start date')
account = models.CharField(max_length=5, choices=ACCOUNTS.items(), null=True)
pid = models.IntegerField(null=True, blank=True) # automaticly generated
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 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, #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