from datetime import datetime from django.db import models from django.utils import timezone from django.utils.html import format_html from django.utils.translation import gettext_lazy as _ # maximum time in years of a material part until being replaced MAX_TIME_MATERIAL = 5 class MaterialCategory(models.Model): """ Describes one kind of material """ name = models.CharField(max_length=40, verbose_name=_('Name')) def __str__(self): return self.name class Meta: verbose_name = _('Material category') verbose_name_plural = _('Material categories') # Create your models here. class MaterialPart(models.Model): """ Represents one part of material, which is owned (and stored) by different members of the association (Ownership) """ name = models.CharField(_('name'), max_length=30) description = models.CharField(_('description'), default='', max_length=140) quantity = models.IntegerField(_('quantity'), default=0) buy_date = models.DateField(_('purchase date'), editable=True) lifetime = models.DecimalField(_('lifetime (years)'), decimal_places=0, max_digits=3) photo = models.ImageField(_('photo'), upload_to='images', blank=True) material_cat = models.ManyToManyField(MaterialCategory, default=None, verbose_name=_('Material category')) def __str__(self): """String representation""" return self.name def quantity_real(self): real = sum([o.count for o in Ownership.objects.filter(material__id=self.pk)]) return str(real) + '/' + str(self.quantity) quantity_real.admin_order_field = 'quantity' quantity_real.short_description = _('Quantity') def admin_thumbnail(self): if self.photo: return format_html(''.format(self.photo.url)) else: return format_html('kein Bild') admin_thumbnail.short_description = _('Thumbnail') def ownership_overview(self): summary = '' for owner in self.ownership_set.all(): summary += '

{}: {}

'.format(str(owner.owner), owner.count) return format_html(summary) ownership_overview.short_description = _('Owners') def not_too_old(self): """Returns wether the part should be replaced cause of age""" buy_time = timezone.make_aware(datetime.combine(self.buy_date, datetime.min.time())) return yearsago(self.lifetime) < buy_time not_too_old.admin_order_field = 'buy_date' not_too_old.boolean = True not_too_old.short_description = _('Not too old?') class Meta: verbose_name = _('material part') verbose_name_plural = _('material parts') class Ownership(models.Model): """Represents the connection between a MaterialPart and a Member""" material = models.ForeignKey(MaterialPart, on_delete=models.CASCADE) owner = models.ForeignKey('members.Member', verbose_name=_('owner'), on_delete=models.CASCADE) count = models.IntegerField(_('count'), default=1) def __str__(self): """String representation""" return str(self.owner) class Meta: verbose_name = _('ownership') verbose_name_plural = _('ownerships') def yearsago(years, from_date=None): """Function to return the date with a delta of years in the past""" if from_date is None: from_date = timezone.now() try: return from_date.replace(year=from_date.year - years) except ValueError: # 29.02 -> use 28.02 assert from_date.month == 2 and from_date.day == 29 return from_date.replace(month=2, day=28, year=from_date.year - years)