axes.py

   1 #-*- coding: utf8 -*-
   2 '''
   3 '''
   4
   5 from openerp.osv import osv, fields
   6 from datetime import date, timedelta, datetime
   7
   8 # ================================ MESURABLE ================================ #
   9 class Mesurable(osv.Model):
  10     _name = "cmmi.axes.mesurable"
  11
  12     _description = "Table de reference des mesusrables."
  13
  14     _states = [("cree", "Crée"), ("encours", "En cours"),
  15                 ("termine", "Terminé"), ("abandonne", "Abandonné"),
  16                 ("suspendu", "Suspendu"), ("generique", "Générique")]
  17
  18
  19     def _nb_jours_init(self, cr, uid, ids, field, arg, context=None):
  20         result = {}
  21         for m in self.browse(cr, uid, ids, context=context):
  22             if not m.date_init_deb or not m.date_init_fin:
  23                 result[m.id] = 0
  24                 continue
  25             result[m.id] = Mesurable._nb_jours_ouvre_entre_2_dates(
  26                         datetime.strptime(m.date_init_deb, "%Y-%m-%d").date(),
  27                         datetime.strptime(m.date_init_fin, "%Y-%m-%d").date())
  28         return result
  29
  30
  31     def _nb_jours_plan(self, cr, uid, ids, field, arg, context=None):
  32         result = {}
  33         for m in self.browse(cr, uid, ids, context=context):
  34             if not m.date_plan_deb or not m.date_plan_fin:
  35                 result[m.id] = 0
  36                 continue
  37             result[m.id] = Mesurable._nb_jours_ouvre_entre_2_dates(
  38                         datetime.strptime(m.date_plan_deb, "%Y-%m-%d").date(),
  39                         datetime.strptime(m.date_plan_fin, "%Y-%m-%d").date())
  40         return result
  41
  42
  43     def _nb_jours_reel(self, cr, uid, ids, field, arg, context=None):
  44         result = {}
  45         for m in self.browse(cr, uid, ids, context=context):
  46             if not m.date_reel_deb or not m.date_reel_fin:
  47                 result[m.id] = 0
  48                 continue
  49             result[m.id] = Mesurable._nb_jours_ouvre_entre_2_dates(
  50                         datetime.strptime(m.date_reel_deb, "%Y-%m-%d").date(),
  51                         datetime.strptime(m.date_reel_fin, "%Y-%m-%d").date())
  52         return result
  53
  54     _columns = {
  55         "name": fields.char(string="Title", size=64, required=True),
  56         "description": fields.text(string="Description"),
  57         "commentaire": fields.text(string="Commentaire"),
  58         "state": fields.selection(_states, string="State"),
  59         "version": fields.char(string="Version", size=16),
  60         "date_jalon": fields.date(string="Jalon"),
  61         "date_init_deb": fields.date(string="Init début"),
  62         "date_init_fin": fields.date(string="Init fin"),
  63         "date_plan_deb": fields.date(string="Plan début"),
  64         "date_plan_fin": fields.date(string="Plan fin"),
  65         "date_reel_deb": fields.date(string="Réel début"),
  66         "date_reel_fin": fields.date(string="Réel fin"),
  67         "nb_jours_init": fields.function(_nb_jours_init,
  68                                             type="integer",
  69                                             string="Nombre de jours initials"),
  70         "nb_jours_plan": fields.function(_nb_jours_plan,
  71                                             type="integer",
  72                                             string="Nombre de jours planifiés"),
  73         "nb_jours_reel": fields.function(_nb_jours_reel,
  74                                             type="integer",
  75                                             string="Nombre de jours réels"),
  76     }
  77
  78     _defaults = {
  79         "state": "cree",
  80     }
  81
  82     _sql_constraints = [
  83         (
  84             "date_init_deb_before_date_init_fin",
  85             "CHECK(date_init_deb <= date_init_fin)",
  86             "The date_init_deb should be previous date_init_fin",
  87         ),
  88         (
  89             "date_plan_deb_before_date_plan_fin",
  90             "CHECK(date_plan_deb <= date_plan_fin)",
  91             "The date_plan_deb should be previous date_plan_fin",
  92         ),
  93         (
  94             "date_reel_deb_before_date_reel_fin",
  95             "CHECK(date_reel_deb <= date_reel_fin)",
  96             "The date_reel_deb should be previous date_reel_fin",
  97         ),
  98     ]
  99
 100
 101     def action_commencer(self, cr, uid, ids, context=None):
 102         if type(ids) == list:
 103             if len(ids) != 1:
 104                 return
 105             ids = ids[0]
 106
 107         palier = self.read(cr, uid, ids, ['date_plan_deb', 'date_plan_fin', 'state'], context)
 108
 109         if palier['state'] != 'cree':
 110             return
 111
 112         self.write(
 113             cr,
 114             uid,
 115             ids, {
 116                 'date_init_deb' : palier['date_plan_deb'],
 117                 'date_init_fin' : palier['date_plan_fin'],
 118                 'state': 'encours'
 119             },
 120             context)
 121         return self
 122
 123
 124     def action_suspendre(self, cr, uid, ids, context=None):
 125         if type(ids) == list:
 126             if len(ids) != 1:
 127                 return # TODO: message d'avertissement
 128             ids = ids[0]
 129
 130         mesurable = self.read(cr, uid, ids, ['state'], context)
 131         if mesurable['state'] != 'encours':
 132             return
 133         self.write(
 134             cr,
 135             uid,
 136             ids,
 137             {'state': 'suspendu'},
 138             context,
 139         )
 140         return self
 141
 142     def action_terminer(self, cr, uid, ids, context=None):
 143         if type(ids) == list:
 144             if len(ids) != 1:
 145                 return # TODO: message d'avertissement
 146             ids = ids[0]
 147
 148         mesurable = self.read(cr, uid, ids, ['state'], context)
 149         if mesurable['state'] != 'encours':
 150             return
 151         self.write(
 152             cr,
 153             uid,
 154             ids,
 155             {'state': 'termine'},
 156             context,
 157         )
 158         return self
 159
 160     def action_abandonner(self, cr, uid, ids, context=None):
 161         if type(ids) == list:
 162             if len(ids) != 1:
 163                 return # TODO: message d'avertissement
 164             ids = ids[0]
 165
 166         mesurable = self.read(cr, uid, ids, ['state'], context)
 167         if not ('encours', 'cree').__contains__(mesurable['state']):
 168             return
 169         self.write(
 170             cr,
 171             uid,
 172             ids,
 173             {'state': 'abandonne'},
 174             context,
 175         )
 176         return self
 177
 178     def action_reprendre(self, cr, uid, ids, context=None):
 179         if type(ids) == list:
 180             if len(ids) != 1:
 181                 return # TODO: message d'avertissement
 182             ids = ids[0]
 183
 184         mesurable = self.read(cr, uid, ids, ['state'], context)
 185         if mesurable['state'] != 'suspendu':
 186             return
 187         self.write(
 188             cr,
 189             uid,
 190             ids,
 191             {'state': 'encours'},
 192             context,
 193         )
 194         return self
 195
 196 #------------ TRAVAIL CALCUL JOURS OUVRES ------------
 197     @staticmethod
 198     def _get_date_paques(annee):
 199         """
 200         Retourne la date du dimanque de pâques pour une année donnée
 201             sous la forme d'un objet date.
 202         """
 203         a = annee % 19
 204         b = annee // 100
 205         c = annee % 100
 206         d = (19 * a + b - b // 4 - ((b - (b + 8) // 25 + 1) // 3) + 15) % 30
 207         e = (32 + 2 * (b % 4) + 2 * (c // 4) - d - (c % 4)) % 7
 208         f = d + e - 7 * ((a + 11 * d + 22 * e) // 451) + 114
 209         mois = f // 31
 210         jours = f % 31 + 1
 211         return date(annee, mois, jours)
 212
 213
 214     @staticmethod
 215     def _get_jours_feries(annee):
 216         """
 217         Retourne une liste contenant les jours fériés d'une année donnée.
 218         """
 219         date_paques = Mesurable._get_date_paques(annee)
 220         return [
 221             date_paques+timedelta(days=1), # Lundi de Pâques
 222             date_paques+timedelta(days=39), # Jeudi de l'Ascension
 223             date_paques+timedelta(days=50), # Lundi de Pentecôte
 224             date(annee, 1, 1), # Jour de l'An
 225             date(annee, 5, 1), # Fête du travail
 226             date(annee, 5, 8), # Armistice 1945
 227             date(annee, 7, 14), # Fête Nationale
 228             date(annee, 8, 15), # Assomption
 229             date(annee, 11, 1), # Toussaint
 230             date(annee, 11, 11), # Armistice 1918
 231             date(annee, 12, 25), # Noël
 232         ]
 233
 234
 235     @staticmethod
 236     def _jour_ouvre(d):
 237         """ Retourne vrai si la date est un jour ouvre, faux sinon."""
 238         if (5,6).__contains__(d.weekday()) or Mesurable._get_jours_feries(d.year).__contains__(d):
 239             return False
 240         else:
 241             return True
 242
 243
 244     @staticmethod
 245     def _nb_jours_ouvre_entre_2_dates(d1, d2):
 246         """
 247         Retourne le nombre de jours ouvres entre deux dates données.
 248
 249         >>> nb_jours_ouvre_entre_2_dates(date(2013, 5, 1), date(2013, 7, 31))
 250         62
 251         >>> nb_jours_ouvre_entre_2_dates(date(2013, 7, 31), date(2013, 5, 1))
 252         62
 253         """
 254         if not d1 or not d2:
 255             return 0
 256         if d1>d2:
 257             d1, d2 = d2, d1 #Switch les 2 dates pour que d1 soit la plus petite
 258         tmp = d1
 259         jour_ouvres = 0
 260         while tmp <= d2:
 261             if Mesurable._jour_ouvre(tmp): # Si tmp est un jour ouvre
 262                 jour_ouvres += 1
 263             tmp += timedelta(days=1)
 264         return jour_ouvres
 265
 266
 267
 268 # ================================= PALIER ================================== #
 269 class Palier(osv.Model):
 270     _name = "cmmi.axes.palier"
 271
 272     _description = "Palier d'un projet."
 273
 274     _inherit = "cmmi.axes.mesurable"
 275
 276     _types_palier = [("normal", "Normal"), ("exceptionnel", "Exceptionnel"),
 277                      ("correctif", "Correctif"), ("autre", "Autre")]
 278
 279     _columns = {
 280         "type_palier": fields.selection(_types_palier, string="Type"),
 281         "projet_id": fields.many2one("cmmi.projet",
 282                                      string="Projet",
 283                                      required=True),
 284         "evolutions": fields.one2many("cmmi.evolution",
 285                                       "palier_id",
 286                                       string="Evolutions"),
 287         "phases": fields.one2many("cmmi.axes.palier.phase",
 288                                   "palier_id",
 289                                   string="Phases"),
 290     }
 291
 292     _defaults = {
 293         "type_palier": "normal",
 294     }
 295
 296
 297     def create(self, cr, uid, vals, context=None):
 298         palier_id = osv.Model.create(self, cr, uid, vals, context=context)
 299
 300         # Récupération des ids de toutes les phases
 301         phase_model = self.pool.get("cmmi.projet.phase")
 302         phases_ids = phase_model.search(cr, uid, [('selectionne', '=', True)])
 303
 304         palier_model = self.pool.get("cmmi.axes.palier")
 305         palier = palier_model.read(cr, uid, palier_id, ['date_plan_deb', 'date_plan_fin'])
 306
 307         # Création des PalierPhase
 308         palier_phase_model = self.pool.get("cmmi.axes.palier.phase")
 309         for phase_id in phases_ids:
 310             palier_phase_model.create(
 311                 cr,
 312                 uid,
 313                 {
 314                     'phase_id': phase_id,
 315                     'palier_id': palier_id,
 316                     'date_plan_deb': palier['date_plan_deb'],
 317                     'date_plan_fin': palier['date_plan_fin'],
 318                 }
 319             )
 320         return palier_id
 321
 322
 323
 324 # =============================== PALIER-PHASE ============================== #
 325 class PalierPhase(osv.Model):
 326     _name = "cmmi.axes.palier.phase"
 327
 328     _description = "Phase d'un palier"
 329
 330     _inherit = "cmmi.axes.mesurable"
 331
 332     def _get_name(self, cr, uid, ids, field_name=None, arg=None, context=None):
 333         if isinstance(ids, (int, long)):
 334             ids = [ids]
 335         return dict([(i, r.phase_id.name) for i, r in
 336                 zip(ids, self.browse(cr, uid, ids, context=context))])
 337
 338
 339     _columns = {
 340         "name": fields.function(_get_name,
 341                                 type='char',
 342                                 store=True,
 343                                 string="Nom de la phase"),
 344         "phase_id": fields.many2one("cmmi.projet.phase",
 345                                     string="Phase du projet"),
 346         "palier_id": fields.many2one("cmmi.axes.palier",
 347                                      string="Palier"),
 348         # backrefs
 349         "charges": fields.one2many("cmmi.evolution.charge",
 350                                    "phase_id",
 351                                    string="Charges"),
 352 #        "evolutions": fields.one2many("cmmi.evolution", #Supprimé !
 353 #                                      "phase_id",
 354 #                                      string="Evolutions"),
 355     }
 356
 357     def create(self, cr, uid, vals, context=None):
 358         # TODO: gérer la création d'une phase de palier.
 359         # Vérifier les valeurs contenues dans vals et les modifier / rajouter si nécessaire selon les cas suivants
 360
 361         # Si description est vide, alors par défaut, recopie de la description du palier et de la phase (concaténés avec un retour à la ligne entre les deux).
 362         # Si commentaire est vide, alors par défaut, recopie du commentaire du palier.
 363         # Si version est vide, alors par dégaut, recopie de la version du palier.
 364
 365         return osv.Model.create(self, cr, uid, vals, context=context)
 366
 367
 368 # ================================ CHANTIER ================================= #
 369 class Chantier(osv.Model):
 370     _name = "cmmi.axes.chantier"
 371
 372     _description = "Chantiers d'un projet."
 373
 374     _inherit = "cmmi.axes.mesurable"
 375
 376     _columns = {
 377         "projet_id": fields.many2one("cmmi.projet",
 378                                      string="Projet",
 379                                      required=True),
 380         "module_ids": fields.many2many("cmmi.description.module",
 381                                        "cmmi_module_chantier_rel",
 382                                        "chantier_id",
 383                                        "module_id",
 384                                        "Modules"),
 385         "evolutions": fields.one2many("cmmi.evolution",
 386                                       "chantier_id",
 387                                       string="Evolutions"),
 388     }