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