Changer le statut du consommable au moment du retrait

[ongolaboy.git] / projets_django / bacgl_utils / bureautique / models.py

# -*-coding:utf-8 -*-

import datetime
from django.db import models
from django.utils import timezone


# Create your models here.
class Personnel(models.Model):
    """Infos générique sur le personnel du bureau"""

    nom = models.CharField(max_length=200,primary_key=True)

    def __unicode__(self):
        return self.nom


class Equipement(models.Model):
    marque = models.CharField(max_length=200,default="HP")
    modele = models.CharField(max_length=200,blank=True)
    num_serie = models.CharField("Numéro de série",
            max_length=200,unique=True)
    date_acquisition = models.DateField("Date d'acquisition")


class Imprimante(Equipement):
    """Informations sur les imprimantes du bureau"""

    CATEGORIE_IMPRIMANTE = (
            ("encre","Jet d'encre"),
            ("laser","Laser"),
            ("multifonction","Multifonction"),
            )

    emplacement = models.CharField(max_length=200)
    fqdn = models.CharField("Nom DNS",max_length=200)
    categorie = \
            models.CharField(\
            "Type d'imprimante",
            max_length=20,
            choices=CATEGORIE_IMPRIMANTE
            )

    def __unicode__(self):
        return "Imprimante: %s %s (%s)" %\
            (self.marque,self.modele,self.emplacement)


class Consommable(Equipement):
    """Informations sur les consommables du bureau"""

    CATEGORIE_CONSOMMABLE = (
            ("encre","Jet d'encre"),
            ("toner","Laser"),
            ("autre","autres types"),
            )

    NOM_CONSOMMABLE = (
            ("45","45"),
            ("56","56"),
            ("57","57"),
            ("78","78"),
            ("130","130"),
            ("134","134"),
            ("339","339"),
            ("344","344"),
            ("05A","05A"),
            ("501A","501A"),
            ("920XL","920XL"),
            )

    NOM_COULEUR = (
            ("noir","noir"),
            ("jaune","jaune"),
            ("cyan","cyan"),
            ("magenta","magenta"),
            ("mixte","mixte"),
            )

    nom = models.CharField(max_length=20,choices=NOM_CONSOMMABLE)
    couleur = models.CharField(max_length=20,choices=NOM_COULEUR)
    categorie = \
            models.CharField(max_length=20,choices=CATEGORIE_CONSOMMABLE)
    date_expiration = models.DateField()
    disponible = models.BooleanField(default=True)

    def __unicode__(self):
        return "%s %s (%s)" %(self.nom,self.couleur,self.num_serie)

class RetraitConsommable(models.Model):
    demandeur = models.ForeignKey(Personnel)
    date_retrait = models.DateField("date du retrait",auto_now_add=True)
    code_consommable = models.OneToOneField(Consommable)
    imprimante_utilisee = models.ForeignKey(Imprimante)

    def __unicode__(self):
        annee_date_retrait = self.date_retrait.year
        mois_date_retrait = self.date_retrait.month
        jour_date_retrait = self.date_retrait.day
        return "%s retire pour %s le %s/%s/%s" % (self.code_consommable,
                self.demandeur,jour_date_retrait,mois_date_retrait,
                annee_date_retrait)

    def save(self,*args,**kwargs):
        """
        Lorsqu'on effectue un retrait, il faut absolument changer
        le statut du consommable afin qu'il soit vu comme
        non disponible
        """
        c = Consommable.objects.get(pk=self.code_consommable)
        c.disponible = False
        c.save()
        super(RetraitConsommable,self).save(*args,**kwargs)