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