ajouté l'initialisation d'une simulation

[auf_rh_dae.git] / project / budget / models.py

# -=- encoding: utf-8 -=-

from django.db import models

from rh import models as rh


class Simulation(models.Model):
    nom = models.CharField(max_length=255,
                            verbose_name=u'Nom de la simulation')

    def copy_rh_model(self, obj):
        """ Crée une copie de simulation d'un modèle rh. """
        initial = dict(['simulation', self] +
                       [(f.name, getattr(obj, f.name))
                        for f in obj._meta.fields
                        if not isinstance(f, models.AutoField) and
                           not f in obj._meta.parents.values()])
        return globals()[obj.__class__.__name__](**initial)


    def initialize(self):
        """ Copie les données nécessaires à une simulation. """
        if (self.poste_set.all().count() > 0 or
          self.dossier_set.all().count() > 0):
            raise ValueError(u'Cette simulation a déjà été initialisée.')

        postes = {}
        for rh_p in rh.Poste.objects.all():
            sim_p = self.copy_rh_model(rh_p)
            sim_p.save()
            postes[rh_p.id] = sim_p

        dossiers = {}
        for rh_d in rh.Dossier.objects.all():
            sim_d = self.copy_rh_model(rh_d)
            if not sim_d.poste.id in postes:
                self.poste_set.all().delete()
                self.dossier_set.all().delete()
                raise ValueError(u'Données inconsistantes.')
            sim_d.poste = postes[sim_d.poste.id]
            sim_d.save()
            dossiers[rh_d.id] = sim_d

class Poste(rh.Poste_):
    __doc__ = rh.Poste_.__doc__
    simulation = models.ForeignKey(Simulation)


class Dossier(rh.Dossier_):
    __doc__ = rh.Dossier_.__doc__
    simulation = models.ForeignKey(Simulation)