from django.db import models from django.core.exceptions import ValidationError from django.utils.translation import ugettext_lazy as _ from event.models import Activity, EventSpecificMixin from .fields import IdField from taggit.managers import TaggableManager from datetime import date class EquipmentOwner(models.Model): name = models.CharField( _("nom"), max_length=200, ) class Meta: verbose_name = _("propriétaire de matériel") verbose_name_plural = _("propriétaires de matériel") def __str__(self): return self.name class EquipmentPole(models.Model): name = models.CharField( _("nom"), max_length=200, ) class Meta: verbose_name = _("pôle") verbose_name_plural = _("pôle") def __str__(self): return self.name class EquipmentCategory(models.Model): name = models.CharField( _("nom"), max_length=200, ) class Meta: verbose_name = _("catégories") verbose_name_plural = _("catégories") def __str__(self): return self.name class Equipment(EventSpecificMixin, models.Model): name = models.CharField( _("nom du matériel"), max_length=200, ) stock = models.PositiveSmallIntegerField(_("quantité disponible")) description = models.TextField(_("description")) activities = models.ManyToManyField( Activity, related_name="equipment", through="EquipmentAttribution", ) owner = models.ForeignKey( EquipmentOwner, blank=True, null=True, ) pole = models.ForeignKey( EquipmentPole, ) category = models.ForeignKey( EquipmentCategory, ) class Meta: verbose_name = _("matériel") verbose_name_plural = _("matériels") def __str__(self): return self.name class EquipmentAttribution(models.Model): equipment = models.ForeignKey(Equipment) activity = models.ForeignKey(Activity) amount = models.BigIntegerField(_("quantité attribuée")) remarks = models.TextField(_("remarques concernant l'attribution")) class Meta: verbose_name = _("attribution de matériel") verbose_name_plural = _("attributions de matériel") def __str__(self): return "%s (%d) -> %s" % (self.equipment.name, self.amout, self.activity.get_herited('title')) def save(self, *args, **kwargs): if self.equipment.event and self.equipment.event != self.activity.event: raise ValidationError super(EquipmentAttribution, self).save(*args, **kwargs) class EquipmentRemark(models.Model): remark = models.TextField(_("remarque sur le matériel")) equipment = models.ForeignKey( Equipment, related_name="remarks", help_text=_("Matériel concerné par la remarque"), ) ids = IdField() is_broken = models.BooleanField() is_lost = models.BooleanField() class Meta: verbose_name = _("remarque sur matériel") verbose_name_plural = _("remarques sur le matériel") def __str__(self): return "%s : %s" % (self.equipment.name, self.remark) class EquipmentRevision(models.Model): date = models.DateField( _("date"), default=date.today, ) equipment = models.ForeignKey( Equipment, related_name="revisions", help_text=_("Matériel concerné par les révisions"), ) remark = models.TextField(_("remarque sur la révision")) ids = IdField() class Meta: verbose_name = _("révision de matériel") verbose_name_plural = _("révisions de matériel") def __str__(self): return "%s : %s" % (self.equipment.name, self.remark)